Update app.py

app.py

@@ -18,28 +18,25 @@ st.set_page_config(
 )
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-if not OPENAI_API_KEY:
-    st.error("Missing OPENAI_API_KEY. Please set it in your Space secrets.")
 client = OpenAI(api_key=OPENAI_API_KEY) if OPENAI_API_KEY else None
 
-# Models
-VISION_MODEL = "gpt-4.1-mini"     # vision-capable
-REASONING_MODEL = "gpt-4.1-mini"  # reasoning / agent
-ASR_MODEL = "whisper-1"           # transcription
+VISION_MODEL = "gpt-4.1-mini"     # vision-capable OpenAI model
+REASONING_MODEL = "gpt-4.1-mini"  # reasoning / agent model
+ASR_MODEL = "whisper-1"           # transcription model
 
-# Health Canada Drug Product Database API (public)
+# Health Canada Drug Product Database API
 DPD_BASE_URL = "https://health-products.canada.ca/api/drug/drugproduct"
 
-# Recalls & Safety Alerts JSON feed (configurable; keep overridable)
+# Recalls & Safety Alerts JSON feed (example; can override via env if GC changes path)
 RECALLS_FEED_URL = os.getenv(
     "RECALLS_FEED_URL",
     "https://recalls-rappels.canada.ca/static-data/items-en.json"
 )
 
-# CIHI wait-times info (general reference, not live triage)
+# 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 (by first letter)
+# Postal code to province mapping (first letter)
 POSTAL_PREFIX_TO_PROVINCE = {
     "A": "Newfoundland and Labrador",
     "B": "Nova Scotia",
@@ -58,12 +55,12 @@ POSTAL_PREFIX_TO_PROVINCE = {
     "T": "Alberta",
     "V": "British Columbia",
     "X": "Northwest Territories / Nunavut",
-    "Y": "Yukon"
+    "Y": "Yukon",
 }
 
 
 # =========================
-# HELPERS
+# GENERIC HELPERS
 # =========================
 
 def safe_get_json(url, params=None, timeout=6):
@@ -86,85 +83,63 @@ def shorten(text, max_chars=400):
 
 
 # =========================
-# REGION CONTEXT (POSTAL CODE)
+# REGION / POSTAL HELPERS
 # =========================
 
 def infer_province_from_postal(postal_code: str):
-    """
-    Very simple FSA-based province inference using the first character.
-    """
     if not postal_code:
         return None
     pc = postal_code.strip().upper().replace(" ", "")
     if not pc:
         return None
-    first = pc[0]
-    return POSTAL_PREFIX_TO_PROVINCE.get(first)
+    return POSTAL_PREFIX_TO_PROVINCE.get(pc[0])
 
 
 def tool_region_context(postal_code: str) -> str:
-    """
-    Build region-aware context for the agent:
-    - Province guess (if possible)
-    - Telehealth-style guidance
-    - Where to check local wait times / services
-    """
     province = infer_province_from_postal(postal_code)
     if not postal_code:
-        base = "No postal code provided. Provide general Canada-wide guidance."
-        return base
+        return "No postal code provided. Use general Canada-wide guidance."
 
     if not province:
         return (
-            f"Postal code '{postal_code}' could not be confidently mapped. "
-            "Provide generic Canada-wide guidance and suggest checking local provincial health websites."
+            f"Postal code '{postal_code}' could not be mapped reliably. "
+            "Provide Canada-wide guidance and advise checking local provincial health websites and Telehealth services."
         )
 
-    # Generic, safe phrasing without hard-coding all numbers:
     telehealth_note = (
-        f"In {province}, guide the user to their official provincial Telehealth or 811-style nurse advice line "
-        "and ministry of health website for region-specific instructions."
+        f"In {province}, suggest the official provincial Telehealth / 811-style nurse advice line "
+        "and Ministry of Health website for region-specific guidance."
     )
-
     wait_note = (
-        "Advise the user that approximate wait-time and access information may be available via their "
-        "provincial or regional health authority website, and through CIHI's access and wait-times resources. "
-        f"Reference (non-exclusively): {WAIT_TIMES_INFO_URL}"
+        "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}"
     )
 
     return (
-        f"User-provided postal code: '{postal_code}' mapped to: {province}.\n"
+        f"User postal code: '{postal_code}' mapped to {province}.\n"
         f"{telehealth_note}\n"
         f"{wait_note}"
     )
 
 
 # =========================
-# TOOL: DRUG PRODUCT LOOKUP (DPD)
+# DRUG PRODUCT LOOKUP (DPD)
 # =========================
 
 def tool_lookup_drug_products(candidates):
     if not candidates:
-        return "No clear medication names detected that require a drug database lookup."
+        return "No clear medication/product names detected that require a Drug Product Database lookup."
 
     found_lines = []
     for name in candidates[:5]:
-        params = {
-            "lang": "en",
-            "type": "json",
-            "brandname": name
-        }
+        params = {"lang": "en", "type": "json", "brandname": name}
         data = safe_get_json(DPD_BASE_URL, params=params)
         if not data:
             continue
 
         unique_brands = set()
         for item in data[:3]:
-            b = (
-                item.get("brand_name")
-                or item.get("brandname")
-                or name
-            )
+            b = item.get("brand_name") or item.get("brandname") or name
             if b:
                 unique_brands.add(b)
 
@@ -181,30 +156,37 @@ def tool_lookup_drug_products(candidates):
         )
 
     found_lines.append(
-        "For definitive product information, ingredients, status, and monographs, "
-        "users must consult the official Health Canada Drug Product Database."
+        "For definitive product information, users must consult the official Health Canada Drug Product Database."
     )
     return "\n".join(found_lines)
 
 
 # =========================
