Update app.py

app.py

@@ -1,30 +1,22 @@
-# app.py — Gradio interface for EvoDecoder conversational chatbot
+# app.py — Gradio chat interface for EvoDecoder with optional RAG (web search)
 import gradio as gr
 from generate import generate_response
 
-# Initialize Gradio Chatbot with OpenAI-style message format
-with gr.Blocks(title="🧠 Chat with EvoDecoder — Lightweight Conversational AI") as demo:
-    gr.Markdown("## 🧠 Chat with EvoDecoder — Lightweight Conversational AI")
-
+with gr.Blocks(title="🧠 EvoDecoder Chat with RAG") as app:
+    gr.Markdown("## 🧠 Chat with EvoDecoder — Now with 🔍 Web Context (RAG)")
     chatbot = gr.Chatbot(label="Chat with Evo", type="messages")
-    msg = gr.Textbox(placeholder="Ask Evo anything...", label="Your message")
-    clear = gr.Button("Clear Chat")
+    msg = gr.Textbox(label="Ask Evo anything...", placeholder="e.g., What is quantum computing?")
+    use_web = gr.Checkbox(label="🔍 Use Web Context (RAG)", value=True)
+    clear = gr.Button("🧹 Clear Chat")
 
-    def chat_fn(history, message):
-        """Handles each user input and appends Evo's response."""
+    def chat_fn(user_message, history, web_flag):
+        response = generate_response(user_message, use_web=web_flag)
         history = history or []
-        response = generate_response(message)
-        history.append({"role": "user", "content": message})
+        history.append({"role": "user", "content": user_message})
         history.append({"role": "assistant", "content": response})
-        return history
-
-    def clear_fn():
-        """Clears the chat history."""
-        return []
+        return history, ""
 
-    msg.submit(chat_fn, [chatbot, msg], chatbot)
-    msg.submit(lambda: "", None, msg)
-    clear.click(clear_fn, outputs=chatbot)
+    msg.submit(chat_fn, inputs=[msg, chatbot, use_web], outputs=[chatbot, msg])
+    clear.click(lambda: ([], ""), outputs=[chatbot, msg])
 
-# Launch the app
-demo.launch()
+app.launch()