Update app.py

app.py

@@ -2,29 +2,27 @@ import gradio as gr
 from generate import generate_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
-
+# 🧠 EvoDecoder Chatbot  
+This chatbot is powered by **EvoDecoder**, a lightweight GPT-style Transformer trained on a small dataset.  
+Ask any question and Evo will respond using its own learned weights (no OpenAI involved).
 """
 
+def chat(prompt):
+    try:
+        response = generate_response(prompt)
+        return response if response else "(No response generated)"
+    except Exception as e:
+        return f"⚠️ Error: {str(e)}"
+
 with gr.Blocks() as demo:
     gr.Markdown(description)
-    chatbot = gr.Chatbot(height=400)
     with gr.Row():
-        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, ""
+        with gr.Column():
+            input_box = gr.Textbox(label="Ask Evo", placeholder="Type a question here...")
+            submit_btn = gr.Button("Generate")
+        with gr.Column():
+            output_box = gr.Textbox(label="Evo's Response")
 
-    send_btn.click(fn=respond, inputs=user_input, outputs=[chatbot, user_input])
+    submit_btn.click(fn=chat, inputs=input_box, outputs=output_box)
 
 demo.launch()