Update app.py

app.py

@@ -1,23 +1,35 @@
-# app.py — EvoChat UI with live generation
+# app.py — Gradio interface for EvoDecoder Chatbot with full conversation context
+
 import gradio as gr
 from generate import generate_response
 
-with gr.Blocks(title="EvoChat — Lightweight Conversational AI") as app:
-    gr.Markdown("## 🤖 EvoChat — Trained on 1K samples | EvoDecoder")
-    
-    chatbot = gr.Chatbot(label="Chat with Evo")
-    msg = gr.Textbox(label="Your message", placeholder="Ask something...")
-    clear = gr.Button("Clear")
+# Define chat function
+def chat_fn(history, message):
+    # Merge all past messages to provide conversation context
+    context = ""
+    for item in history:
+        context += f"User: {item['user']}\nAssistant: {item['assistant']}\n"
+    context += f"User: {message}\n"
+
+    # Generate Evo's reply based on full context
+    reply = generate_response(context)
+
+    # Append the new exchange to history
+    history.append({"user": message, "assistant": reply})
+    return history, ""
+
+# Gradio UI setup
+with gr.Blocks() as demo:
+    gr.Markdown("## 🧠 Chat with EvoDecoder — A Lightweight Conversational AI")
 
-    history = []
+    chatbot = gr.Chatbot(label="Chat with Evo", type="messages")
+    state = gr.State([])
 
-    def chat_fn(message):
-        history.append(("User: " + message, None))
-        response = generate_response(message)
-        history[-1] = (history[-1][0], response)
-        return history
+    with gr.Row():
+        msg = gr.Textbox(label="Your message", placeholder="Ask Evo anything...")
+        send_btn = gr.Button("Send")
 
-    msg.submit(chat_fn, inputs=msg, outputs=chatbot)
-    clear.click(lambda: [], None, chatbot)
+    send_btn.click(chat_fn, inputs=[state, msg], outputs=[chatbot, msg])
+    msg.submit(chat_fn, inputs=[state, msg], outputs=[chatbot, msg])  # Enter key submits too
 
-app.launch()
+demo.launch()