Update app.py

app.py

@@ -343,17 +343,58 @@ def send_ally_ai_email(to_email, subject, body, user_email,
 # GRADIO UI
 # ============================
 def run_search(story, country):
-    out = find_professionals_from_story(story, country=country, results_per_query=RESULTS_PER_QUERY)
-    df = pd.DataFrame(out["professionals"])
-    # Build dropdown labels: "0 — Title (email/phone)"
+    """Robust search wrapper: returns (summary, table_records, dropdown_update, anonymized_text)"""
+    try:
+        out = find_professionals_from_story(story, country=country, results_per_query=RESULTS_PER_QUERY)
+    except Exception as e:
+        # If the backend search crashed, return a friendly error and placeholder dropdown
+        err_msg = f"Search failed: {e}"
+        placeholder = "0 — No results (search failed)"
+        return err_msg, [], gr.update(choices=[placeholder], value=placeholder), ""
+
+    # Build dataframe records (list of dicts)
+    pros = out.get("professionals", []) or []
+    try:
+        df_records = pd.DataFrame(pros).to_dict(orient="records")
+    except Exception:
+        # fallback to empty list
+        df_records = []
+
+    # Build dropdown options safely (guarantee at least one option)
     options = []
-    for i, r in enumerate(out["professionals"]):
-        label_contact = r.get("email") if r.get("email") and r.get("email")!="Not found" else (r.get("phone","No contact"))
-        label = f"{i} — {r.get('title','(no title)')} ({label_contact})"
+    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"))
+        title = r.get("title") or r.get("url") or "(no title)"
+        label = f"{i} — {title} ({label_contact})"
         options.append(label)
-    # Anonymize the story
-    anon = anonymize_story(story) or "I am seeking confidential support regarding gender-based violence."
-    return out["summary"], df.to_dict(orient="records"), gr.update(choices=options, value=(options[0] if options else None)), anon
+
+    if not options:
+        options = ["0 — No results (try a different country/query)"]
+
+    # Anonymize story in a safe way (if anonymizer fails, fallback to short template)
+    try:
+        anon = anonymize_story(story) or "I am seeking confidential support regarding gender-based violence."
+    except Exception as e:
+        print("[anonymize error]", e)
+        anon = "I am seeking confidential support regarding gender-based violence."
+
+    # Return summary, table records, dropdown update and anonymized text
+    summary = out.get("summary", "No results found.")
+    return summary, df_records, gr.update(choices=options, value=options[0]), anon
+
+# -------------------------
+# If you used a _on_search wrapper, replace it similarly (example below)
+# This is the function wired to the search button in your Gradio UI.
+# -------------------------
+def _on_search(story, country):
+    """
+    This wrapper simply calls run_search and also returns a safe prefilled message body.
+    It is the function connected to search_btn.click(...)
+    """
+    summary, records, dropdown_update, anon = run_search(story, country)
+    # Prefill message body with anonymized summary (user email blank for now)
+    prefill = make_body(anon, story, True, "")
+    return summary, records, dropdown_update, anon, prefill
 
 def make_body(anon_text, full_story, use_anon, user_email):
     core = (anon_text or "").strip() if use_anon else (full_story or "").strip()