Update app.py

app.py

@@ -258,7 +258,7 @@ def beam_generate_v2(src_tensor, beam=5, max_len=50, alpha=0.7):
 
 
 # ------------- wrapper to go from user text → reply -------------
-def generate_reply(user_text: str) -> str:
+def generate_reply_no_attn(user_text: str) -> str:
     # replicate notebook logic: reverse the input sentence
     user_text_rev = reverse(user_text)
     tokens = tokenize(user_text_rev)
@@ -269,32 +269,37 @@ def generate_reply(user_text: str) -> str:
         return "I'm a chatbot trained on Ubuntu Linux support conversations, so I may not understand this question."
     return reply
 
+def generate_reply_attn(user_text: str) -> str:
+    """
+    Inference using the ATTENTION model.
+    Replace body with your encoder/decoder calls (beam or greedy).
+    """
+    # Example placeholder:
+    # src = text_to_tensor(user_text)
+    # return beam_generate_attn(src)
+    return "Reply from ATTENTION model (replace with real function)"
 
 # ------------- Gradio ChatInterface -------------
-def respond(message, history):
-    reply = generate_reply(message)
-    return reply
-
-
-# Create two UI windows side by side (both use the same model for now)
-chat1 = gr.ChatInterface(
-    fn=respond,
-    title="Ubuntu Chatbot without attention",
-    description="A generative chatbot trained on Ubuntu dialogue pairs (seq2seq with attention)."
-)
-
-chat2 = gr.ChatInterface(
-    fn=respond,
-    title="Ubuntu Chatbot with attention",
-    description="A generative chatbot trained on Ubuntu dialogue pairs (seq2seq with attention)."
-)
-
 with gr.Blocks() as demo:
-    gr.Markdown("# Ubuntu Chatbot (Seq2Seq + GRU) Developed by Group E")
-    gr.Row([
-        chat1,
-        chat2
-    ])
+    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)
 
 if __name__ == "__main__":
     demo.launch()