File size: 2,015 Bytes
9db894e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
import os
from typing import List
import gradio as gr
from qa.manager import YoutubeQA
DESCRIPTION = """
<h1> <center> 🤗 Hello. This App will help you do questions on youtube videos.</center> </h1>
<h4>
Follow this steps to use 😉:
</h4>
<ol>
<li>Set your OpenAI Key</li>
<li>Set your Youtube URL</li>
<li>Ask!</li>
</ol>
"""
qa = YoutubeQA()
def set_openai_key(key: str):
os.environ["OPENAI_API_KEY"] = key
# Set status field to Not Ready
return gr.update(lines=1, value="Not Ready 🥴")
def instanciate_retriver(url: str):
qa.load_model()
qa.load_vector_store(url)
qa.load_retriever()
# Set status field to Ready
return gr.update(lines=1, value="Ready 😎")
def respond(message: str, chat_history: List[str]):
bot_message = qa.run(message)
chat_history.append((message, bot_message))
return "", chat_history
with gr.Blocks() as app:
gr.Markdown(DESCRIPTION)
with gr.Tab("QA"):
status = gr.Textbox(label="🤔 Vector DB Status:", interactive=False)
chatbot = gr.Chatbot(label="🤖 Bot Answer:")
question = gr.Textbox(label="🕵️♀️ Question:", placeholder="Write your question here and press enter")
clear = gr.Button("Clear")
question.submit(respond, [question, chatbot], [question, chatbot])
clear.click(lambda: None, None, chatbot, queue=False)
with gr.Tab("Youtube URL"):
url = gr.Textbox(label="🎞️ URL:", lines=1, placeholder="Set your Youtube URL here...")
url_button = gr.Button("Set URL")
with gr.Tab("OpenAI Key"):
key = gr.Textbox(label="🔑 Key:", type="password", placeholder="Set your OpenAI Key here...")
key_button = gr.Button("Set Key")
#with gr.Accordion("Click me. About this App"):
# gr.Markdown("Look at me...")
url_button.click(instanciate_retriver, inputs=url, outputs=status)
key_button.click(set_openai_key, inputs=key, outputs=status)
app.launch() |