Upload app.py

app.py

@@ -0,0 +1,72 @@
+# Importamos la librería
+from youtube_transcript_api import YouTubeTranscriptApi
+import re
+from langchain.vectorstores import FAISS
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain.embeddings import AwaEmbeddings
+import os
+import openai
+import gradio as gr
+
+
+def get_transcript(url):
+  video_id = re.search(r"(?<=v=)([^&#]+)", url)
+  video_id = video_id.group(0)
+
+    # retrieve the available transcripts
+  transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
+
+  # iterate over all available transcripts
+  for transcript in transcript_list:
+    subtitles = transcript.translate('en').fetch()
+
+  # Imprimimos los transcript
+  text = ''
+  for sub in subtitles:
+      text = text + ' ' + sub['text']
+  return text
+
+
+embeddings = AwaEmbeddings()
+
+text_splitter = RecursiveCharacterTextSplitter(
+    # Set a really small chunk size, just to show.
+    chunk_size = 1500,
+    chunk_overlap  = 100,
+    length_function = len,
+    is_separator_regex = False,
+)
+
+def chat(url, question, api_key):
+  os.environ["OPENAI_API_KEY"] = api_key
+  openai.api_key = api_key
+
+  info = get_transcript(url)
+  texts = text_splitter.create_documents([info])
+  db = FAISS.from_documents(texts, embeddings)
+  docs = db.similarity_search(question)
+
+  prompt = [
+      {"role": "system", "content": """You are my Youtube Asisstant. I will pass you texts from a Youtube Video Transcrip and I need you to use them to answer my question from the Youtube Video.
+      Please do not invent any information, and I am asking about information in the Youtube Video."""},
+      {"role":"user", "content": f"Context:{docs}"},
+      {"role":"user", "content": f"Question:{question}"},
+  ]
+  response = openai.ChatCompletion.create(
+    model="gpt-3.5-turbo-0613",
+    messages=prompt,
+    temperature = 0
+    )
+
+  return response["choices"][0]["message"]["content"]
+
+
+# Create a gradio interface with two inputs and one output
+demo = gr.Interface(
+  fn=chat, # The function to call
+  inputs=[gr.Textbox(label="Youtube URL"), gr.Textbox(label="Question"), gr.Textbox(label="API_KEY")], # The input components
+  outputs=gr.Textbox(label="Answer") # The output component
+)
+
+# Launch the interface
+demo.launch(share=True)