feat: update app

app.py

@@ -49,6 +49,7 @@ class MdnaQA:
         return self.chain.run(input_documents=input_documents, question=question)
 
 
+
 filename = '2023-05-12_2023_q1_goog_mdna.txt'
 loader = TextLoader(filename)
 documents = loader.load()
@@ -58,12 +59,13 @@ docs = split_documents(documents, tiktoken_len)
 tokens_sum = sum(tiktoken_len(d.page_content) for d in docs)
 
 title = "Alphabet's Q1 2023 10-Q MD&A"
+video = '<iframe width="560" height="315" src="https://www.youtube.com/embed/mI6_SF3bYJs" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" allowfullscreen></iframe>'
 
 with gr.Blocks(title=title) as demo:
     gr.Markdown(f'# {title}')
-    gr.Markdown('Video tutorial ')
-    gr.Markdown('Link to a blog post the video tutorial')
-    gr.Markdown("You can an API key [from OpenAI](https://platform.openai.com/account/api-keys)")
+    # gr.HTML(video)
+    gr.Markdown('Blog post https://blog.experienced.dev')
+    gr.Markdown("You can get an API key [from OpenAI](https://platform.openai.com/account/api-keys)")
     openai_api_key = gr.Text(
         value=os.getenv("OPENAI_API_KEY"),
         type="password",
@@ -80,8 +82,8 @@ with gr.Blocks(title=title) as demo:
 
             def summarize_mdna(docs, api_key, temp):
                 llm = OpenAI(temperature=temp, openai_api_key=api_key)
-                # mdna_summary = summarize_docs(llm, docs)
-                return 'HaHa'
+                mdna_summary = summarize_docs(llm, docs)
+                return mdna_summary
 
             summarize.click(summarize_mdna, inputs=[mdna, openai_api_key, temperature], outputs=[summary])
         with gr.TabItem("QA with MD&A"):
@@ -94,8 +96,7 @@ with gr.Blocks(title=title) as demo:
             send = gr.Button("Ask question", variant='primary', visible=False)
 
             def start_chat(docs, openai_api_key):
-                # qa_chat = MdnaQA(docs, openai_api_key)
-                qa_chat = MDNAQAMock()
+                qa_chat = MdnaQA(docs, openai_api_key)
                 return (
                     qa_chat,
                     gr.Textbox.update(visible=True),
@@ -110,7 +111,6 @@ with gr.Blocks(title=title) as demo:
             def respond(qa_chat, question, chat_history):
                 answer = qa_chat.ask(question)
                 chat_history.append((question, answer))
-                time.sleep(3)
                 return "", chat_history
 
             send.click(respond, [qa_chat, question, chatbot], [question, chatbot])