Update app.py

app.py

@@ -281,25 +281,60 @@ def generate_reply_attn(user_text: str) -> str:
 
 # ------------- Gradio ChatInterface -------------
 with gr.Blocks() as demo:
-    gr.Markdown("# Ubuntu Chatbot Comparison — No Attention vs Attention")
+    gr.Markdown("# **Ubuntu Chatbot Comparison — No Attention vs Attention**")
 
     with gr.Row():
-        
-        # Left column — No Attention model
-        with gr.Column():
-            gr.Markdown("### No Attention Model")
-            inp1 = gr.Textbox(label="Your question (No-Attn)", placeholder="Type here...")
-            out1 = gr.Textbox(label="Bot Reply (No-Attn)")
-            btn1 = gr.Button("Ask No-Attn Bot")
-            btn1.click(fn=generate_reply_no_attn, inputs=inp1, outputs=out1)
-
-        # Right column — Attention model
-        with gr.Column():
-            gr.Markdown("### Attention Model")
-            inp2 = gr.Textbox(label="Your question (Attn)", placeholder="Type here...")
-            out2 = gr.Textbox(label="Bot Reply (Attn)")
-            btn2 = gr.Button("Ask Attn Bot")
-            btn2.click(fn=generate_reply_attn, inputs=inp2, outputs=out2)
+
+        # ---------- LEFT: No Attention ----------
+        with gr.Column(scale=1):
+            gr.Markdown("### **No Attention Model**")
+
+            chatbox1 = gr.Textbox(
+                label="Bot Reply (No-Attention)",
+                lines=8,
+                interactive=False
+            )
+
+            user_input1 = gr.Textbox(
+                label="Your Message",
+                placeholder="Type a message for the NO-ATTN model...",
+                lines=2
+            )
+
+            send1 = gr.Button("Send")
+
+            # User clicks send → reply appears in big chat window
+            send1.click(
+                fn=generate_reply_no_attn,
+                inputs=user_input1,
+                outputs=chatbox1
+            )
+
+
+        # ---------- RIGHT: With Attention ----------
+        with gr.Column(scale=1):
+            gr.Markdown("### **Attention Model**")
+
+            chatbox2 = gr.Textbox(
+                label="Bot Reply (Attention)",
+                lines=8,
+                interactive=False
+            )
+
+            user_input2 = gr.Textbox(
+                label="Your Message",
+                placeholder="Type a message for the ATTENTION model...",
+                lines=2
+            )
+
+            send2 = gr.Button("Send")
+
+            send2.click(
+                fn=generate_reply_attn,
+                inputs=user_input2,
+                outputs=chatbox2
+            )
+
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()