feat: update app

app.py

@@ -50,7 +50,7 @@ class MdnaQA:
 
 
 
-filename = '2023-05-12_2023_q1_goog_mdna.txt'
+filename = "2023-05-12_2023_q1_goog_mdna.txt"
 loader = TextLoader(filename)
 documents = loader.load()
 model_name = "text-davinci-003"
@@ -62,50 +62,73 @@ 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(f"# {title}")
     # 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)")
+    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",
         label="OpenAI API key",
     )
-    temperature = gr.Slider(0, 2, value=0, step=0.1, label="Temperature", info="adjusts a model's output from predictable to random")
+    temperature = gr.Slider(
+        0,
+        2,
+        value=0,
+        step=0.1,
+        label="Temperature",
+        info="adjusts a model's output from predictable to random",
+    )
     mdna = gr.State(docs)
-    tokens_total = gr.Textbox(label="Total input tokens", value=tokens_sum, info='how many tokens will be spent on input / embeddings')
+    tokens_total = gr.Textbox(
+        label="Total input tokens",
+        value=tokens_sum,
+        info="how many tokens will be spent on input / embeddings",
+    )
     with gr.Tabs(visible=True) as tabs:
         with gr.TabItem("Summary"):
-            
-            summarize = gr.Button("Summarize MD&A", variant='primary', info='On click you spent tokens on input, instructions and output')
-            summary = gr.TextArea(label='Summary')
+            summarize = gr.Button(
+                "Summarize MD&A",
+                variant="primary",
+                info="On click you spent tokens on input, instructions and output",
+            )
+            summary = gr.TextArea(label="Summary")
 
             def summarize_mdna(docs, api_key, temp):
                 llm = OpenAI(temperature=temp, openai_api_key=api_key)
                 mdna_summary = summarize_docs(llm, docs)
                 return mdna_summary
 
-            summarize.click(summarize_mdna, inputs=[mdna, openai_api_key, temperature], outputs=[summary])
+            summarize.click(
+                summarize_mdna,
+                inputs=[mdna, openai_api_key, temperature],
+                outputs=[summary],
+            )
         with gr.TabItem("QA with MD&A"):
-            start_qa = gr.Button("Start QA with MD&A", variant='primary')
+            start_qa = gr.Button("Start QA with MD&A", variant="primary")
             chatbot = gr.Chatbot(label="QA with MD&A", visible=False)
             question = gr.Textbox(
                 label="Your question", interactive=True, visible=False
             )
             qa_chat = gr.State()
-            send = gr.Button("Ask question", variant='primary', visible=False)
+            send = gr.Button("Ask question", variant="primary", visible=False)
 
-            def start_chat(docs, openai_api_key):
-                qa_chat = MdnaQA(docs, openai_api_key)
+            def start_chat(docs, api_key, temp):
+                llm = OpenAI(temperature=temp, openai_api_key=api_key)
+                qa_chat = MdnaQA(llm, docs)
                 return (
                     qa_chat,
                     gr.Textbox.update(visible=True),
                     gr.Textbox.update(visible=True),
-                    gr.Button.update(visible=True)
+                    gr.Button.update(visible=True),
                 )
 
             start_qa.click(
-                start_chat, [openai_api_key], [qa_chat, chatbot, question, send]
+                start_chat,
+                [mdna, openai_api_key, temperature],
+                [qa_chat, chatbot, question, send],
             )
 
             def respond(qa_chat, question, chat_history):
@@ -114,9 +137,7 @@ with gr.Blocks(title=title) as demo:
                 return "", chat_history
 
             send.click(respond, [qa_chat, question, chatbot], [question, chatbot])
-            question.submit(
-                respond, [qa_chat, question, chatbot], [question, chatbot]
-            )
+            question.submit(respond, [qa_chat, question, chatbot], [question, chatbot])
 
 
 demo.launch()