Update app.py

app.py

@@ -3,6 +3,7 @@ import io
 import base64
 import textwrap
 import csv
+import re
 import requests
 import streamlit as st
 from PIL import Image
@@ -19,9 +20,9 @@ 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"      # image-capable model
-REASONING_MODEL = "gpt-4.1-mini"   # reasoning model
-ASR_MODEL = "whisper-1"            # transcription
+VISION_MODEL = "gpt-4.1-mini"
+REASONING_MODEL = "gpt-4.1-mini"
+ASR_MODEL = "whisper-1"
 
 DPD_BASE_URL = "https://health-products.canada.ca/api/drug/drugproduct"
 RECALLS_FEED_URL = os.getenv(
@@ -30,10 +31,8 @@ RECALLS_FEED_URL = os.getenv(
 )
 WAIT_TIMES_INFO_URL = "https://www.cihi.ca/en/topics/access-and-wait-times"
 
-# Optional: your own wait-time microservice (if you create one)
 WAIT_TIMES_API_BASE = os.getenv("WAIT_TIMES_API_BASE", "").rstrip("/")
 
-# Google Custom Search (JSON API)
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 GOOGLE_CSE_ID = os.getenv("GOOGLE_CSE_ID")
 
@@ -58,16 +57,12 @@ POSTAL_PREFIX_TO_PROVINCE = {
     "Y": "Yukon",
 }
 
-USER_AGENT = "carecall-ai-googlecse-hqo-city/1.0"
+USER_AGENT = "carecall-ai-cityfacilities/1.0"
 
 
-# ============== POSTAL -> CITY MAP USING canadacities.csv ==============
+# ============== POSTAL -> CITY MAP (canadacities.csv) ==============
 
 def load_postal_city_map(path: str = "canadacities.csv"):
-    """
-    Build a map: first 3 chars of postal code (FSA) -> (city, province).
-    Tries to auto-detect column names in the CSV.
-    """
     mapping = {}
     if not os.path.exists(path):
         return mapping
@@ -78,7 +73,6 @@ def load_postal_city_map(path: str = "canadacities.csv"):
             header = next(reader, None)
             if not header:
                 return mapping
-
             lower = [h.strip().lower() for h in header]
 
             def find_col(candidates):
@@ -113,10 +107,8 @@ def load_postal_city_map(path: str = "canadacities.csv"):
                     else ""
                 )
                 if prefix and (city or prov):
-                    # first entry wins; that's fine at FSA granularity
                     mapping.setdefault(prefix, (city, prov))
     except Exception:
-        # On any parsing error, just fall back silently to province-only logic
         return mapping
 
     return mapping
@@ -131,8 +123,7 @@ def get_city_from_postal(postal_code: str):
     pc = postal_code.strip().upper().replace(" ", "")
     if len(pc) < 3:
         return None, None
-    prefix = pc[:3]
-    return POSTAL_CITY_MAP.get(prefix, (None, None))
+    return POSTAL_CITY_MAP.get(pc[:3], (None, None))
 
 
 # ============== GENERIC HELPERS ==============
@@ -203,7 +194,16 @@ def geocode_postal(postal_code: str):
     return None
 
 
-# ============== GOOGLE CUSTOM SEARCH (CITY-BASED) ==============
+def extract_phone(text: str):
+    if not text:
+        return ""
+    # simple North American phone pattern
+    pattern = r"(\(?\d{3}\)?[\s\-\.]?\d{3}[\s\-\.]?\d{4})"
+    m = re.search(pattern, text)
+    return m.group(1) if m else ""
+
+
+# ============== GOOGLE CSE HELPERS ==============
 
 def google_cse_search(query: str, num: int = 5):
     if not GOOGLE_API_KEY or not GOOGLE_CSE_ID:
@@ -218,27 +218,64 @@ def google_cse_search(query: str, num: int = 5):
     return safe_get_json(url, params=params)
 
 
