Create ingest/coverage.py

ingest/coverage.py

@@ -1,29 +1,5 @@
-import plotly.graph_objects as go
+from collections import Counter
 
-AGENCY_STATE_MAP = {
-    "CIA FOIA Reading Room": "VA",
-}
-
-def build_coverage_heatmap(counts):
-    return counts
-
-
-def build_agency_map(counts):
-    states = []
-    values = []
-
-    for agency, count in counts.items():
-        if agency in AGENCY_STATE_MAP:
-            states.append(AGENCY_STATE_MAP[agency])
-            values.append(count)
-
-    fig = go.Figure(
-        go.Choropleth(
-            locations=states,
-            locationmode="USA-states",
-            z=values,
-            colorscale="Blues",
-        )
-    )
-    fig.update_layout(title="Agency FOIA Coverage (US)")
-    return fig
+def agency_coverage(results):
+    counts = Counter(r.get("agency", "Unknown") for r in results)
+    return dict(counts)