Update thicc/app.py

thicc/app.py

@@ -107,7 +107,9 @@ def respond(message, history):
         hospitals_list = "\n".join(f"• {name}" for name in HOSPITAL_CHOICES)
         plans_list = "\n".join(f"• {name}" for name in PLAN_CHOICES)
 
-        return f"""Hi, I'm the THICC Cost Chatbot. 🏥
+        # Case 1: Both missing - show full welcome message
+        if hospital_name is None and plan_name is None:
+            return f"""Hi, I'm the THICC Cost Chatbot. 🏥
 
 Before I can estimate your costs, tell me **which hospital** you're using and **what insurance plan** you have.
 
@@ -120,6 +122,30 @@ Before I can estimate your costs, tell me **which hospital** you're using and **
 Please reply with something like:
 • "I'm going to UCLA Medical Center and I have a PPO plan."
 • "Cedars-Sinai Medical Center with No Insurance."
+"""
+
+        # Case 2: Have hospital, need insurance plan
+        if hospital_name is not None and plan_name is None:
+            return f"""Great, I see you're going to **{hospital_name}**! 🏥
+
+Now, what **insurance plan** do you have?
+
+**Insurance options I support:**
+{plans_list}
+
+Please reply with your plan, like "PPO" or "No Insurance".
+"""
+
+        # Case 3: Have insurance plan, need hospital
+        if hospital_name is None and plan_name is not None:
+            return f"""Got it, you have **{plan_name}**! 📋
+
+Now, which **hospital** are you going to?
+
+**Hospitals I currently support:**
+{hospitals_list}
+
+Please reply with the hospital name, like "UCLA Medical Center".
 """
 
     # --- 1) Coverage questions first ---
@@ -248,7 +274,7 @@ This estimate assumes your deductible has been met.
 # Gradio interface
 demo = gr.ChatInterface(
     fn=respond,
-    title="THICC Cost Chatbot 🏥",
+    title="THICC Cost Chatbot - Now with Coverage Explanations! 🏥",
     description="""Get healthcare cost estimates and understand your insurance coverage.
     
 **What you can ask:**