Serve the graph data from a static file

Instead of fetching the data from the `/extract` endpoint (which expects a `graph.json` file to be present, and then adds extra data that it fetches from S3), use a new `/graph-viewmodel` endpoint, which simply serves the data from a static file.

Author: mtaylorgds

Dockerfile

@@ -17,6 +17,7 @@ RUN uv sync --no-dev --no-install-project
 COPY src/ ./src/
 COPY static/ ./static/
 COPY templates/ ./templates/
+COPY graph-viewmodel.json ./graph-viewmodel.json
 COPY app.py ./app.py
 
 # Set environment variables

app.py

@@ -2,7 +2,7 @@
 import logging
 from flask import Flask, request, jsonify, render_template
 from dotenv import load_dotenv
-from src.generate_graph import generate_graph
+from src.generate_graph import generate_graph, load_graph_viewmodel
 
 load_dotenv()
 
@@ -24,6 +24,18 @@ def graph_page():
         """Serve the Cytoscape graph viewer page."""
         return render_template('graph.html')
 
+    @app.route('/graph-viewmodel', methods=['GET'])
+    async def graph_viewmodel():
+        """Serve the graph data as JSON for the frontend."""
+        try:
+            logger.info('Loading graph data for viewmodel endpoint...')
+            graph_data = load_graph_viewmodel("graph-viewmodel.json")
+            logger.info('Graph data loaded successfully.')
+            return jsonify(graph_data), 200
+        except Exception as e:
+            app.logger.error(f"Error loading graph data: {str(e)}")
+            return jsonify({"error": str(e)}), 500
+
     @app.route('/healthcheck/ready', methods=['GET'])
     def health_check():
         """Simple health check endpoint."""

graph-viewmodel.json

@@ -150,5 +150,17 @@ async def generate_graph(input_data: Union[str, Dict[str, Any]], output_path: Op
 
     return cy_json
 
+def load_json_file(file_path: str) -> Dict[str, Any]:
+    """Utility function to load JSON data from a file."""
+    if not os.path.exists(file_path):
+        logger.error(f"File {file_path} not found.")
+        return {}
+    with open(file_path, "r") as f:
+        return json.load(f)
+
+def load_graph_viewmodel(file_path: str) -> Dict[str, Any]:
+    """Loads the graph viewmodel JSON for the frontend."""
+    return load_json_file(file_path)
+
 if __name__ == "__main__":
     asyncio.run(generate_graph("graph.json", "outputs/graphNode.json"))

src/generate_graph.py

@@ -58,7 +58,7 @@ <h2 id="sidebar-title">No node selected</h2>
 
       async function loadGraph() {
         try {
-          const response = await fetch('/extract');
+          const response = await fetch('/graph-viewmodel');
           if (!response.ok) {
             throw new Error(`Failed to load graph: ${response.status} ${response.statusText}`);
           }