-# TOOL: RECALLS & SAFETY ALERTS
+# RECALLS & SAFETY ALERTS
 # =========================
 
 def tool_get_recent_recalls_snippet():
     data = safe_get_json(RECALLS_FEED_URL)
-    if not data or not isinstance(data, list):
+    items = []
+
+    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:
         return (
-            "Unable to fetch recent recalls. Direct users to the official Government of Canada "
+            "Unable to load recent recalls. Direct users to the official Government of Canada "
             "Recalls and Safety Alerts website for up-to-date information."
         )
 
-    items = data[:5]
-    lines = ["Recent recalls and safety alerts (high-level snapshot):"]
+    lines = ["Recent recalls & safety alerts (high-level snapshot):"]
     for item in items:
         title = (
             item.get("title")
             or item.get("english_title")
+            or item.get("name")
             or "Recall / Alert"
         )
         date = item.get("date_published") or item.get("date") or ""
@@ -217,23 +199,23 @@ def tool_get_recent_recalls_snippet():
 
 
 # =========================
-# TOOL: WAIT TIMES AWARENESS (GENERIC)
+# WAIT TIMES AWARENESS (GENERIC)
 # =========================
 
 def tool_get_wait_times_awareness():
     return textwrap.dedent(f"""
-    Use open Canadian data conceptually:
-    - CIHI and several provinces publish indicators and dashboards for emergency room, surgery, and diagnostic wait times.
-    - These are averages and not real-time triage.
+    Use Canadian open-data wait-time indicators conceptually:
+    - CIHI and several provinces publish average wait-time dashboards.
+    - These are not real-time triage.
     Guidance:
-    - For non-urgent issues: consider checking local clinic or urgent care centre information online.
-    - For urgent symptoms: do not spend time checking wait-time tools; go to the nearest ER or call 911.
-    Reference (general): {WAIT_TIMES_INFO_URL}
+    - 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.
+    General reference: {WAIT_TIMES_INFO_URL}
     """).strip()
 
 
 # =========================
-# TOOL: CANDIDATE TERM EXTRACTION
+# TERM EXTRACTION
 # =========================
 
 def extract_candidate_terms(text: str):
