Update app.py

app.py

@@ -1,19 +1,30 @@
 import gradio as gr
-from generate.py import generate_response
+from generate import generate_response
 
-def chat_with_evo(user_input):
-    response = generate_response(user_input)
-    return response
+description = """
+### 🤖 EvoDecoder Chatbot  
+Powered by a lightweight, GPT-style evolved Transformer (EvoDecoder) trained from scratch.
+
+- Trained on dialogue data
+- Uses 6 decoder blocks (512-dim, 8 heads)
+- Fine-tuned to respond to real prompts
+
+"""
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 EvoDecoder Chatbot (Lightweight Conversational AI)")
+    gr.Markdown(description)
+    chatbot = gr.Chatbot(height=400)
     with gr.Row():
-        with gr.Column(scale=4):
-            user_input = gr.Textbox(label="Ask Evo anything...", placeholder="Type your question here...")
-            submit_btn = gr.Button("Submit")
-        with gr.Column(scale=6):
-            response_output = gr.Textbox(label="Evo's Response")
+        user_input = gr.Textbox(placeholder="Type a message...", scale=4)
+        send_btn = gr.Button("Send", scale=1)
+
+    history = []
+
+    def respond(message):
+        response = generate_response(message)
+        history.append((message, response))
+        return history, ""
 
-    submit_btn.click(fn=chat_with_evo, inputs=user_input, outputs=response_output)
+    send_btn.click(fn=respond, inputs=user_input, outputs=[chatbot, user_input])
 
 demo.launch()