Update app.py

app.py

@@ -20,14 +20,14 @@ st.set_page_config(
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 client = OpenAI(api_key=OPENAI_API_KEY) if OPENAI_API_KEY else None
 
-VISION_MODEL = "gpt-4.1-mini"     # vision-capable OpenAI model
+VISION_MODEL = "gpt-4.1-mini"     # vision-capable model
 REASONING_MODEL = "gpt-4.1-mini"  # reasoning / agent model
 ASR_MODEL = "whisper-1"           # transcription model
 
 # Health Canada Drug Product Database API
 DPD_BASE_URL = "https://health-products.canada.ca/api/drug/drugproduct"
 
-# Recalls & Safety Alerts JSON feed (example; can override via env if GC changes path)
+# Recalls & Safety Alerts JSON feed (can override with env variable if path changes)
 RECALLS_FEED_URL = os.getenv(
     "RECALLS_FEED_URL",
     "https://recalls-rappels.canada.ca/static-data/items-en.json"
@@ -36,7 +36,7 @@ RECALLS_FEED_URL = os.getenv(
 # CIHI wait-times info (general reference)
 WAIT_TIMES_INFO_URL = "https://www.cihi.ca/en/topics/access-and-wait-times"
 
-# Postal code to province mapping (first letter)
+# Postal code to province mapping (by first letter)
 POSTAL_PREFIX_TO_PROVINCE = {
     "A": "Newfoundland and Labrador",
     "B": "Nova Scotia",
@@ -73,7 +73,7 @@ def safe_get_json(url, params=None, timeout=6):
     return None
 
 
-def shorten(text, max_chars=400):
+def shorten(text, max_chars=280):
     if not text:
         return ""
     text = text.strip()
@@ -107,12 +107,12 @@ def tool_region_context(postal_code: str) -> str:
         )
 
     telehealth_note = (
-        f"In {province}, suggest the official provincial Telehealth / 811-style nurse advice line "
-        "and Ministry of Health website for region-specific guidance."
+        f"In {province}, advise the user to use the official provincial Telehealth or 811-style nurse advice line "
+        "for real-time clinical advice."
     )
     wait_note = (
-        "For non-urgent issues, users can check provincial or regional health authority pages and CIHI indicators "
-        f"for access and wait-time information. Reference (general): {WAIT_TIMES_INFO_URL}"
+        "For non-urgent issues, they can check provincial or regional health authority websites and CIHI indicators "
+        f"for access/wait-time information. Reference (general): {WAIT_TIMES_INFO_URL}"
     )
 
     return (
@@ -122,6 +122,74 @@ def tool_region_context(postal_code: str) -> str:
     )
 
 
+def tool_find_nearby_clinics_and_waits(postal_code: str) -> str:
+    """
+    Province-aware pointers to official 'find a clinic' / wait-time resources.
+    Not real-time; we send the user to authoritative tools.
+    """
+    province = infer_province_from_postal(postal_code)
+    if not postal_code or not province:
+        return (
+            "No precise local clinic data. Suggest the user search 'walk-in clinic near me' plus their city, and use "
+            "their provincial health website or health authority finder for verified locations and hours."
+        )
+
+    base = f"For users in {province}, suggest these options to find nearby care and typical waits:\n"
+    province_links = []
+
+    # Very high-level, non-exhaustive hints (you can refine for your production app)
+    if province == "Ontario":
+        province_links.append(
+            "- Use Ontario's official 'Health811' and 'Find a clinic/hospital' tools from Ontario Health."
+        )
+    elif province == "British Columbia":
+        province_links.append(
+            "- Use HealthLink BC (811) and the 'Find Health Services' directory for clinics and urgent care."
+        )
+    elif province == "Alberta":
+        province_links.append(
+            "- Use Alberta Health Services 'Find Healthcare' directory for walk-ins and urgent care clinics."
+        )
+    elif province == "Quebec":
+        province_links.append(
+            "- Use your local CISSS/CIUSSS websites and provincial resources (e.g., Bonjour-santé) for clinic bookings."
+        )
+    elif province == "Manitoba":
+        province_links.append(
+            "- Use Manitoba Health / regional health authority sites to locate walk-in and primary care clinics."
+        )
+    elif province == "Saskatchewan":
+        province_links.append(
+            "- Use Saskatchewan Health Authority tools to find urgent care and clinics."
+        )
+    elif province == "Nova Scotia":
+        province_links.append(
+            "- Use Nova Scotia Health 'Find a Location/Service' for primary care and walk-in options."
+        )
+    elif province == "New Brunswick":
+        province_links.append(
+            "- Use Horizon / Vitalité Health Network resources for clinics and after-hours care."
+        )
+    elif province == "Newfoundland and Labrador":
+        province_links.append(
+            "- Use provincial health authority websites to locate clinics and ERs."
+        )
+    elif province == "Prince Edward Island":
+        province_links.append(
+            "- Use Health PEI resources to find walk-in clinics and primary care."
+        )
+    else:
+        province_links.append(
+            "- Use your territorial/provincial health authority website and Telehealth for locations and typical waits."
+        )
+
+    province_links.append(
+        "- For emergencies or red-flag symptoms, emphasize going to the nearest emergency department or calling 911 immediately."
+    )
+
+    return base + "\n".join(province_links)
+
+
 # =========================
 # DRUG PRODUCT LOOKUP (DPD)
 # =========================
@@ -156,7 +224,7 @@ def tool_lookup_drug_products(candidates):
         )
 
     found_lines.append(
-        "For definitive product information, users must consult the official Health Canada Drug Product Database."
+        "For definitive product information, they must consult the official Health Canada Drug Product Database."
     )
     return "\n".join(found_lines)
 
@@ -172,7 +240,6 @@ def tool_get_recent_recalls_snippet():
     if isinstance(data, list):
         items = data[:5]
     elif isinstance(data, dict):
-        # some feeds use { items: [...] }
         items = (data.get("items") or data.get("results") or [])[:5]
 
     if not items:
@@ -181,7 +248,7 @@ def tool_get_recent_recalls_snippet():
             "Recalls and Safety Alerts website for up-to-date information."
         )
 
-    lines = ["Recent recalls & safety alerts (high-level snapshot):"]
+    lines = ["Recent recalls & safety alerts (snapshot):"]
     for item in items:
         title = (
             item.get("title")
@@ -191,9 +258,10 @@ def tool_get_recent_recalls_snippet():
         )
         date = item.get("date_published") or item.get("date") or ""
         category = item.get("category") or item.get("type") or ""
-        lines.append(f"- {shorten(title, 90)} ({category}, {date})")
+        lines.append(f"- {shorten(title)} ({category}, {date})")
+
     lines.append(
-        "For full details or to check a specific product, direct users to the official Recalls and Safety Alerts portal."
+        "For details or to check a specific product, direct users to the official Recalls and Safety Alerts portal."
     )
     return "\n".join(lines)
 
@@ -205,11 +273,11 @@ def tool_get_recent_recalls_snippet():
 def tool_get_wait_times_awareness():
     return textwrap.dedent(f"""
     Use Canadian open-data wait-time indicators conceptually:
-    - CIHI and several provinces publish average wait-time dashboards.
-    - These are not real-time triage.
+    - CIHI and several provinces publish average wait-time dashboards for ER, surgeries, diagnostics.
+    - These are not real-time triage tools.
     Guidance:
-    - For mild/non-urgent issues: consider local clinics or urgent care; check provincial resources.
-    - For red-flag or severe symptoms: skip tools and go directly to ER or call 911.
+    - For mild issues: local clinics/urgent care may be suitable; check provincial wait-time/clinic-finder tools.
+    - For red-flag or severe symptoms: skip online tools and go directly to ER or call 911.
     General reference: {WAIT_TIMES_INFO_URL}
     """).strip()
 
@@ -241,12 +309,12 @@ def call_vision_summarizer(image_bytes: bytes) -> str:
 
     prompt = (
         "You are a cautious Canadian health-information assistant.\n"
-        "Describe only what is visibly observable in neutral terms.\n"
-        "- If skin: describe general appearance (location, size, colour, pattern). Do NOT name diseases.\n"
-        "- If pill/box: mention visible drug name/strength/markings if legible.\n"
-        "- If document: say what type (e.g., lab report, appointment letter) without copying identifiers.\n"
-        "- If irrelevant/unclear: say that it is not medically interpretable.\n"
-        "Never diagnose. Never prescribe. Limit to about 80-100 words."
+        "Describe only visible features in neutral terms.\n"
+        "- If skin/toenail etc.: describe colour, texture, dryness, cracks, swelling, but DO NOT name diseases.\n"
+        "- If medication label: read visible drug name/strength if legible.\n"
+        "- If health letter/report: say what type it appears to be, without copying identifiers.\n"
+        "- If unclear: say it is not medically interpretable.\n"
+        "Never diagnose. Never prescribe. 80-100 words max."
     )
 
     try:
@@ -302,11 +370,12 @@ def call_reasoning_agent(
     recalls_context: str,
     wait_times_context: str,
     region_context: str,
+    nearby_services_context: str,
 ) -> str:
     if not client:
         return (
             "AI backend not configured. With a valid API key, this agent would combine your inputs and "
-            "open-data context into cautious, non-diagnostic guidance."
+            "open-data context into cautious, structured guidance."
         )
 
     narrative_parts = []
@@ -321,23 +390,43 @@ def call_reasoning_agent(
     system_prompt = """
 You are CareCall AI, an agentic Canadian health information assistant.
 
-Rules:
-- Do NOT diagnose.
-- Do NOT prescribe or give dosing.
-- Do NOT claim exact real-time wait times.
-- Use only the information and tool outputs provided.
-- Use open-data-style context (Drug Product DB, recalls, wait-time awareness, region) to improve safety.
-- Map concerns to broad categories only (e.g., irritation vs infection vs allergy vs medication vs admin).
-- Give practical next steps in Canada:
-    * self-care when clearly mild,
-    * pharmacist / family doctor / walk-in clinic,
-    * provincial Telehealth / nurse line,
-    * ER / 911 for red-flag symptoms.
-- Red flags (chest pain, breathing trouble, stroke signs, severe bleeding, suicidal thoughts,
-  major trauma, high fever with stiff neck, rapidly worsening symptoms, etc.) -> clearly say ER/911.
-- Mention official sources: Health Canada, Public Health Agency of Canada, CIHI,
-  and the user's provincial health services.
-- Maximum length: 350 words. Clear and calm.
+HARD RULES:
+- DO NOT give a medical diagnosis.
+- DO NOT prescribe or specify individual dosing.
+- DO NOT guarantee exact real-time wait times or locations.
+- Use ONLY the provided content and tool outputs (vision summary, DPD, recalls, wait-times context, region info, nearby services guide).
+- You MUST classify guidance into a clear, practical structure without pretending to be a doctor.
+
+WHAT YOU SHOULD DO:
+
+1. Identify red-flag indicators (e.g., chest pain, trouble breathing, stroke signs, rapidly spreading infection,
+   severe pain, heavy bleeding, high fever with stiff neck, suicidal thoughts, major trauma).
+   - If any such red flags are present or strongly implied: 
+     PRIMARY PATHWAY = ER/911.
+2. If no red flags and issue seems mild/local (e.g., minor rash, mild toe pain, dry skin, chipped nail, etc.):
+   - You may choose PRIMARY PATHWAY = Home care + possibly pharmacist.
+3. If there is uncertainty, ongoing pain, risk factors (diabetes, immune issues, poor circulation),
+   or symptoms that are not clearly mild:
+   - You may choose PRIMARY PATHWAY = Clinic / family doctor / Telehealth.
+
+ALWAYS:
+
+- Output a section called "Primary Recommended Pathway" with ONE of:
+    * "Home care likely reasonable (monitor closely)"
+    * "Pharmacist / Walk-in clinic / Family doctor recommended"
+    * "Call provincial Telehealth / nurse line for guidance"
+    * "Go to Emergency / Call 911 now"
+- Then:
+    * Give a short explanation for WHY that pathway was chosen.
+    * Provide a "Home Care (if appropriate)" section with 3–6 simple, generic measures
+      (e.g., keep area clean/dry, rest, elevation, cold compress, OTC pain relief as per label, avoid friction),
+      ONLY when not clearly contradicted.
+    * Provide "When to seek clinic/Telehealth" with concrete triggers.
+    * Provide "When this is an emergency" with explicit red-flag bullets.
+    * Mention provincial/nearby resources using the region and nearby-services context.
+    * If medications/recalls context is relevant, remind the user to verify via Health Canada DPD and Recalls portal.
+- Keep total answer under ~350 words.
+- Reiterate that this is informational only, not medical care, and that professionals/Telehealth/ER override all advice.
 """
 
     user_prompt = f"""
@@ -358,22 +447,25 @@ INPUT CONTEXT
 5) Wait-time awareness context:
 {wait_times_context}
 
-6) Region (postal-code-based) context:
+6) Region (postal code) context:
 {region_context}
 
+7) Nearby services & finder resources:
+{nearby_services_context}
+
 TASK
 
-Based only on this information:
-
-1. Briefly summarize what the user seems concerned about (neutral; no diagnosis).
-2. Briefly outline broad possible categories (without asserting any specific disease).
-3. Provide clear next steps in bullet form, aligned with Canadian context and region info where available:
-   - when self-care might be reasonable,
-   - when to speak with a pharmacist, family doctor, or walk-in clinic,
-   - when to use provincial Telehealth / nurse lines,
-   - when to go to ER or call 911.
-4. If medication/recall context looks relevant, remind them to confirm via official Health Canada resources.
-5. End with a short disclaimer that this is informational only and not medical care.
+Using ONLY the above, produce a structured response with these sections:
+
+1. "Primary Recommended Pathway": one of the four options (see system rules) + 1–2 sentence reasoning.
+2. "What this might relate to (in general terms)": 3–5 bullets of broad categories, no diagnoses.
+3. "Home Care (if appropriate)": 3–6 safe, generic self-care steps OR a note that home care alone is not advised.
+4. "When to see a Pharmacist / Clinic / Telehealth": 3–6 concrete triggers.
+5. "When this is an Emergency (ER/911)": bullet list of red-flag situations.
+6. "Local & Official Resources": 2–5 bullets using the region + nearby-services context and open-data style info.
+7. "Important Disclaimer": 2–4 lines stating this is informational only and not medical care.
+
+Be firm and clear but never overconfident.
 """
 
     try:
@@ -389,7 +481,7 @@ Based only on this information:
     except Exception as e:
         return (
             f"(Reasoning agent unavailable: {e})\n\n"
-            "Please contact a licensed provider, Telehealth service, or emergency care as appropriate."
+            "Please contact a licensed provider, Telehealth, or emergency services as appropriate."
         )
 
 
@@ -401,37 +493,35 @@ st.title("CareCall AI (Canada)")
 st.caption("Agentic, vision + voice + postal-aware health info companion. Not a doctor. Not a diagnosis.")
 
 if not OPENAI_API_KEY:
-    st.warning("Set OPENAI_API_KEY in your Hugging Face Space secrets to enable the AI backend.")
+    st.warning("Set OPENAI_API_KEY in your Space secrets to enable the AI backend.")
 
 st.markdown(
     """
-**How it works**
-
-1. Optionally enter your **Canadian postal code** (to tailor guidance to your province).
-2. Optionally upload a **photo** (rash, pill bottle, letter, etc.).
-3. Optionally upload a **voice note**, and/or type a **short description**.
-4. The agent:
-   - analyzes the image with a vision model (safely),
-   - transcribes your voice with Whisper,
-   - looks up related info via:
-        - Health Canada Drug Product Database (high-level),
-        - Recalls & Safety Alerts feed (snapshot),
-        - CIHI/provincial wait-time context (awareness),
-   - then generates structured, non-diagnostic next steps.
+**How to use**
+
+1. Enter your **Canadian postal code** (optional but helps tailor guidance).
+2. Upload a **photo** related to your concern (e.g., rash, nail, pill bottle, letter) — optional.
+3. Upload a **voice note** and/or type a **short description**.
+4. Click **Run CareCall Agent** to get:
+   - A clear **primary recommended pathway**,
+   - Safe **home-care tips** when appropriate,
+   - Triggers for **clinic/Telehealth**,
+   - **Emergency red-flag guidance**,
+   - Links to **official & provincial resources**.
 """
 )
 
 # 1) Postal code
 st.markdown("### 1. Postal code (optional, Canada only)")
 postal_code = st.text_input(
-    "Canadian postal code (e.g., M5V 2T6). Leave blank to keep guidance general.",
+    "Canadian postal code (e.g., M5V 2T6). Leave blank for general guidance.",
     max_chars=7,
 )
 
 # 2) Image upload
 st.markdown("### 2. Upload an image (optional)")
 image_file = st.file_uploader(
-    "Upload a photo (JPG/PNG). Avoid very identifying or explicit content.",
+    "Upload a photo (JPG/PNG). Avoid highly identifying or explicit content.",
     type=["jpg", "jpeg", "png"],
     accept_multiple_files=False,
 )
@@ -439,7 +529,7 @@ image_file = st.file_uploader(
 image_bytes = None
 if image_file is not None:
     try:
-        # Convert to RGB to avoid RGBA -> JPEG error
+        # Convert to RGB so we can safely save as JPEG (avoids RGBA error)
         img = Image.open(image_file).convert("RGB")
         st.image(img, caption="Image received", use_container_width=True)
         buf = io.BytesIO()
@@ -461,7 +551,7 @@ audio_file = st.file_uploader(
 st.markdown("### 4. Or type a short description")
 user_text = st.text_area(
     "Describe what’s going on:",
-    placeholder='Example: "Red itchy patch on forearm for 3 days, mild burning, no fever, using a new soap."',
+    placeholder='Example: "Pain and discoloration around big toenail when wearing shoes for 1 week, no fever, no major swelling."',
     height=120,
 )
 
@@ -482,7 +572,7 @@ if st.button("Run CareCall Agent"):
             if audio_file is not None:
                 voice_text = call_asr(audio_file)
 
-            # Candidate terms (for Drug DB)
+            # Candidate terms for Drug DB
             combined_for_terms = " ".join(
                 t for t in [user_text or "", voice_text or "", vision_summary or ""] if t
             )
@@ -492,14 +582,20 @@ if st.button("Run CareCall Agent"):
             # Recalls
             recalls_context = tool_get_recent_recalls_snippet()
 
-            # Wait-times awareness
+            # Wait-time awareness
             wait_times_context = tool_get_wait_times_awareness()
 
             # Region context
             region_context = (
                 tool_region_context(postal_code)
                 if postal_code
-                else "No postal code provided. Use Canada-wide guidance and suggest provincial resources."
+                else "No postal code provided. Use Canada-wide guidance and mention provincial resources generally."
+            )
+
+            # Nearby clinics / wait finder resources
+            nearby_services_context = tool_find_nearby_clinics_and_waits(postal_code) if postal_code else (
+                "No postal code. Suggest searchable options like 'walk-in clinic near me' plus city, "
+                "and provincial clinic/ER finder tools."
             )
 
             # Final reasoning
@@ -511,24 +607,8 @@ if st.button("Run CareCall Agent"):
                 recalls_context=recalls_context,
                 wait_times_context=wait_times_context,
                 region_context=region_context,
+                nearby_services_context=nearby_services_context,
             )
 
         st.markdown("### CareCall AI Summary (Informational Only)")
         st.write(final_answer)
-
-        st.markdown(
-            """
----
-**Critical Safety Notice**
-
-- This tool does **not** provide medical diagnoses, prescriptions, or emergency triage.
-- It may be incomplete or incorrect and must **not** replace a doctor, nurse, pharmacist,
-  Telehealth service, or emergency department.
-- If you have severe or rapidly worsening symptoms (chest pain, trouble breathing, signs of stroke,
-  heavy bleeding, suicidal thoughts, major injury, high fever with stiff neck, etc.),
-  call **911** or go to the **nearest emergency department** immediately.
-- For trusted information, consult:
-  Health Canada, the Public Health Agency of Canada, CIHI,
-  and your provincial/territorial health services.
-"""
-        )