@@ -248,7 +230,7 @@ def extract_candidate_terms(text: str):
 
 
 # =========================
-# OPENAI: VISION, ASR, AGENT
+# OPENAI CALLS
 # =========================
 
 def call_vision_summarizer(image_bytes: bytes) -> str:
@@ -256,14 +238,15 @@ def call_vision_summarizer(image_bytes: bytes) -> str:
         return ""
 
     b64 = base64.b64encode(image_bytes).decode("utf-8")
+
     prompt = (
         "You are a cautious Canadian health-information assistant.\n"
-        "Look at the image and ONLY describe observable details in neutral terms.\n"
-        "- If it's a skin issue: describe general appearance (location/size/colour/pattern). Do NOT name diseases.\n"
-        "- If it's a pill bottle or box: note visible drug name/strength/marks if legible.\n"
-        "- If it's a health letter/report: state the type (e.g., appointment letter, lab report) without exposing identifiers.\n"
-        "- If it's irrelevant or unclear: say it is not medically interpretable.\n"
-        "Never diagnose. Never prescribe. Keep within ~80-100 words."
+        "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."
     )
 
     try:
@@ -339,29 +322,26 @@ def call_reasoning_agent(
 You are CareCall AI, an agentic Canadian health information assistant.
 
 Rules:
-- You DO NOT diagnose.
-- You DO NOT prescribe or give specific dosing.
-- You DO NOT claim exact real-time wait times.
-- You MAY:
-  - Summarize concerns neutrally.
-  - Map them to broad, possible categories (e.g., irritation, infection, allergy, medication question, admin/letter).
-  - Use the provided Drug Database context, Recalls context, wait-time awareness, and region context
-    to make the guidance safer and more relevant.
-  - Suggest next steps:
-      * Self-care for clearly mild issues,
-      * Pharmacist / family doctor / walk-in clinic,
-      * Provincial Telehealth / 811-style nurse line,
-      * ER / 911 if red-flag symptoms are present.
-- Red flags (e.g., chest pain, difficulty breathing, signs of stroke, heavy bleeding, suicidal thoughts,
-  severe abdominal pain, high fever with stiff neck, major trauma, rapidly worsening symptoms)
-  -> clearly recommend ER/911.
-- Always mention that official sources include:
-  Health Canada, Public Health Agency of Canada, CIHI, and the user's provincial health services.
-- Max 350 words. Clear, calm, practical.
+- 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.
 """
 
     user_prompt = f"""
-INPUT TO THE AGENT
+INPUT CONTEXT
 
 1) User narrative:
 {narrative}
@@ -383,17 +363,17 @@ INPUT TO THE AGENT
 
 TASK
 
-Using ONLY the above information:
-
-1. Briefly summarize what the user seems worried about (neutral; no diagnosis).
-2. Describe, in broad terms, what categories of issues it could relate to, without asserting a specific disease.
-3. Provide clear next-step options in bullet form, aligned with Canadian context and (if available) their province:
-   - When self-care might be reasonable.
-   - When to talk to a pharmacist, family doctor, or walk-in clinic.
-   - When to call their provincial Telehealth / nurse line.
-   - When to bypass everything and go to ER / call 911.
-4. If any medication/recall hints appear, remind them to confirm via Health Canada resources and follow professional advice.
-5. End with a concise disclaimer that this is informational only and not medical care.
+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.
 """
 
     try:
@@ -409,7 +389,7 @@ Using ONLY the above information:
     except Exception as e:
         return (
             f"(Reasoning agent unavailable: {e})\n\n"
-            "Please contact a licensed provider, Telehealth, or emergency services as appropriate."
+            "Please contact a licensed provider, Telehealth service, or emergency care as appropriate."
         )
 
 
@@ -420,31 +400,38 @@ Using ONLY the above information:
 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.markdown(
     """