+def tool_city_facilities_google(postal_code: str) -> str:
+    """
+    City-level fallback using Google CSE:
+    - Find city from canadacities.csv.
+    - Search for hospitals/clinics in that city.
+    - Extract name, rough location, phone if visible.
+    """
+    if not postal_code:
+        return ""
+
+    city, prov = get_city_from_postal(postal_code)
+    if city:
+        loc = f"{city}, {prov or 'Canada'}"
+    else:
+        loc = f"{postal_code} Canada"
+
+    data = google_cse_search(
+        query=f"hospital OR walk-in clinic OR urgent care in {loc}",
+        num=5,
+    )
+    if not data or "items" not in data:
+        return ""
+
+    lines = [f"City-based facilities from official/clinical search results around {loc} (verify details):"]
+    for it in data["items"][:5]:
+        title = it.get("title", "Result")
+        link = it.get("link", "")
+        snippet = shorten(it.get("snippet", "") or it.get("htmlSnippet", ""), 200)
+        phone = extract_phone(title + " " + snippet)
+        entry = f"- {title}"
+        if phone:
+            entry += f" | Phone: {phone}"
+        if snippet:
+            entry += f" | Info: {snippet}"
+        if link:
+            entry += f" | {link}"
+        lines.append(entry)
+
+    return "\n".join(lines)
+
+
 def tool_search_wait_times_google(postal_code: str) -> str:
     """
-    Build a query biased to the user's city (from canadacities.csv) instead of
-    just generic province, so we get 'hospitals/clinics in <city>'.
+    For context links only; separate from facilities list.
     """
     if not postal_code:
         return "Postal code missing; Google Custom Search not used."
 
     city, prov = get_city_from_postal(postal_code)
     if city:
-        location_phrase = f"{city}, {prov or 'Canada'}"
+        loc = f"{city}, {prov or 'Canada'}"
     else:
-        location_phrase = f"{postal_code} Canada"
+        loc = f"{postal_code} Canada"
 
     data = google_cse_search(
-        query=(
-            f"hospital OR walk-in clinic OR urgent care OR emergency department in {location_phrase}"
-        ),
-        num=6,
+        query=f"emergency department wait times {loc}",
+        num=5,
     )
-
     if data is None:
         return (
             "Google Custom Search is not configured (set GOOGLE_API_KEY and GOOGLE_CSE_ID). "
@@ -248,15 +285,15 @@ def tool_search_wait_times_google(postal_code: str) -> str:
     items = data.get("items", [])
     if not items:
         return (
-            f"No strong Google CSE matches for {location_phrase}. "
+            f"No clear wait-time dashboards found for {loc}. "
             "Check provincial or hospital websites directly."
         )
 
-    lines = [f"Google CSE results for clinics/hospitals around {location_phrase} (verify directly):"]
-    for it in items[:6]:
+    lines = [f"Google CSE links for wait-time / access info near {loc} (verify sources):"]
+    for it in items[:5]:
         title = it.get("title", "Result")
         link = it.get("link", "")
-        snippet = shorten(it.get("snippet", "") or it.get("htmlSnippet", ""), 180)
+        snippet = shorten(it.get("snippet", "") or it.get("htmlSnippet", ""), 160)
         lines.append(f"- {title} — {link} — {snippet}")
 
     return "\n".join(lines)
@@ -269,10 +306,9 @@ def tool_hqontario_context(postal_code: str) -> str:
     if province != "Ontario":
         return ""
     return (
-        "For Ontario-specific emergency department locations and typical wait times near your postal code, "
-        "use Ontario Health's official tool: "
+        "For Ontario ED locations and typical wait times near your postal code, use Ontario Health's official tool: "
         "https://www.hqontario.ca/System-Performance/Time-Spent-in-Emergency-Departments "
-        "(select 'Area by Postal Code', enter your code, and review nearby hospitals)."
+        "(select 'Area by Postal Code')."
     )
 
 
@@ -288,31 +324,33 @@ def tool_region_context(postal_code: str) -> str:
             "Use Canada-wide guidance and suggest checking local provincial health ministry and Telehealth services."
         )
     telehealth_note = (
-        f"In {province}, advise use of the official provincial Telehealth or 811-style nurse line "
-        "for real-time nurse assessment."
+        f"In {province}, advise use of the official provincial Telehealth or 811-style nurse line."
     )
     wait_note = (
         "Users can check provincial/health authority resources and CIHI indicators "
         f"for average wait-time context (not real-time). Ref: {WAIT_TIMES_INFO_URL}"
     )
