Update app.py

app.py

@@ -9,7 +9,7 @@ CONTACTS = """
 🌐 Website: [elimuhub.simdif.com](https://elimuhub.simdif.com)  
 """
 
-# --- Tuition & Homeschooling Packages (as DataFrame for table display) ---
+# --- Tuition & Homeschooling Packages (DataFrame for display) ---
 fees_data = [
     ["Homeschooling - Standard (3 days/wk, 2 hrs/day, 24 hrs)", "KES 12,000"],
     ["Homeschooling - Intensive (5 days/wk, 4 hrs/day, 80 hrs)", "KES 20,000"],
@@ -57,40 +57,34 @@ FAQS = {
 def chatbot(user_input):
     text = user_input.lower()
 
-    # Fees (return table separately)
     if "fee" in text or "price" in text or "package" in text:
-        return ("Here are our tuition & homeschooling packages:", fees_df)
-
-    # FAQs
+        return "Here are our tuition & homeschooling packages 👇 (see the table below).", fees_df
     elif "subject" in text:
-        return (FAQS["subjects"], None)
+        return FAQS["subjects"], None
     elif "curriculum" in text or "syllabus" in text:
-        return (FAQS["curriculum"], None)
+        return FAQS["curriculum"], None
     elif "homeschool" in text:
-        return (FAQS["homeschool"], None)
+        return FAQS["homeschool"], None
     elif "tutor" in text or "teacher" in text:
-        return (FAQS["tutor"], None)
+        return FAQS["tutor"], None
     elif "location" in text or "where" in text:
-        return (FAQS["location"], None)
+        return FAQS["location"], None
     elif "contact" in text or "phone" in text or "email" in text or "website" in text:
-        return (FAQS["contact"], None)
-
-    # Greeting
+        return FAQS["contact"], None
     elif "hello" in text or "hi" in text:
-        return (f"👋 Hello, welcome to {ORG_NAME}!\nAsk about *fees*, *subjects*, *curriculum*, *homeschooling*, or *contacts*.", None)
-
-    # Fallback
+        return f"👋 Hello, welcome to {ORG_NAME}!\nAsk about *fees*, *subjects*, *curriculum*, *homeschooling*, or *contacts*.", None
     else:
-        return ("I can help with tuition, homeschooling, packages, and curricula. Try asking: 'What are your fees?' or use the quick menu below ⬇️", None)
+        return "I can help with tuition, homeschooling, packages, and curricula. Try asking: 'What are your fees?' or use the quick menu below ⬇️", None
 
 # --- Gradio UI ---
 with gr.Blocks() as demo:
     gr.Markdown(f"## 🤝 Welcome to {ORG_NAME}\nAsk us about our tuition & homeschooling services!")
 
-    chatbot_ui = gr.ChatInterface(
-        fn=chatbot,
-        additional_outputs=[gr.DataFrame(label="Fee Packages", visible=True)]
-    )
+    chatbox = gr.Chatbot(label="Chat with Elimuhub")
+    msg = gr.Textbox(placeholder="Type your question here...")
+
+    fees_table = gr.DataFrame(value=pd.DataFrame([], columns=["Package / Service", "Price"]),
+                              label="Fee Packages", visible=False)
 
     with gr.Row():
         gr.Markdown("### 📌 Quick Menu")
@@ -103,14 +97,25 @@ with gr.Blocks() as demo:
         btn_location = gr.Button("📍 Locations")
         btn_contact = gr.Button("☎️ Contact Info")
 
-    # Button actions
-    btn_fees.click(lambda: chatbot("fees"), outputs=[chatbot_ui, chatbot_ui.additional_outputs[0]])
-    btn_subjects.click(lambda: chatbot("subjects"), outputs=[chatbot_ui, chatbot_ui.additional_outputs[0]])
-    btn_curriculum.click(lambda: chatbot("curriculum"), outputs=[chatbot_ui, chatbot_ui.additional_outputs[0]])
-    btn_homeschool.click(lambda: chatbot("homeschool"), outputs=[chatbot_ui, chatbot_ui.additional_outputs[0]])
-    btn_tutors.click(lambda: chatbot("tutor"), outputs=[chatbot_ui, chatbot_ui.additional_outputs[0]])
-    btn_location.click(lambda: chatbot("location"), outputs=[chatbot_ui, chatbot_ui.additional_outputs[0]])
-    btn_contact.click(lambda: chatbot("contact"), outputs=[chatbot_ui, chatbot_ui.additional_outputs[0]])
+    # Handle text input
+    def respond(message, history):
+        response, table = chatbot(message)
+        history.append((message, response))
+        if table is not None:
+            return history, table, gr.update(visible=True)
+        else:
+            return history, pd.DataFrame([], columns=["Package / Service", "Price"]), gr.update(visible=False)
+
+    msg.submit(respond, [msg, chatbox], [chatbox, fees_table, fees_table])
+    
+    # Button actions (simulate user queries)
+    btn_fees.click(lambda: respond("fees", []), outputs=[chatbox, fees_table, fees_table])
+    btn_subjects.click(lambda: respond("subjects", []), outputs=[chatbox, fees_table, fees_table])
+    btn_curriculum.click(lambda: respond("curriculum", []), outputs=[chatbox, fees_table, fees_table])
+    btn_homeschool.click(lambda: respond("homeschool", []), outputs=[chatbox, fees_table, fees_table])
+    btn_tutors.click(lambda: respond("tutor", []), outputs=[chatbox, fees_table, fees_table])
+    btn_location.click(lambda: respond("location", []), outputs=[chatbox, fees_table, fees_table])
+    btn_contact.click(lambda: respond("contact", []), outputs=[chatbox, fees_table, fees_table])
 
 # Launch app
 if __name__ == "__main__":