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()