-    return (
-        f"User postal code '{postal_code}' mapped to {province}.\n"
-        f"{telehealth_note}\n"
-        f"{wait_note}"
-    )
+    return f"User postal code '{postal_code}' mapped to {province}.\n{telehealth_note}\n{wait_note}"
 
 
-def tool_find_nearby_clinics_osm(postal_code: str) -> str:
-    coords = geocode_postal(postal_code)
-    if not coords:
-        return (
-            "Could not geocode the postal code. Suggest using maps or provincial clinic finders."
-        )
+def tool_find_nearby_clinics(postal_code: str) -> str:
+    """
+    Main facility tool:
+    1) Try OSM within ~3km.
+    2) If none, fall back to city-based Google CSE facilities.
+    """
+    if not postal_code:
+        return "No postal code. Suggest using maps and provincial clinic/ER finders."
 
-    lat, lon = coords
-    radius_m = 3000
-    overpass_url = "https://overpass-api.de/api/interpreter"
-    query = f"""[out:json][timeout:8];
+    coords = geocode_postal(postal_code)
+    facilities_lines = []
+
+    # 1) OSM-based if geocoding worked
+    if coords:
+        lat, lon = coords
+        radius_m = 3000
+        overpass_url = "https://overpass-api.de/api/interpreter"
+        query = f"""[out:json][timeout:8];
 (
   node["amenity"="clinic"](around:{radius_m},{lat},{lon});
   node["amenity"="hospital"](around:{radius_m},{lat},{lon});
@@ -321,76 +359,74 @@ def tool_find_nearby_clinics_osm(postal_code: str) -> str:
 );
 out center;
 """
-    data = safe_get_json(overpass_url, method="POST", data=query)
-    if not data or "elements" not in data:
-        return (
-            "Nearby facility lookup via open map data is unavailable. "
-            "Use maps or provincial tools to find the closest clinic or ER."
-        )
-
-    facilities = []
-    for el in data["elements"]:
-        tags = el.get("tags", {})
-        name = tags.get("name")
-        if not name:
-            continue
-
-        amenity = tags.get("amenity", "clinic")
-        lat2 = el.get("lat") or (el.get("center") or {}).get("lat")
-        lon2 = el.get("lon") or (el.get("center") or {}).get("lon")
-        if lat2 is None or lon2 is None:
-            continue
-
-        dist = haversine_km(lat, lon, float(lat2), float(lon2))
-
-        addr_parts = []
-        for k in ("addr:street", "addr:city"):
-            if tags.get(k):
-                addr_parts.append(tags[k])
-        addr = ", ".join(addr_parts)
-
-        wait = None
-        if WAIT_TIMES_API_BASE:
-            jt = safe_get_json(
-                f"{WAIT_TIMES_API_BASE}/wait",
-                params={"name": name},
-                timeout=4
-            )
-            if jt and isinstance(jt, dict) and jt.get("estimated_minutes") is not None:
-                wait = int(jt["estimated_minutes"])
-
-        facilities.append(
-            {
-                "name": name,
-                "type": amenity,
-                "distance_km": dist,
-                "address": addr,
-                "wait": wait,
-            }
-        )
-
-    facilities = sorted(facilities, key=lambda x: x["distance_km"])[:5]
-    if not facilities:
+        data = safe_get_json(overpass_url, method="POST", data=query)
+        if data and "elements" in data:
+            facilities = []
+            for el in data["elements"]:
+                tags = el.get("tags", {})
+                name = tags.get("name")
+                if not name:
+                    continue
+                amenity = tags.get("amenity", "clinic")
+                lat2 = el.get("lat") or (el.get("center") or {}).get("lat")
+                lon2 = el.get("lon") or (el.get("center") or {}).get("lon")
+                if lat2 is None or lon2 is None:
+                    continue
+                dist = haversine_km(lat, lon, float(lat2), float(lon2))
+                addr_parts = []
+                for k in ("addr:street", "addr:city"):
+                    if tags.get(k):
+                        addr_parts.append(tags[k])
+                addr = ", ".join(addr_parts)
+                wait = None
+                if WAIT_TIMES_API_BASE:
+                    jt = safe_get_json(
+                        f"{WAIT_TIMES_API_BASE}/wait",
+                        params={"name": name},
+                        timeout=4
+                    )
+                    if jt and isinstance(jt, dict) and jt.get("estimated_minutes") is not None:
+                        wait = int(jt["estimated_minutes"])
+                facilities.append(
+                    {
+                        "name": name,
+                        "type": amenity,
+                        "distance_km": dist,
+                        "address": addr,
+                        "wait": wait,
+                    }
+                )
+            facilities = sorted(facilities, key=lambda x: x["distance_km"])[:5]
+            if facilities:
+                facilities_lines.append("Closest facilities from open map data (approx., within ~3 km):")
+                for f in facilities:
+                    s = f"- {f['name']} ({f['type']}, ~{f['distance_km']:.1f} km"
+                    if f["address"]:
+                        s += f", {f['address']}"
+                    if f["wait"] is not None:
+                        s += f", est. wait ~{f['wait']} min"
+                    s += ")"
+                    facilities_lines.append(s)
+
+    # 2) City-based Google CSE fallback (if OSM empty or as extra signal)
+    city_facilities = tool_city_facilities_google(postal_code)
+    if city_facilities:
+        if facilities_lines:
+            facilities_lines.append("")  # spacing
+        facilities_lines.append(city_facilities)
+
+    if not facilities_lines:
         return (
-            "No nearby facilities found within about 3 km. Use maps or provincial tools to locate the closest clinic/ER."
+            "Unable to list nearby facilities automatically. "
+            "Please use provincial or hospital 'find a clinic/ER' tools or maps with your postal code."
         )
 
-    lines = ["Closest facilities (approximate from open data, within ~3 km):"]
-    for f in facilities:
-        s = f"- {f['name']} ({f['type']}, ~{f['distance_km']:.1f} km"
-        if f["address"]:
-            s += f", {f['address']}"
-        if f["wait"] is not None:
-            s += f", est. wait ~{f['wait']} min"
-        s += ")"
-        lines.append(s)
-
     if not WAIT_TIMES_API_BASE:
-        lines.append(
-            "Note: Live wait times are not provided here. Check the facility or provincial website directly."
+        facilities_lines.append(
+            "Note: Any wait-time or status information must be checked directly on the facility or provincial website."
         )
 
-    return "\n".join(lines)
+    return "\n".join(facilities_lines)
 
 
 def tool_lookup_drug_products(text: str) -> str:
@@ -483,19 +519,13 @@ def tool_get_wait_times_awareness() -> str:
 def call_vision_summarizer(image_bytes: bytes) -> str:
     if not client or not image_bytes:
         return ""
-
     b64 = base64.b64encode(image_bytes).decode("utf-8")
-
     prompt = (
         "You are a cautious Canadian health-information assistant.\n"
         "Describe ONLY what is visible in neutral terms.\n"
-        "- For skin/nails: describe colour, dryness, cracks, swelling, etc. Do NOT name diseases.\n"
-        "- For medication labels: mention visible drug names/strengths if legible.\n"
-        "- For letters/reports: say what type of document it appears to be; avoid identifiers.\n"
-        "- If unclear: say it is not medically interpretable.\n"
-        "Never diagnose or prescribe. Keep to ~80-100 words."
+        "No diagnoses or prescriptions.\n"
+        "Keep to about 80-100 words."
     )
-
     try:
         resp = client.chat.completions.create(
             model=VISION_MODEL,
@@ -528,9 +558,7 @@ def call_asr(audio_file) -> str:
             file=bio
         )
         txt = getattr(transcript, "text", None)
-        if isinstance(txt, str):
-            return txt.strip()
-        return str(transcript)
+        return txt.strip() if isinstance(txt, str) else str(transcript)
     except Exception as e:
         return f"(Transcription unavailable: {e})"
 
@@ -543,8 +571,8 @@ def call_reasoning_agent(
     recalls_context: str,
     wait_awareness_context: str,
     region_context: str,
-    nearby_osm_context: str,
-    google_search_context: str,
+    facilities_context: str,
+    wait_links_context: str,
     hqontario_context: str,
 ) -> str:
     if not client:
@@ -553,55 +581,68 @@ def call_reasoning_agent(
             "into cautious, structured guidance."
         )
 
-    system_prompt = """You are CareCall AI, an agentic Canadian health information assistant.
+    system_prompt = """
+You are CareCall AI, an agentic Canadian health information assistant.
 
-You receive:
-- Narrative, vision summary, postal code.
-- Drug Product Database context.
-- Recalls & Safety Alerts snapshot.
-- Wait-time awareness notes.
-- Region context (province + Telehealth).
-- Nearby facilities from OSM.
-- Google Custom Search context (official pages).
-- HQOntario/Ontario Health tool context when applicable.
+Use ONLY the provided context:
+- narrative, vision summary, postal;
+- drug product info;
+- recalls info;
+- wait-time awareness;
+- region context;
+- facilities context (OSM + city-based Google CSE);
+- wait-time links context;
+- HQOntario context.
 
 Rules:
-- Do NOT give a diagnosis.
-- Do NOT prescribe or specify dosing.
-- Do NOT invent real-time wait times.
-- Use ONLY the provided context.
-- Keep response under ~350 words.
-- Be clear, neutral, practical.
-
-Pick EXACTLY ONE primary pathway:
+- NO diagnosis.
+- NO prescribing or exact dosing.
+- NO fabricated real-time wait times.
+- Max ~350 words.
+- Clear, neutral, practical.
+
+Choose EXACTLY ONE primary pathway:
 A) "Home care likely reasonable (monitor closely)"
 B) "Pharmacist / Walk-in clinic / Family doctor recommended"
 C) "Call provincial Telehealth / nurse line for guidance"
 D) "Go to Emergency / Call 911 now"
 
-Conditional sections:
-- If A: include Home Care; When to see Pharmacist/Clinic/Telehealth; When this is an Emergency (ER/911);
-        Local & Official Resources; Important Disclaimer.
-- If B or C: include When to see Pharmacist/Clinic/Telehealth (as primary); When this is an Emergency (ER/911);
-        Local & Official Resources; Important Disclaimer. Do NOT include a Home Care block.
-- If D: include ONLY When this is an Emergency (ER/911); Local & Official Resources; Important Disclaimer.
-
-In Local & Official Resources:
+Sections:
+- Always start with **Primary Recommended Pathway** (one bullet + short reason).
+- If A: then show
+    **Home Care**,
+    **When to see a Pharmacist / Clinic / Telehealth**,
+    **When this is an Emergency (ER/911)**,
+    **Local & Official Resources**,
+    **Important Disclaimer**.
+- If B or C: then show
+    **When to see a Pharmacist / Clinic / Telehealth** (primary),
+    **When this is an Emergency (ER/911)**,
+    **Local & Official Resources**,
+    **Important Disclaimer**.
+    (No Home Care block.)
+- If D: then show ONLY
+    **When this is an Emergency (ER/911)**,
+    **Local & Official Resources**,
+    **Important Disclaimer**.
+    (No Home Care or Walk-in blocks.)
+
+In **Local & Official Resources**:
 - Use region_context.
-- Use nearby_osm_context.
-- Include hqontario_context when relevant (Ontario).
-- Optionally include google_search_context (brief).
-- Mention Health Canada / PHAC / Recalls / DPD where useful.
-
-Use only broad, non-diagnostic categories (e.g., "possible infection risk", "nail or soft tissue injury")."""
+- Use facilities_context (names/locations/phones/links).
+- Include hqontario_context when present.
+- Optionally summarize wait_links_context.
+- Mention Health Canada / Recalls / DPD if relevant.
 
