Update app.py

app.py

@@ -77,24 +77,25 @@ After writing the document, please provide a list of sources used in your respon
     else:
         return f"Error: API returned status code {response.status_code}", ""
 
-def gradio_interface(query):
-    main_content, sources = get_response_with_search(query)
+def chatbot_interface(message, history):
+    main_content, sources = get_response_with_search(message)
     formatted_response = f"{main_content}\n\nSources:\n{sources}"
     return formatted_response
 
-# Gradio interface
-iface = gr.Interface(
-    fn=gradio_interface,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your question here..."),
-    outputs="text",
+# Gradio chatbot interface
+iface = gr.ChatInterface(
+    fn=chatbot_interface,
     title="AI-powered Web Search Assistant",
-    description="Ask a question, and I'll search the web and provide an answer using the Mistral-7B-Instruct model.",
+    description="Ask questions, and I'll search the web and provide answers using the Mistral-7B-Instruct model.",
     examples=[
-        ["Latest news about Yann LeCun"],
-        ["Latest news site:github.blog"],
-        ["Where I can find best hotel in Galapagos, Ecuador intitle:hotel"],
-        ["filetype:pdf intitle:python"]
-    ]
+        ["What are the latest developments in AI?"],
+        ["Tell me about recent updates on GitHub"],
+        ["What are the best hotels in Galapagos, Ecuador?"],
+        ["Summarize recent advancements in Python programming"],
+    ],
+    retry_btn="Retry",
+    undo_btn="Undo",
+    clear_btn="Clear",
 )
 
 if __name__ == "__main__":