-This demo:
-- Accepts a photo (rash, pill label, letter, etc.).
-- Accepts a voice note and/or typed description.
-- Optionally uses your **postal code** to tailor guidance to your province.
-- Uses:
-  - OpenAI vision (for safe visual summary),
-  - Whisper (for transcription),
-  - Canadian open-data style tools (Drug Product Database, recalls, wait-time context),
-  - A reasoning agent with strict safety rules.
+**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.
 """
 )
 
-# Postal code
+# 1) Postal code
 st.markdown("### 1. Postal code (optional, Canada only)")
 postal_code = st.text_input(
-    "Enter your Canadian postal code (e.g., M5V 2T6). Leave blank if you prefer not to share.",
-    max_chars=7
+    "Canadian postal code (e.g., M5V 2T6). Leave blank to keep guidance general.",
+    max_chars=7,
 )
 
-# Image upload
+# 2) Image upload
 st.markdown("### 2. Upload an image (optional)")
 image_file = st.file_uploader(
-    "Upload a photo (JPG/PNG). Avoid sharing highly identifying or explicit content.",
+    "Upload a photo (JPG/PNG). Avoid very identifying or explicit content.",
     type=["jpg", "jpeg", "png"],
     accept_multiple_files=False,
 )
@@ -452,8 +439,9 @@ image_file = st.file_uploader(
 image_bytes = None
 if image_file is not None:
     try:
-        img = Image.open(image_file)
-        st.image(img, caption="Image received", use_column_width=True)
+        # Convert to RGB to avoid RGBA -> JPEG error
+        img = Image.open(image_file).convert("RGB")
+        st.image(img, caption="Image received", use_container_width=True)
         buf = io.BytesIO()
         img.save(buf, format="JPEG")
         image_bytes = buf.getvalue()
@@ -461,7 +449,7 @@ if image_file is not None:
         st.warning(f"Could not read that image: {e}")
         image_bytes = None
 
-# Audio upload
+# 3) Audio upload
 st.markdown("### 3. Add a voice note (optional)")
 audio_file = st.file_uploader(
     "Upload a short voice note (wav/mp3/m4a).",
@@ -469,9 +457,10 @@ audio_file = st.file_uploader(
     accept_multiple_files=False,
 )
 
-# Text description
+# 4) Text description
+st.markdown("### 4. Or type a short description")
 user_text = st.text_area(
-    "Or type a short description:",
+    "Describe what’s going on:",
     placeholder='Example: "Red itchy patch on forearm for 3 days, mild burning, no fever, using a new soap."',
     height=120,
 )
@@ -493,7 +482,7 @@ if st.button("Run CareCall Agent"):
             if audio_file is not None:
                 voice_text = call_asr(audio_file)
 
-            # Candidates for DPD lookup
+            # Candidate terms (for Drug DB)
             combined_for_terms = " ".join(
                 t for t in [user_text or "", voice_text or "", vision_summary or ""] if t
             )
@@ -503,12 +492,14 @@ if st.button("Run CareCall Agent"):
             # Recalls
             recalls_context = tool_get_recent_recalls_snippet()
 
-            # Wait-time awareness
+            # Wait-times awareness
             wait_times_context = tool_get_wait_times_awareness()
 
-            # Region context from postal
-            region_context = tool_region_context(postal_code) if postal_code else (
-                "No postal code provided. Use Canada-wide guidance and suggest provincial resources."
+            # 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."
             )
 
             # Final reasoning
@@ -528,15 +519,16 @@ if st.button("Run CareCall Agent"):
         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 care.
-- If you have severe or worsening symptoms (such as chest pain, difficulty breathing,
-  signs of stroke, heavy bleeding, suicidal thoughts, major injury, high fever with stiff neck),
-  call 911 or go to the nearest emergency department immediately.
-- For dependable information, consult:
-  Health Canada, the Public Health Agency of Canada, CIHI, and your provincial/territorial health services.
+**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.
 """
         )