Update app.py

app.py

@@ -1,73 +1,151 @@
-from huggingface_hub import InferenceClient
-from sentence_transformers import SentenceTransformer
-import torch
 import gradio as gr
-import random
 
-client = InferenceClient("Qwen/Qwen2.5-72B-Instruct")
+# helper to get clinics by county
+def get_clinics_by_county(county_name, file_path="info.txt", limit=5):
+    clinics = []
+    with open(file_path, "r", encoding="utf-8") as file:
+        lines = file.readlines()
+
+    inside_county = False
+    count = 0
+    for line in lines:
+        if county_name.lower() in line.lower():
+            inside_county = True
+            continue
+        if inside_county:
+            if "county" in line.lower() and county_name.lower() not in line.lower():
+                break
+            if line.strip():
+                clinics.append(line.strip())
+                count += 1
+            if count >= limit * 5:
+                break
+    return "\n".join(clinics) if clinics else "No clinics found for that county."
 
-# Load info.txt at start
+
+# Load clinic info (organized by county)
 with open("info.txt", "r", encoding="utf-8") as file:
-    info_lines = file.readlines()
-
-# list of counties to check against
-possible_counties = [
-    "adams", "asotin", "benton", "chelan", "clallam", "clark", "columbia", "cowlitz",
-    "douglas", "ferry", "franklin", "garfield", "grant", "grays harbor", "island",
-    "jefferson", "king", "kitsap", "kittitas", "klickitat", "lewis", "lincoln", "mason",
-    "okanogan", "pacific", "pend oreille", "pierce", "san juan", "skagit", "skamania",
-    "snohomish", "spokane", "stevens", "thurston", "wahkiakum", "walla walla", "whatcom",
-    "whitman", "yakima"
-]
-
-# pulls clinics from info.txt
-def get_clinics_by_county(county_name, limit=5):
+    clinic_data = file.read().lower()
+
+# Keep track of state
+conversation_state = {
+    "stage": "intro",  # intro → wait_for_choice → wait_for_eligibility / wait_for_county
+    "age": None,
+    "income": None,
+}
+
+program_explanations = {
+    "medicaid": "Medicaid (called Apple Health in Washington) is a free or low-cost health coverage program for eligible low-income adults, children, pregnant women, and people with disabilities.",
+    "medicare": "Medicare is a federal health insurance program mainly for people age 65 or older and certain younger people with disabilities.",
+    "aca": "ACA subsidies are financial help from the Affordable Care Act that reduce your monthly insurance premium based on your income.",
+}
+
+def find_clinic_by_county(county):
+    county = county.lower()
+    sections = clinic_data.split("###")
+    for section in sections:
+        if county in section:
+            lines = section.strip().split("\n")
+            if lines and county in lines[0].lower():
+                lines = lines[1:]
+            return "\n".join(lines).strip()
+    return "Sorry, I couldn't find any clinics for that county. Please check your spelling or try a nearby county."
+    
+# helper to get clinics by county
+def get_clinics_by_county(county_name, file_path="info.txt", limit=5):
     clinics = []
+    with open(file_path, "r", encoding="utf-8") as file:
+        lines = file.readlines()
+
     inside_county = False
     count = 0
-    for line in info_lines:
+    for line in lines:
         if county_name.lower() in line.lower():
             inside_county = True
             continue
         if inside_county:
             if "county" in line.lower() and county_name.lower() not in line.lower():
                 break
-            if line.strip() != "":
+            if line.strip():
                 clinics.append(line.strip())
                 count += 1
             if count >= limit * 5:
                 break
     return "\n".join(clinics) if clinics else "No clinics found for that county."
 
-# main chatbot function for chat interface
+
 def chatbot(message, history):
-    user_input = message.lower()
+    message = message.strip().lower()
+
+    # Check if user asks about programs
+    for key in program_explanations:
+        if key in message:
+            return program_explanations[key] + "\n\nAnything else you'd like help with?"
+
+    response = ""
+
+    if conversation_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."
+        conversation_state["stage"] = "wait_for_choice"
+
+    elif conversation_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"
+        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"
+        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":
+        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
+        elif len(numbers) == 1:
+            num = numbers[0]
+            if num < 120:
+                conversation_state["age"] = num
+                response = "Got it — now how much do you make each year?"
+                return response
+            else:
+                conversation_state["income"] = num
+                response = "Thanks! Now please tell me your age."
+                return response
+        else:
+            return "Could you tell me your age and income? For example: 'I'm 30 and make $25,000'."
 
-    if "what's that" in user_input:
-        return ("Affordable healthcare includes programs like Medicaid (Apple Health in WA), which is free or low-cost health insurance for people with low income. "
-                "It covers doctor visits, prescriptions, mental health, dental, and vision care. You usually apply through the WA Healthplanfinder.")
+        age = conversation_state["age"]
+        income = conversation_state["income"]
+        if income < 20000:
+            eligible = "free Medicaid coverage"
+        elif income < 40000:
+            eligible = "ACA subsidies that reduce the cost of insurance"
+        else:
+            eligible = "some limited public programs, depending on your needs"
 
-    if user_input in possible_counties:
-        clinics = get_clinics_by_county(user_input)
-        return f"Here are some free clinics in {message.title()} County:\n\n{clinics}"
+        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"
 
-    # default: send message to LLM
-    full_prompt = ""
-    for user, bot in history:
-        full_prompt += f"<|user|>\n{user}<|end|>\n<|assistant|>\n{bot}<|end|>\n"
-    full_prompt += f"<|user|>\n{message}<|end|>\n<|assistant|>"
+    elif conversation_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"
 
-    temperature = random.uniform(0.3, 0.9)
+    else:
+        response = "I'm not sure what you meant. Would you like help with free clinics or program eligibility?"
 
-    response = client.text_generation(
-        prompt=full_prompt,
-        max_new_tokens=150,
-        temperature=temperature,
-        top_p=0.9,
-        stop=["<|user|>", "<|assistant|>"]
-    )
+    return response
 
-    return response.strip()
+# Gradio app
+with gr.Blocks() as demo:
+    with gr.Row():
+        gr.Markdown("# HealthPal.io")
+    with gr.Row():
+        gr.Markdown("## Helping Washington State residents find free clinics and health programs.")
+    gr.ChatInterface(chatbot)
 
-# the correct format you're using
-gr.ChatInterface(chatbot, title="Washington Healthcare Bot").launch()
+demo.launch()