Create app.py

app.py

@@ -7,6 +7,8 @@ from ingest.health import get_adapter_health
 from ingest.coverage import coverage_summary
 from ingest.discovery import agency_discovery
 from ingest.semantic import semantic_refine, semantic_available
+from ingest.timeline import release_timeline
+from ingest.latency import latency_badges
 from analytics.events import log_event
 
 ALL_ADAPTERS = get_all_adapters()
@@ -19,12 +21,12 @@ def run_search(
     enable_semantic: bool,
 ) -> List[Dict]:
 
-    log_event("search", {"query_length": len(query or "")})
+    log_event("search", {"len": len(query or "")})
 
     if not query:
         return []
 
-    results: List[Dict] = []
+    results = []
 
     for adapter in ALL_ADAPTERS:
         if not include_stubs and not adapter.is_live:
@@ -35,12 +37,12 @@ def run_search(
                 continue
 
         try:
-            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)
+            docs = adapter.search(query)
+            for d in docs:
+                d.setdefault("agency", adapter.name)
+                d.setdefault("status", "🟢 Live" if adapter.is_live else "🔒 Stub")
+                d.setdefault("exportable", adapter.is_live)
+                results.append(d)
         except Exception as e:
             results.append({
                 "agency": adapter.name,
@@ -57,38 +59,35 @@ def run_search(
     return results
 
 
-def table_from_results(results: List[Dict]):
+def table_from_results(results):
     return [
         [
-            r.get("agency", ""),
-            r.get("status", ""),
-            r.get("title", ""),
-            r.get("snippet", ""),
-            r.get("url", ""),
+            r.get("agency"),
+            r.get("status"),
+            r.get("title"),
+            r.get("snippet"),
+            r.get("url"),
         ]
         for r in results
     ]
 
 
-def export_handler(results: List[Dict]):
+def export_handler(results):
     exportable = [r for r in results if r.get("exportable")]
     if not exportable:
         return gr.File.update(visible=False)
-    return gr.File.update(
-        value=export_results(exportable),
-        visible=True
-    )
+    return gr.File.update(value=export_results(exportable), visible=True)
 
 
 with gr.Blocks() as app:
     gr.Markdown(
         "# 🏛️ Federal FOIA Intelligence Search\n"
-        "### Public Electronic Reading Rooms Only"
+        "Public Electronic Reading Rooms only"
     )
 
     gr.Markdown(
-        "ℹ️ **Stub results are informational and cannot be exported.**\n\n"
-        "Semantic search is optional and operates only on returned results."
+        "ℹ️ Stub results are informational and cannot be exported.\n\n"
+        "Semantic refinement is optional and runs only on returned results."
     )
 
     query = gr.Textbox(label="Search query")
@@ -113,18 +112,32 @@ with gr.Blocks() as app:
 
     results_table = gr.Dataframe(
         headers=["Agency", "Status", "Title", "Snippet", "URL"],
-        interactive=False,
-        wrap=True
+        wrap=True,
+        interactive=False
     )
 
     export_btn = gr.Button("Export Results (ZIP)", interactive=False)
     export_file = gr.File(visible=False)
 
+    gr.Markdown("## 📊 Coverage Heatmap")
     coverage_table = gr.Dataframe(
         headers=["Agency", "Result Count"],
         interactive=False
     )
 
+    gr.Markdown("## 🕒 Release Timeline")
+    timeline_table = gr.Dataframe(
+        headers=["Period", "Documents"],
+        interactive=False
+    )
+
+    gr.Markdown("## ⚡ Agency Latency Badges")
+    latency_table = gr.Dataframe(
+        headers=["Agency", "Latency (s)", "Badge"],
+        interactive=False
+    )
+
+    gr.Markdown("## 🌐 Agency Discovery")
     discovery_table = gr.Dataframe(
         headers=["Agency", "Status", "Reason"],
         interactive=False
@@ -144,10 +157,16 @@ with gr.Blocks() as app:
         fn=lambda r: (
             table_from_results(r),
             coverage_summary(r),
+            release_timeline(r),
             gr.Button.update(interactive=any(x.get("exportable") for x in r))
         ),
         inputs=results_state,
-        outputs=[results_table, coverage_table, export_btn]
+        outputs=[
+            results_table,
+            coverage_table,
+            timeline_table,
+            export_btn
+        ]
     )
 
     export_btn.click(
@@ -156,7 +175,7 @@ with gr.Blocks() as app:
         outputs=export_file
     )
 
-    gr.Markdown("## 🌐 Agency Coverage")
+    latency_table.value = latency_badges(ALL_ADAPTERS)
     discovery_table.value = agency_discovery()
 
     gr.Markdown("## 🔍 Adapter Health")