-    user_prompt = f"""INPUT CONTEXT
+Use broad categories only (e.g., "possible infection risk", "soft tissue or nail injury").
+"""
 
+    user_prompt = f"""
 User narrative:
 {narrative or "(none provided)"}
 
 Vision summary:
-{vision_summary or "(no image or no usable visual details)"}
+{vision_summary or "(none)"}
 
 Postal code:
 {postal_code or "(not provided)"}
@@ -609,25 +650,25 @@ Postal code:
 Drug Product Database context:
 {dpd_context}
 
-Recalls & Safety Alerts context:
+Recalls context:
 {recalls_context}
 
-Wait-time awareness (general):
+Wait-time awareness:
 {wait_awareness_context}
 
 Region context:
 {region_context}
 
-Nearby facilities (OSM-based):
-{nearby_osm_context}
+Facilities context (OSM + city-based Google CSE):
+{facilities_context}
 
-Google search context:
-{google_search_context}
+Wait-time links context:
+{wait_links_context}
 
-HQOntario/Ontario Health context:
+HQOntario context:
 {hqontario_context or "(not applicable)"}
 
-Now follow the system instructions and generate the final structured response.
+Now follow the system instructions exactly.
 """
 
     try:
@@ -650,34 +691,30 @@ Now follow the system instructions and generate the final structured response.
 # ============== STREAMLIT UI ==============
 
 st.title("CareCall AI (Canada)")
