Update app.py

app.py

@@ -1,28 +1,17 @@
+# app.py
+
 import gradio as gr
 from generate import generate_response
 
-description = """
-# 🧠 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)
-    with gr.Row():
-        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")
+def chat_fn(user_input):
+    return generate_response(user_input)
 
-    submit_btn.click(fn=chat, inputs=input_box, outputs=output_box)
+iface = gr.Interface(
+    fn=chat_fn,
+    inputs=gr.Textbox(lines=2, placeholder="Ask Evo anything..."),
+    outputs="text",
+    title="🧠 EvoDecoder Chatbot",
+    description="A lightweight, custom-trained GPT-style decoder built from scratch."
+)
 
-demo.launch()
+iface.launch()