Update app.py

app.py

@@ -212,6 +212,26 @@ def rebuild_index():
     return gr.update(value="✅ Index rebuilt from KB.")
 
 # ----------- Gradio UI -----------
+def process_message(user_input, history):
+    """
+    history is a list of dicts: [{"role":"user","content":...}, {"role":"assistant","content":...}, ...]
+    We return updated history and a cleared textbox.
+    """
+    user_input = (user_input or "").strip()
+    if not user_input:
+        return history, gr.update(value="")
+    reply = respond(user_input, history or [])
+    new_history = (history or []) + [
+        {"role": "user", "content": user_input},
+        {"role": "assistant", "content": reply}
+    ]
+    return new_history, gr.update(value="")
+
+def process_quick(label, history):
+    # turn a quick button into a user message
+    q = quick_intent(label)
+    return process_message(q, history)
+
 with gr.Blocks(title="Self-Service KB Assistant", fill_height=True) as demo:
     gr.Markdown(
         """
@@ -219,32 +239,40 @@ with gr.Blocks(title="Self-Service KB Assistant", fill_height=True) as demo:
 Ask about setup, automations, billing, or troubleshooting.  
 The assistant answers **only from the Knowledge Base** and cites the articles it used.
 
-**Quick actions:** Try one of the buttons below.
+**Quick actions:** Use a button below to try a common task.
         """
     )
 
     with gr.Row():
-        outputs = []
-        for label, _ in HELPFUL_SUGGESTIONS:
-            btn = gr.Button(label)
-            btn.click(fn=lambda L=label: quick_intent(L), outputs=None)\
-               .then(fn=respond, inputs=[gr.State(quick_intent(label)), gr.State([])], outputs=gr.Chatbot())
-            outputs.append(btn)
-
-    chat = gr.ChatInterface(
-        fn=respond,
-        chatbot=gr.Chatbot(height=420, show_copy_button=True),
-        textbox=gr.Textbox(placeholder="e.g., How do I connect WhatsApp?"),
-        retry_btn="Retry",
-        undo_btn="Undo",
-        clear_btn="Clear",
-    )
+        chat = gr.Chatbot(height=420, show_copy_button=True, type="messages")  # ✅ modern format
+    with gr.Row():
+        txt = gr.Textbox(placeholder="e.g., How do I connect WhatsApp?", scale=10)
+        send = gr.Button("Send", variant="primary")
+
+    with gr.Row():
+        # Quick action buttons
+        btn_whatsapp = gr.Button("Connect WhatsApp")
+        btn_reset = gr.Button("Reset Password")
+        btn_first = gr.Button("First Automation")
+        btn_billing = gr.Button("Billing & Invoices")
+        btn_ig = gr.Button("Fix Instagram Connect")
 
     with gr.Accordion("Admin", open=False):
         gr.Markdown("Rebuild the search index after changing files in `/kb`.")
         rebuild = gr.Button("Rebuild Index")
         status = gr.Markdown("")
-        rebuild.click(fn=rebuild_index, outputs=status)
+
+    # Wire up events
+    send.click(process_message, inputs=[txt, chat], outputs=[chat, txt])
+    txt.submit(process_message, inputs=[txt, chat], outputs=[chat, txt])  # Enter key
+
+    btn_whatsapp.click(process_quick, inputs=[gr.State("Connect WhatsApp"), chat], outputs=[chat, txt])
+    btn_reset.click(process_quick, inputs=[gr.State("Reset Password"), chat], outputs=[chat, txt])
+    btn_first.click(process_quick, inputs=[gr.State("First Automation"), chat], outputs=[chat, txt])
+    btn_billing.click(process_quick, inputs=[gr.State("Billing & Invoices"), chat], outputs=[chat, txt])
+    btn_ig.click(process_quick, inputs=[gr.State("Fix Instagram Connect"), chat], outputs=[chat, txt])
+
+    rebuild.click(lambda: (kb.build(KB_DIR), "✅ Index rebuilt from KB.")[1], outputs=status)
 
 if __name__ == "__main__":
     demo.launch()