-st.caption("Vision + Voice + Postal + City-aware search • Pathway-specific non-diagnostic guidance")
+st.caption("Vision + Voice + Postal + City-aware facilities • Pathway-specific non-diagnostic guidance")
 
 if not OPENAI_API_KEY:
     st.stop()
 
 st.markdown(
     """
-Upload a photo, add a voice note, and/or type what's going on.  
-Optionally enter your **Canadian postal code** so we can:
-- infer province and Telehealth options,
-- look up your city from `canadacities.csv`,
-- search for nearby clinics/hospitals in that city,
-- and suggest local official resources.
-
-The agent chooses **one** primary pathway and only shows sections for that pathway.
+Enter your details below. The assistant will:
+- interpret your description and image (non-diagnostically),
+- detect your city from the postal code (via an internal table),
+- list nearby hospitals/clinics using open map data and trusted Google CSE results,
+- link to official wait-time tools where appropriate,
+- and give ONE clear primary pathway (home care vs clinic vs ER), with only relevant sections shown.
 """
 )
 
-# Postal code
 postal_code = st.text_input(
-    "Canadian postal code (e.g., M5V 2T6). Leave blank for general guidance.",
+    "Canadian postal code (e.g., M5V 2T6).",
     max_chars=7,
 )
 
-# Image
 st.markdown("### Image (optional)")
 image_file = st.file_uploader(
-    "Upload a photo (JPG/PNG). Avoid highly identifying content.",
+    "Upload a photo (JPG/PNG).",
     type=["jpg", "jpeg", "png"],
     accept_multiple_files=False,
 )
