Update app.py

app.py

@@ -1,51 +1,96 @@
-import gradio as gr
 import asyncio
+import gradio as gr
+from typing import List, Dict
+
 from ingest.cia_reading_room import CIAAdapter
-from ingest.fbi_vault_live import FBIAdapter
-from ingest.dod_reading_room_live import DoDAdapter
-from core.async_search import fanout_search
-from core.cache import dedupe
-from core.cluster import cluster_results
-from core.citations import citation_block
-from core.redaction import redaction_confidence
-from core.journalist import journalist_export
-from core.explain import explain
-
-cia, fbi, dod = CIAAdapter(), FBIAdapter(), DoDAdapter()
-
-async def run(q):
-    res = await fanout_search([cia,fbi,dod], q)
-    return dedupe(res)
+from ingest.fbi_vault import FBIAdapter
+
+# Register adapters here
+ADAPTERS = [
+    CIAAdapter(),        # stub (safe)
+    FBIAdapter(),        # live
+]
+
+# -------------------------
+# Async search core
+# -------------------------
+async def federated_search_async(query: str) -> List[Dict]:
+    tasks = [adapter.search(query) for adapter in ADAPTERS]
+    results = await asyncio.gather(*tasks, return_exceptions=True)
+
+    flat: List[Dict] = []
+    for r in results:
+        if isinstance(r, Exception):
+            continue
+        flat.extend(r)
+
+    return flat
+
+
+# -------------------------
+# Sync wrapper (HF-safe)
+# -------------------------
+def federated_search(query: str) -> List[Dict]:
+    loop = asyncio.get_event_loop()
+    if loop.is_running():
+        # HF / Gradio case
+        return asyncio.create_task(federated_search_async(query))
+    else:
+        return loop.run_until_complete(federated_search_async(query))
+
+
+# -------------------------
+# UI helpers
+# -------------------------
+def format_result(r: Dict) -> str:
+    badge = "🟢 LIVE" if r.get("live") else "⚪ STUB"
+    return f"""
+### {r['title']}
+**{badge} · {r['source']}**
 
+{r['snippet']}
+
+🔗 {r['url']}
+"""
+
+
+def run_search(query: str, hide_stubs: bool):
+    results = federated_search(query)
+
+    # If async task returned
+    if asyncio.isfuture(results):
+        results = asyncio.get_event_loop().run_until_complete(results)
+
+    if hide_stubs:
+        results = [r for r in results if r.get("live")]
+
+    if not results:
+        return "No results found."
+
+    return "\n\n---\n\n".join(format_result(r) for r in results)
+
+
+# -------------------------
+# Gradio UI
+# -------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("# FOIA Federated Search — Supreme")
-
-    q = gr.Textbox(label="Query")
-    results_state = gr.State([])
-
-    with gr.Tabs():
-        with gr.Tab("Clustered Results"):
-            clusters = gr.JSON()
-        with gr.Tab("Citations"):
-            cites = gr.Markdown()
-        with gr.Tab("Explainability"):
-            explain_box = gr.JSON()
-
-    preview = gr.JSON(label="Redaction Confidence")
-
-    def _run(q):
-        res = asyncio.run(run(q))
-        cl = cluster_results(res)
-        cites_md = "\n".join(citation_block(r) for r in res[:5])
-        explain_data = explain(res)
-        red = {r.get("url"): redaction_confidence(r) for r in res}
-        return res, cl, cites_md, explain_data, red
-
-    btn = gr.Button("Search")
-    btn.click(_run, inputs=q, outputs=[results_state, clusters, cites, explain_box, preview])
-
-    exp = gr.Button("Journalist Export")
-    out = gr.File()
-    exp.click(lambda r: journalist_export(r, "/tmp/journalist_export.zip"), inputs=results_state, outputs=out)
+    gr.Markdown("# 📜 FOIA Federated Search")
+    gr.Markdown(
+        "Search across public government FOIA reading rooms. "
+        "Stub sources are clearly labeled."
+    )
+
+    query = gr.Textbox(label="Search term", placeholder="e.g. UAP radar")
+    hide_stubs = gr.Checkbox(
+        label="Hide stub (placeholder) sources",
+        value=False
+    )
+    output = gr.Markdown()
+
+    gr.Button("Search").click(
+        run_search,
+        inputs=[query, hide_stubs],
+        outputs=output
+    )
 
 demo.launch()