Update app.py

app.py

@@ -1,31 +1,28 @@
 import gradio as gr
 from generate import generate_response
-from web_search import web_search
 
-def chat_with_evo(prompt, temperature=1.0, use_rag=False):
+def chat_interface(question, context, use_rag, temperature):
     try:
-        context = web_search(prompt) if use_rag else ""
-        response = generate_response(prompt, temperature=temperature, external_context=context)
-        return response if response else "(No meaningful response generated.)"
+        answer = generate_response(question, context, use_rag=use_rag, temperature=temperature)
+        return answer if answer else "(No meaningful response generated.)"
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"(Error: {str(e)})"
 
-with gr.Blocks(title="EvoRAG Chatbot") as demo:
-    gr.Markdown("## 🧠 EvoRAG — Lightweight Chatbot with Real-Time Retrieval")
+title = "🧠 EvoRAG — SQuAD-Tuned QA with Optional Web Retrieval"
+description = "Ask a question with optional context. Evo was fine-tuned on SQuAD v2.\nToggle RAG for live web search if no context is available."
 
-    with gr.Row():
-        prompt = gr.Textbox(label="💬 Ask Anything", placeholder="Type your question here...", lines=2)
-        response = gr.Textbox(label="🤖 Evo's Answer", interactive=False)
+iface = gr.Interface(
+    fn=chat_interface,
+    inputs=[
+        gr.Textbox(label="❓ Question"),
+        gr.Textbox(label="📄 Context (Optional)", lines=5, placeholder="Paste context or leave blank to use RAG"),
+        gr.Checkbox(label="🔍 Use RAG (Live Web Search)"),
+        gr.Slider(0.1, 1.5, value=1.0, label="🔥 Temperature")
+    ],
+    outputs=gr.Textbox(label="🤖 Evo's Answer"),
+    title=title,
+    description=description,
+)
 
-    with gr.Row():
-        temp = gr.Slider(0.1, 1.5, step=0.1, value=1.0, label="🔥 Temperature")
-        use_rag = gr.Checkbox(label="🔍 Use RAG (live web context)", value=True)
-
-    with gr.Row():
-        clear_btn = gr.Button("🧹 Clear")
-        send_btn = gr.Button("🚀 Ask Evo")
-
-    send_btn.click(fn=chat_with_evo, inputs=[prompt, temp, use_rag], outputs=response)
-    clear_btn.click(fn=lambda: ("", ""), inputs=[], outputs=[prompt, response])
-
-demo.launch()
+if __name__ == "__main__":
+    iface.launch()