Update thicc/app.py

thicc/app.py

@@ -4,22 +4,88 @@ from data.data_loader import list_services, load_services_data, HOSPITALS
 from insurance import plans, cost_estimator
 from insurance.coverage_explainer import CoverageExplainer
 
-# UI dropdowns
-plan_dropdown = gr.Dropdown(
-    choices=list(plans.SAMPLE_PLANS.keys()) + ["No Insurance"],
-    value="PPO",
-    label="Select plan",
-)
+# Valid options for hospitals and plans (used in the conversational flow)
+HOSPITAL_CHOICES = list(HOSPITALS.keys())
+PLAN_CHOICES = list(plans.SAMPLE_PLANS.keys()) + ["No Insurance"]
 
-hospital_dropdown = gr.Dropdown(
-    choices=list(HOSPITALS.keys()),
-    value=list(HOSPITALS.keys())[0],
-    label="Select a Hospital",
-)
 
+def _extract_hospital_and_plan(text: str):
+    """Best-effort extraction of hospital and plan names from free text."""
+    if not text:
+        return None, None
+
+    lower = text.lower()
+    hospital_name = None
+    plan_name = None
+
+    # Match hospitals by substring
+    for name in HOSPITAL_CHOICES:
+        if name.lower() in lower:
+            hospital_name = name
+
+    # Match plans by key
+    for plan_key in plans.SAMPLE_PLANS.keys():
+        if plan_key.lower() in lower:
+            plan_name = plan_key
+
+    # Allow "no insurance" as a phrase
+    if "no insurance" in lower:
+        plan_name = "No Insurance"
+
+    return hospital_name, plan_name
+
+
+def _get_current_hospital_and_plan(message, history):
+    """
+    Look through the chat history (and latest message) for the most recent
+    hospital & plan mentioned by the user.
+    """
+    hospital_name = None
+    plan_name = None
+
+    # History is a list of [user, bot] message pairs
+    for user_msg, bot_msg in history or []:
+        h, p = _extract_hospital_and_plan(user_msg)
+        if h:
+            hospital_name = h
+        if p:
+            plan_name = p
+
+    # Also extract from the current message
+    h, p = _extract_hospital_and_plan(message)
+    if h:
+        hospital_name = h
+    if p:
+        plan_name = p
 
-def respond(message, history, plan_name: str, hospital_name: str) -> str:
-    """Main response function - handles cost estimation and coverage explanations."""
+    return hospital_name, plan_name
+
+
+def respond(message, history) -> str:
+    """Main response function - now handles hospital/plan conversationally."""
+
+    # Determine the user's current hospital & plan from history/message
+    hospital_name, plan_name = _get_current_hospital_and_plan(message, history)
+
+    # If either is missing, show the options and ask the user to choose
+    if hospital_name is None or plan_name is None:
+        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. 🏥
+
+Before I can estimate your costs, tell me **which hospital** you're using and **what insurance plan** you have.
+
+**Hospitals I currently support:**
+{hospitals_list}
+
+**Insurance options I support:**
+{plans_list}
+
+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."
+"""
 
     # --- 1) Coverage questions first ---
     if CoverageExplainer.identify_coverage_question(message):
@@ -80,7 +146,7 @@ def respond(message, history, plan_name: str, hospital_name: str) -> str:
     service_description = service_data.iloc[0]["description"]
     price = service_data.iloc[0]["negotiated_rate"]
 
-    # Map dropdown choice -> InsurancePlan object
+    # Map plan name -> InsurancePlan object
     if plan_name == "No Insurance":
         plan = plans.NO_INSURANCE_PLAN
     else:
@@ -105,7 +171,7 @@ Because you selected **No Insurance**, this demo assumes you pay the full negoti
 • Your insurance covers: $0.00
 • You pay: ${cost:.2f}
 
-If you want to see how deductibles, copays, and coinsurance work, switch to a sample plan above and ask something like:
+If you want to see how deductibles, copays, and coinsurance work, try asking:
 • "What is a deductible?"
 • "Explain coinsurance"
 """
@@ -155,9 +221,8 @@ demo = gr.ChatInterface(
 • Coverage terms: "What is a deductible?" or "Explain coinsurance"
 • Available services: "What services are available?"
 """,
-    additional_inputs=[plan_dropdown, hospital_dropdown],
-    additional_inputs_accordion="Tell us about your insurance plan and hospital",
 )
 
 if __name__ == "__main__":
-    demo.launch(ssr_mode=False)
+    # Keep SSR disabled for Hugging Face Spaces stability
+    demo.launch(ssr_mode=False)