#22 added the graph endpoint (#32)

EconLabs · Feb 18, 2025 · e7a4fed · e7a4fed
1 parent d1f25b6
commit e7a4fed
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/routers/graphs.py b/src/routers/graphs.py
@@ -1,8 +1,12 @@
 from fastapi import APIRouter
+from ..submodules.jp_qcew.src.visualization.graph import graphGenerator
 
 router = APIRouter()
 
 
 @router.get("/graph/tmp")
-async def get_trade_file():
-    return {"This has not been implemented yet"}
+async def get_trade_file(
+    naics_code : str
+):
+    graph = graphGenerator().create_graph(naics_code)
+    return graph.to_json()