Create app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from typing import List, Dict
+
 from ingest.registry import get_all_adapters
 from ingest.export import export_results
 from ingest.health import get_adapter_health
@@ -11,21 +12,32 @@ def run_search(
     query: str,
     include_stubs: bool,
     enable_extended: bool,
-    acknowledge: bool
+    acknowledge_extended: bool,
 ) -> List[Dict]:
-    log_event("search", {"query_len": len(query)})
-    results = []
+    log_event("search", {"query_length": len(query or "")})
+
+    if not query:
+        return []
+
+    results: List[Dict] = []
 
     for adapter in ALL_ADAPTERS:
+        # Skip stub adapters if not enabled
         if not include_stubs and not adapter.is_live:
             continue
-        if adapter.is_extended and not enable_extended:
-            continue
-        if adapter.is_extended and not acknowledge:
-            continue
+
+        # Extended adapters require explicit enable + acknowledgement
+        if adapter.is_extended:
+            if not enable_extended or not acknowledge_extended:
+                continue
 
         try:
-            results.extend(adapter.search(query))
+            adapter_results = adapter.search(query)
+            for r in adapter_results:
+                r.setdefault("agency", adapter.name)
+                r.setdefault("exportable", adapter.is_live)
+                r.setdefault("status", "🟢 Live" if adapter.is_live else "🔒 Stub")
+                results.append(r)
         except Exception as e:
             results.append({
                 "agency": adapter.name,
@@ -33,68 +45,109 @@ def run_search(
                 "snippet": str(e),
                 "url": "",
                 "status": "⚠️ Error",
-                "exportable": False
+                "exportable": False,
             })
 
     return results
 
-def handle_export(results: List[Dict]):
+
+def table_from_results(results: List[Dict]):
+    rows = []
+    for r in results:
+        rows.append([
+            r.get("agency", ""),
+            r.get("status", ""),
+            r.get("title", ""),
+            r.get("snippet", ""),
+            r.get("url", ""),
+        ])
+    return rows
+
+
+def export_handler(results: List[Dict]):
     if not results:
         return gr.File.update(visible=False)
 
     exportable = [r for r in results if r.get("exportable")]
-    return gr.File.update(
-        visible=True,
-        value=export_results(exportable if exportable else results)
-    )
+
+    if not exportable:
+        return gr.File.update(visible=False)
+
+    path = export_results(exportable)
+    return gr.File.update(value=path, visible=True)
+
 
 with gr.Blocks() as app:
-    gr.Markdown("## 🏛️ FOIA Document Search (Live + Public Sources Only)")
+    gr.Markdown(
+        "# 🏛️ Federal FOIA Intelligence Search\n"
+        "### Public Electronic Reading Rooms Only"
+    )
+
     gr.Markdown(
         "ℹ️ **Stub results are informational and cannot be exported.**\n\n"
-        "Extended sources require explicit consent before searching."
+        "Extended sources require explicit user consent before searching."
     )
 
-    query = gr.Textbox(label="Search query")
-    include_stubs = gr.Checkbox(label="Include Stub Results", value=True)
-    enable_extended = gr.Checkbox(label="Enable Extended Sources")
-    acknowledge = gr.Checkbox(
+    with gr.Row():
+        query = gr.Textbox(label="Search query", placeholder="e.g. MKULTRA")
+
+    with gr.Row():
+        include_stubs = gr.Checkbox(
+            label="Include Stub Results",
+            value=True
+        )
+        enable_extended = gr.Checkbox(
+            label="Enable Extended Coverage (opt-in)",
+            value=False
+        )
+
+    acknowledge_extended = gr.Checkbox(
         label="I understand some agencies block automated access",
-        visible=True
+        value=False
     )
 
     search_btn = gr.Button("Search")
-    export_btn = gr.Button("Export Results (ZIP)", interactive=False)
 
     results_state = gr.State([])
     results_table = gr.Dataframe(
-        headers=["agency", "status", "title", "snippet", "url"],
+        headers=["Agency", "Status", "Title", "Snippet", "URL"],
+        interactive=False,
+        wrap=True
+    )
+
+    export_btn = gr.Button(
+        "Export Results (ZIP)",
         interactive=False
     )
     export_file = gr.File(visible=False)
 
     search_btn.click(
         fn=run_search,
-        inputs=[query, include_stubs, enable_extended, acknowledge],
-        outputs=[results_state]
+        inputs=[
+            query,
+            include_stubs,
+            enable_extended,
+            acknowledge_extended,
+        ],
+        outputs=results_state,
     ).then(
         fn=lambda r: (
-            r,
+            table_from_results(r),
             gr.Button.update(
                 interactive=any(x.get("exportable") for x in r)
-            )
+            ),
         ),
-        inputs=[results_state],
-        outputs=[results_table, export_btn]
+        inputs=results_state,
+        outputs=[results_table, export_btn],
     )
 
     export_btn.click(
-        fn=handle_export,
-        inputs=[results_state],
-        outputs=[export_file]
+        fn=export_handler,
+        inputs=results_state,
+        outputs=export_file,
     )
 
-    gr.Markdown("### 🔍 Adapter Health")
+    gr.Markdown("## 🔍 Adapter Health")
     gr.JSON(get_adapter_health())
 
 app.launch()