Create ingest/coverage.py

ingest/coverage.py

@@ -1,9 +1,29 @@
-def build_coverage_heatmap(per_agency_counts):
-    # Simple numeric heatmap data
-    return {
-        agency: {
-            "count": count,
-            "intensity": min(count / 10.0, 1.0)
-        }
-        for agency, count in per_agency_counts.items()
-    }
+import plotly.graph_objects as go
+
+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