Update app.py

app.py

@@ -9,6 +9,7 @@ import urllib.parse
 from bs4 import BeautifulSoup
 import json
 from pathlib import Path
+import traceback
 
 
 import torch
@@ -580,25 +581,34 @@ def send_ally_ai_email(to_email, subject, body, user_email,
 
 def run_search(story, country):
     """
-    Robust search wrapper: returns (summary, table_records, dropdown_options, anonymized_text).
-    Avoids returning gr.update(...) to prevent KeyError during serialization.
+    Robust wrapper around find_professionals_from_story.
+    Always returns: (summary:str, records:list[dict], options:list[str], anonymized_text:str)
     """
+    # default safe return
+    default_summary = "No results found (an error occurred or no matches)."
+    default_records, default_options, default_anon = [], ["0 — No results (try again)"], "I am seeking confidential support."
+
     try:
         out = find_professionals_from_story(story, country=country, results_per_query=RESULTS_PER_QUERY)
+        # If the function returned None or invalid type, replace with defaults
+        if not isinstance(out, dict):
+            raise ValueError("find_professionals_from_story returned non-dict or None")
     except Exception as e:
-        err_msg = f"Search failed: {e}"
-        placeholder = ["0 — No results (search failed)"]
-        return err_msg, [], placeholder, ""
+        # Log the full traceback so you can inspect logs in the Space
+        print("[run_search] find_professionals_from_story error:", e)
+        traceback.print_exc()
+        out = {"summary": f"Search failed: {str(e)}", "professionals": [], "queries_used": []}
 
     pros = out.get("professionals", []) or []
 
-    # build table records
+    # Build table records safely
     try:
         records = pd.DataFrame(pros).to_dict(orient="records") if pros else []
-    except Exception:
+    except Exception as e:
+        print("[run_search] DataFrame conversion error:", e)
         records = []
 
-    # build dropdown options as list of strings (guarantee at least one)
+    # Build dropdown options as plain list
     options = []
     for i, r in enumerate(pros):
         label_contact = r.get("email") if r.get("email") and r.get("email") != "Not found" else (r.get("phone", "No contact"))
@@ -607,37 +617,33 @@ def run_search(story, country):
         options.append(label)
 
     if not options:
-        options = ["0 — No results (try a different country/query)"]
+        options = default_options
 
-    # anonymize safely
+    # Anonymize safely (fallback if anonymizer fails)
     try:
-        anon = anonymize_story(story) or "I am seeking confidential support regarding gender-based violence."
+        anon = anonymize_story(story) or default_anon
     except Exception as e:
-        print("[anonymize error]", e)
-        anon = "I am seeking confidential support regarding gender-based violence."
+        print("[run_search] anonymize_story error:", e)
+        anon = default_anon
 
-    summary = out.get("summary", "No results found.")
+    summary = out.get("summary", default_summary)
     return summary, records, options, anon
 
 
 def _on_search(story, country):
     """
-    Function wired to the search button.
-    Returns exactly 5 outputs to match:
+    Function wired to search_btn.click(...)
+    Must return exactly these outputs:
       [summary_out, results_table, dropdown_sel, anon_out, message_in]
     """
     summary, records, options, anon = run_search(story, country)
 
-    # pre-fill message body with anonymized text (user email left empty for now)
+    # Prefill message body (user email left blank for now)
     prefill = make_body(anon, story, True, "")
 
     # Return plain serializable values (not gr.update)
-    # summary -> str
-    # records -> list[dict] (or [])
-    # options -> list[str] for dropdown (Gradio will accept it)
-    # anon -> str
-    # prefill -> str (message body)
     return summary, records, options, anon, prefill
+# ---------- end replacements ----------
 
 
 def make_body(anon_text, full_story, use_anon, user_email):