Update app.py

app.py

@@ -1,37 +1,31 @@
-# app.py
-
 import gradio as gr
 from generate import generate_response
+from web_search import web_search
 
-chat_history = []
-
-def chat(user_input):
-    global chat_history
-
-    # Append user input
-    chat_history.append(f"User: {user_input}")
-
-    # Prepare full conversation context
-    context = "\n".join(chat_history)
-    evo_reply = generate_response(context)
-
-    # Append Evo's response
-    chat_history.append(f"Evo: {evo_reply}")
+def chat_with_evo(prompt, temperature=1.0, use_rag=False):
+    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.)"
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-    # Keep last 10 turns to avoid overlong context
-    if len(chat_history) > 20:
-        chat_history = chat_history[-20:]
+with gr.Blocks(title="EvoRAG Chatbot") as demo:
+    gr.Markdown("## 🧠 EvoRAG — Lightweight Chatbot with Real-Time Retrieval")
 
-    return "\n".join(chat_history)
+    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)
 
-with gr.Blocks() as demo:
-    gr.Markdown("### 🧠 Evo Chat (Stateful Level 2)")
-    chatbot = gr.Textbox(label="Chat History", lines=15, interactive=False)
-    user_input = gr.Textbox(label="Your Message")
+    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)
 
-    def submit_message(message):
-        return chat(message)
+    with gr.Row():
+        clear_btn = gr.Button("🧹 Clear")
+        send_btn = gr.Button("🚀 Ask Evo")
 
-    user_input.submit(submit_message, inputs=user_input, outputs=chatbot)
+    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()