Update app.py

app.py

@@ -7,18 +7,6 @@ program_explanations = {
     "aca": "ACA subsidies are financial help from the Affordable Care Act that reduce your monthly insurance premium based on your income.",
 }
 
-# Load clinic info (organized by county)
-with open("info.txt", "r", encoding="utf-8") as file:
-    clinic_data = file.read().lower()
-
-# Keep track of conversation state
-conversation_state = {
-    "stage": "intro",
-    "age": None,
-    "income": None,
-    "last_eligible_program": None,  # <--- NEW FIELD
-}
-
 # Helper to get clinics by county
 def find_clinic_by_county(county_name, file_path="info.txt"):
     clinics = []
@@ -36,102 +24,117 @@ def find_clinic_by_county(county_name, file_path="info.txt"):
                 clinics.append(line.strip())
     if not clinics:
         return "Sorry, I couldn't find any clinics for that county. Please check your spelling or try a nearby county."
-    
-    # Add Medicaid/EBT explanation
     clinics.append(
         "---\n\nMedicaid (Apple Health) is Washington’s free health insurance for low-income residents. "
         "It covers doctor visits, mental health, prescriptions, dental, and more.\n\n"
         "EBT (SNAP) helps you buy groceries if you qualify. You’ll get a card you can use at most stores."
     )
-    
-    # Return with double spacing between clinics
     return "\n\n".join(clinics)
 
-# Chatbot logic
-def chatbot(message, history):
+# Chatbot logic with persistent state passed in and returned
+def chatbot(message, history, state):
     message = message.strip().lower()
-    
+
+    # Unpack state dict or initialize if empty
+    if state is None or not state:
+        state = {
+            "stage": "intro",
+            "age": None,
+            "income": None,
+            "last_eligible_program": None,
+        }
+
+    # Explanation trigger phrases
+    explain_triggers = [
+        "what does that mean", "what's that", "explain", "what is that",
+        "can you explain", "tell me about that", "what do you mean"
+    ]
+
     # Check for program keywords in message
     for key in program_explanations:
         if key in message:
-            return program_explanations[key] + "\n\nAnything else you'd like help with?"
-    
-    # EXPLANATION TRIGGER FIX
-    if any(phrase in message for phrase in ["what does that mean", "what's that", "explain", "what is that"]):
-        program = conversation_state.get("last_eligible_program")
+            return program_explanations[key] + "\n\nAnything else you'd like help with?", history, state
+
+    # Check if user asks for explanation of last eligible program
+    if any(phrase in message for phrase in explain_triggers):
+        program = state.get("last_eligible_program")
         if program and program in program_explanations:
-            return program_explanations[program] + "\n\nAnything else you'd like help with?"
+            return program_explanations[program] + "\n\nAnything else you'd like help with?", history, state
 
     response = ""
-    
-    if conversation_state["stage"] == "intro":
-        response = ("Hi! I can help you find free clinics in your Washington county, "
+
+    if state["stage"] == "intro":
+        response = (
+            "Hi! I can help you find free clinics in your Washington county, "
             "or figure out what health programs you may qualify for. "
-            "What would you like help with? Just say 'clinic' or 'eligibility'.")
-        conversation_state["stage"] = "wait_for_choice"
+            "What would you like help with? Just say 'clinic' or 'eligibility'."
+        )
+        state["stage"] = "wait_for_choice"
 
-    elif conversation_state["stage"] == "wait_for_choice":
+    elif state["stage"] == "wait_for_choice":
         if "eligib" in message or "qualify" in message or "insurance" in message:
             response = "Sure, I can help with that. How old are you and what’s your yearly income?"
-            conversation_state["stage"] = "wait_for_eligibility"
+            state["stage"] = "wait_for_eligibility"
         elif "clinic" in message or "help" in message or "doctor" in message:
             response = "What county in Washington are you in?"
-            conversation_state["stage"] = "wait_for_county"
+            state["stage"] = "wait_for_county"
         else:
             response = "Would you like help finding a clinic or seeing what programs you're eligible for? Just say 'clinic' or 'eligibility'."
 
-    elif conversation_state["stage"] == "wait_for_eligibility":
+    elif state["stage"] == "wait_for_eligibility":
         numbers = [int(w) for w in message.split() if w.isdigit()]
         if len(numbers) >= 2:
             age, income = numbers[0], numbers[1]
-            conversation_state["age"] = age
-            conversation_state["income"] = income
+            state["age"] = age
+            state["income"] = income
         elif len(numbers) == 1:
             num = numbers[0]
             if num < 120:
-                conversation_state["age"] = num
-                return "Got it — now how much do you make each year?"
+                state["age"] = num
+                return "Got it — now how much do you make each year?", history, state
             else:
-                conversation_state["income"] = num
-                return "Thanks! Now please tell me your age."
+                state["income"] = num
+                return "Thanks! Now please tell me your age.", history, state
         else:
-            return "Could you tell me your age and income? For example: 'I'm 30 and make $25,000'."
+            return "Could you tell me your age and income? For example: 'I'm 30 and make $25,000'.", history, state
 
-        age = conversation_state["age"]
-        income = conversation_state["income"]
+        age = state["age"]
+        income = state["income"]
 
         # Decide eligibility
         if income < 20000:
             eligible = "free Medicaid coverage"
-            conversation_state["last_eligible_program"] = "medicaid"
+            state["last_eligible_program"] = "medicaid"
         elif income < 40000:
             eligible = "ACA subsidies that reduce the cost of insurance"
-            conversation_state["last_eligible_program"] = "aca"
+            state["last_eligible_program"] = "aca"
         else:
             eligible = "some limited public programs, depending on your needs"
-            conversation_state["last_eligible_program"] = "medicare"
+            state["last_eligible_program"] = "medicare"
 
         response = f"Based on what you shared, you may qualify for {eligible}. Is there anything else I can help you with?"
-        conversation_state["stage"] = "wait_for_choice"
+        state["stage"] = "wait_for_choice"
 
-    elif conversation_state["stage"] == "wait_for_county":
+    elif state["stage"] == "wait_for_county":
         county = message.strip()
         clinic_info = find_clinic_by_county(county)
         response = clinic_info + "\n\nNeed help with anything else?"
-        conversation_state["stage"] = "wait_for_choice"
+        state["stage"] = "wait_for_choice"
 
     else:
         response = "I'm not sure what you meant. Would you like help with free clinics or program eligibility?"
-    
-    return response
 
-# Gradio UI
+    return response, history, state
+
+
+# Gradio UI with state persistence
 with gr.Blocks(css="body { background-color: #EDE8E8; }") as demo:
     with gr.Column():
         gr.Image("heading.png", show_label=False, show_download_button=False)
-        gr.ChatInterface(
+        chat = gr.ChatInterface(
             fn=chatbot,
             title="HealthPal.io",
             description="Use this chatbot to help you find affordable healthcare and figure out what medical subsidies you are eligible for!",
+            state={"stage": "intro", "age": None, "income": None, "last_eligible_program": None},
         )
 demo.launch()