Update entity_graph.py

entity_graph.py

@@ -1,48 +1,11 @@
-import json
-
-KEYWORDS = ["TENCAP", "AATIP", "NRO", "DIA", "NSA"]
-
-def extract_entities(text):
-    return [k for k in KEYWORDS if k.lower() in text.lower()]
-
-def build_entity_graph(documents):
-    nodes = []
-    links = []
-    seen = set()
-
-    for d in documents:
-        agency = d["agency"]
-        if agency not in seen:
-            nodes.append({"id": agency, "label": agency, "type": "agency"})
-            seen.add(agency)
-
-        for ent in extract_entities(d["text"]):
-            if ent not in seen:
-                nodes.append({"id": ent, "label": ent, "type": "entity"})
-                seen.add(ent)
-            links.append({"source": agency, "target": ent})
-
-    return {"nodes": nodes, "links": links}
-
-def render_d3(graph):
-    return f"""
-<div id="graph"></div>
-<script src="https://d3js.org/d3.v7.min.js"></script>
-<script>
-const data = {json.dumps(graph)};
-const w=600,h=400;
-const svg=d3.select("#graph").append("svg").attr("width",w).attr("height",h);
-const sim=d3.forceSimulation(data.nodes)
- .force("link",d3.forceLink(data.links).id(d=>d.id))
- .force("charge",d3.forceManyBody().strength(-200))
- .force("center",d3.forceCenter(w/2,h/2));
-const link=svg.selectAll("line").data(data.links).enter().append("line").attr("stroke","#999");
-const node=svg.selectAll("circle").data(data.nodes).enter().append("circle")
- .attr("r",6).attr("fill","#4f46e5");
-sim.on("tick",()=>{
- link.attr("x1",d=>d.source.x).attr("y1",d=>d.source.y)
-     .attr("x2",d=>d.target.x).attr("y2",d=>d.target.y);
- node.attr("cx",d=>d.x).attr("cy",d=>d.y);
-});
-</script>
-"""
+FOIA_REQUESTS = []
+
+def add_foia_request(agency, subject, status):
+    req = {
+        "id": len(FOIA_REQUESTS),
+        "agency": agency,
+        "subject": subject,
+        "status": status
+    }
+    FOIA_REQUESTS.append(req)
+    return FOIA_REQUESTS