@@ -693,7 +730,6 @@ if image_file is not None:
         st.warning(f"Could not read that image: {e}")
         image_bytes = None
 
-# Audio
 st.markdown("### Voice note (optional)")
 audio_file = st.file_uploader(
     "Upload a short voice note (wav/mp3/m4a).",
@@ -701,15 +737,12 @@ audio_file = st.file_uploader(
     accept_multiple_files=False,
 )
 
-# Text
 st.markdown("### Description")
 user_text = st.text_area(
-    "Describe what's going on:",
-    placeholder='Example: "Big toenail is sore and a bit red for 5 days when wearing tight shoes, no fever, I can walk okay."',
+    "Describe what is happening:",
     height=120,
 )
 
-# Run
 if st.button("Run CareCall Agent"):
     with st.spinner("Analyzing inputs and generating guidance..."):
         vision_summary = call_vision_summarizer(image_bytes) if image_bytes else ""
@@ -723,33 +756,20 @@ if st.button("Run CareCall Agent"):
         narrative = "\n".join(parts)
 
         combined_for_drugs = " ".join(x for x in [narrative, vision_summary] if x)
-        dpd_context = (
-            tool_lookup_drug_products(combined_for_drugs)
-            if combined_for_drugs else "No medication context."
-        )
-
+        dpd_context = tool_lookup_drug_products(combined_for_drugs) if combined_for_drugs else "No medication context."
         recalls_context = tool_get_recent_recalls_snippet()
         wait_awareness_context = tool_get_wait_times_awareness()
 
-        region_context = (
-            tool_region_context(postal_code)
-            if postal_code
-            else "No postal code provided. Use Canada-wide guidance and mention provincial resources."
-        )
-
-        nearby_osm_context = (
-            tool_find_nearby_clinics_osm(postal_code)
-            if postal_code
-            else "No postal code. Suggest using maps and provincial clinic/ER finder tools."
-        )
-
-        google_search_context = (
-            tool_search_wait_times_google(postal_code)
-            if postal_code
-            else "Postal code missing; Google Custom Search not used."
-        )
-
-        hqontario_context = tool_hqontario_context(postal_code) if postal_code else ""
+        if postal_code:
+            region_context = tool_region_context(postal_code)
+            facilities_context = tool_find_nearby_clinics(postal_code)
+            wait_links_context = tool_search_wait_times_google(postal_code)
+            hqontario_context = tool_hqontario_context(postal_code)
+        else:
+            region_context = "No postal code provided. Use Canada-wide guidance."
+            facilities_context = "No postal code; cannot list nearby facilities."
+            wait_links_context = "No postal code; no local wait-time links."
+            hqontario_context = ""
 
         final_answer = call_reasoning_agent(
             narrative=narrative,
@@ -759,8 +779,8 @@ if st.button("Run CareCall Agent"):
             recalls_context=recalls_context,
             wait_awareness_context=wait_awareness_context,
             region_context=region_context,
-            nearby_osm_context=nearby_osm_context,
-            google_search_context=google_search_context,
+            facilities_context=facilities_context,
+            wait_links_context=wait_links_context,
             hqontario_context=hqontario_context,
         )