Update app.py

app.py

@@ -1,4 +1,3 @@
-import asyncio
 import time
 from typing import List, Dict
 
@@ -14,16 +13,16 @@ import coverage
 import saved_searches
 
 
-# -----------------------------
+# ======================================================
 # Adapter Registry
-# -----------------------------
+# ======================================================
 
 ALL_ADAPTERS = get_all_adapters()
 
 
-# -----------------------------
-# Async Search Engine
-# -----------------------------
+# ======================================================
+# Async Federated Search Engine
+# ======================================================
 
 async def _search_adapter(adapter, query: str) -> Dict:
     start = time.time()
@@ -57,7 +56,7 @@ async def federated_search_async(
     ]
 
     tasks = [_search_adapter(a, query) for a in adapters]
-    responses = await asyncio.gather(*tasks)
+    responses = await gradio_async_gather(tasks)
 
     flat_results = []
     health = {}
@@ -73,9 +72,17 @@ async def federated_search_async(
     return flat_results, health
 
 
-# -----------------------------
+async def gradio_async_gather(tasks):
+    """
+    HF-safe async gather wrapper.
+    """
+    import asyncio
+    return await asyncio.gather(*tasks, return_exceptions=False)
+
+
+# ======================================================
 # Rendering Helpers
-# -----------------------------
+# ======================================================
 
 def badge(result: Dict) -> str:
     badges = []
@@ -113,11 +120,11 @@ def render_health(health: Dict) -> str:
     return "\n".join(lines)
 
 
-# -----------------------------
-# Sync Wrapper (HF Safe)
-# -----------------------------
+# ======================================================
+# HF-SAFE ASYNC SEARCH HANDLER
+# ======================================================
 
-def run_search(
+async def run_search(
     query: str,
     hide_stubs: bool,
     enable_extended: bool,
@@ -133,10 +140,7 @@ def run_search(
             [a for a in enabled_extended_agencies if a in EXTENDED_AGENCIES]
         )
 
-    loop = asyncio.get_event_loop()
-    results, health = loop.run_until_complete(
-        federated_search_async(query, agencies)
-    )
+    results, health = await federated_search_async(query, agencies)
 
     saved_searches.save(query, enable_extended)
 
@@ -150,28 +154,28 @@ def run_search(
     return rendered, render_health(health), coverage.render_coverage()
 
 
-# -----------------------------
+# ======================================================
 # Gradio UI
-# -----------------------------
+# ======================================================
 
-with gr.Blocks(title="Federated FOIA Search") as demo:
+with gr.Blocks(title="Federated FOIA Document Search") as demo:
     gr.Markdown(
         """
 # 🏛️ Federated FOIA Document Search
-Search **public government FOIA reading rooms** across multiple agencies.
 
-- Default: **Safe, public-only sources**
-- Extended mode: **Additional public releases (opt-in)**
-- No authentication
-- No classified or restricted systems
+Search **public government FOIA electronic reading rooms** across multiple agencies.
+
+- **Default:** Safe, public-only sources  
+- **Extended mode:** Additional public releases (opt-in)  
+- No authentication  
+- No classified or restricted systems  
 """
     )
 
-    with gr.Row():
-        query = gr.Textbox(
-            label="Search term",
-            placeholder="e.g. UAP radar incident",
-        )
+    query = gr.Textbox(
+        label="Search term",
+        placeholder="e.g. UAP, radar incident, AATIP",
+    )
 
     with gr.Row():
         hide_stubs = gr.Checkbox(
@@ -219,7 +223,7 @@ Search **public government FOIA reading rooms** across multiple agencies.
 **Legal Notice**
 
 This tool:
-- Accesses **only publicly available FOIA Electronic Reading Rooms**
+- Accesses **only publicly available FOIA electronic reading rooms**
 - Respects robots.txt and rate limits
 - Does not bypass access controls
 - Is intended for journalism, research, and public-interest use
@@ -227,9 +231,9 @@ This tool:
         )
 
 
-# -----------------------------
-# HF Entry Point
-# -----------------------------
+# ======================================================
+# HF ENTRY POINT
+# ======================================================
 
 if __name__ == "__main__":
     demo.launch()