Update app.py

app.py

@@ -1,6 +1,8 @@
 import os
 from getpass import getpass
 import gradio as gr
+import random
+import time
 
 pinecone_api_key = os.getenv("PINECONE_API_KEY") or getpass("Enter your Pinecone API Key: ")
 openai_api_key = os.getenv("OPENAI_API_KEY") or getpass("Enter your OpenAI API Key: ")
@@ -63,13 +65,28 @@ def query_anual_report(query):
     response = query_engine.query(query)
     return response.response
 
-# Define Gradio Interface
-iface = gr.Interface(
-    fn=query_anual_report,
-    inputs=gr.Textbox(lines=2, placeholder="Ask something..."),
-    outputs="text",
-    title="Annual Report Query",
-    description="Ask questions about the annual report."
-)
-
-iface.launch()
+# Define the chat functions
+def user(user_message, history):
+    return "", history + [[user_message, None]]
+
+def bot(history):
+    bot_message = query_anual_report(history[-1][0])
+    history[-1][1] = ""
+    for character in bot_message:
+        history[-1][1] += character
+        time.sleep(0.05)
+        yield history
+
+# Define Gradio Blocks interface
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.Button("Clear")
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, chatbot, chatbot
+    )
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+if __name__ == "__main__":
+    demo.launch()