Update app.py

app.py

@@ -289,21 +289,30 @@ PREDEFINED = [
 ]
 
 # ---------- Reply functions compatible with ChatInterface ----------
-# ChatInterface expects a function: (message, history) -> (reply, history)
 def reply_no_attn(message, history):
     if not message or not str(message).strip():
-        return "Please enter a question."
-    return generate_reply_no_attn(message)
+        bot_reply = "Please enter a question."
+    else:
+        bot_reply = generate_reply_no_attn(message)
+    if history is None:
+        history = []
+    history = history + [(message, bot_reply)]
+    return bot_reply, history
 
 def reply_attn(message, history):
     if not message or not str(message).strip():
-        return "Please enter a question."
-    return generate_reply_attn(message)
+        bot_reply = "Please enter a question."
+    else:
+        bot_reply = generate_reply_attn(message)
+    if history is None:
+        history = []
+    history = history + [(message, bot_reply)]
+    return bot_reply, history
+
 
-# ---------- Build UI: ChatInterface side-by-side inside Blocks ----------
 with gr.Blocks() as demo:
     gr.Markdown("## Ubuntu Chatbot Comparison — No Attention (left) vs Attention (right)")
-    gr.Markdown("Use dropdown to quickly fill the chat input. Each chat keeps its own history. Clear button is built in per chat.")
+    gr.Markdown("Use dropdown to quickly submit predefined prompts. Each chat keeps its own history. Clear button is built in per chat.")
 
     with gr.Row():
         # Left column: ChatInterface + dropdown
@@ -311,18 +320,15 @@ with gr.Blocks() as demo:
             gr.Markdown("### No Attention Model")
             chat_left = gr.ChatInterface(fn=reply_no_attn, title="No Attention", description="Clear button is on the chat box")
             dd_left = gr.Dropdown(choices=PREDEFINED, label="Quick prompts (left)", interactive=True)
-            def set_input_left(selected):
-                return gr.update(value=selected)
-            dd_left.change(fn=set_input_left, inputs=dd_left, outputs=chat_left.input_components[0])
+            # CALL the same reply function and output to the ChatInterface component
+            dd_left.select(fn=reply_no_attn, inputs=[dd_left, chat_left], outputs=[chat_left])
 
         # Right column: ChatInterface + dropdown
         with gr.Column(scale=1):
             gr.Markdown("### With Attention Model")
             chat_right = gr.ChatInterface(fn=reply_attn, title="With Attention", description="Clear button is on the chat box")
             dd_right = gr.Dropdown(choices=PREDEFINED, label="Quick prompts (right)", interactive=True)
-            def set_input_right(selected):
-                return gr.update(value=selected)
-            dd_right.change(fn=set_input_right, inputs=dd_right, outputs=chat_right.input_components[0])
+            dd_right.select(fn=reply_attn, inputs=[dd_right, chat_right], outputs=[chat_right])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()