Create app.py

app.py

@@ -1,64 +1,46 @@
-# Federal FOIA Intelligence Search
-# Public Electronic Reading Rooms Only
-
-import gradio as gr
-import asyncio
+import gradio as gr, asyncio, plotly.express as px
 from ingest.registry import get_all_adapters
-from ingest.cluster import semantic_cluster
-from ingest.export import export_zip
-from ingest.foia_request import generate_foia_packet
-from ingest.health import adapter_health
+from ingest.cluster import semantic_cluster_plot
+from ingest.coverage import agency_coverage
+from ingest.citations import bluebook_pdf
 
 adapters = get_all_adapters()
 
-async def run_search(query, live_only):
+async def run_search(query):
     results = []
-    for adapter in adapters:
-        if live_only and not adapter.is_live:
-            continue
+    for a in adapters:
         try:
-            r = await adapter.search(query)
-            results.extend(r)
+            results.extend(await a.search(query))
         except Exception:
-            adapter.health = "down"
+            pass
     return results
 
-def search_ui(query, live_only):
-    results = asyncio.run(run_search(query, live_only))
-    clusters = semantic_cluster(results)
-    return results, clusters
+def search_ui(query):
+    results = asyncio.run(run_search(query))
+    coverage = agency_coverage(results)
+    heatmap = px.imshow(
+        [list(coverage.values())],
+        labels=dict(x="Agency", color="Docs"),
+        x=list(coverage.keys())
+    )
+    cluster_fig = semantic_cluster_plot(results)
+    return results, heatmap, cluster_fig
 
 with gr.Blocks() as demo:
     gr.Markdown("## Federal FOIA Intelligence Search")
     gr.Markdown("*Public Electronic Reading Rooms Only*")
 
-    query = gr.Textbox(label="Search FOIA Reading Rooms")
-    live_toggle = gr.Checkbox(label="Enable Live Agencies (Public Only)")
+    q = gr.Textbox(label="Search term")
+    btn = gr.Button("Search")
 
-    search_btn = gr.Button("Search")
-
-    results_table = gr.Dataframe(label="Results")
+    results_df = gr.Dataframe(label="Results")
+    heatmap_plot = gr.Plot(label="Agency Coverage")
     cluster_plot = gr.Plot(label="Semantic Clusters")
 
-    with gr.Tab("📄 Document Preview"):
-        preview = gr.HTML()
-
-    with gr.Tab("🧾 FOIA Request Generator"):
-        agency = gr.Textbox(label="Agency")
-        subject = gr.Textbox(label="Subject")
-        packet_btn = gr.Button("Generate FOIA Packet")
-        packet_out = gr.File()
+    pdf_btn = gr.Button("Export Bluebook PDF")
+    pdf_out = gr.File()
 
-    search_btn.click(
-        search_ui,
-        inputs=[query, live_toggle],
-        outputs=[results_table, cluster_plot]
-    )
-
-    packet_btn.click(
-        generate_foia_packet,
-        inputs=[agency, subject],
-        outputs=packet_out
-    )
+    btn.click(search_ui, q, [results_df, heatmap_plot, cluster_plot])
+    pdf_btn.click(bluebook_pdf, results_df, pdf_out)
 
 demo.launch()