From 49c42476e9d77c6a730e5e4bd3e0b2c47d128e56 Mon Sep 17 00:00:00 2001
From: Jedrzej Kosinski <kosinkadink1@gmail.com>
Date: Mon, 20 Oct 2025 13:18:19 -0700
Subject: [PATCH] Fix path and actually return real data

---
 app/subgraph_manager.py | 31 +++++++++++++++++++++----------
 server.py               |  2 +-
 2 files changed, 22 insertions(+), 11 deletions(-)

diff --git a/app/subgraph_manager.py b/app/subgraph_manager.py
index a8a0d82f2..b314936b3 100644
--- a/app/subgraph_manager.py
+++ b/app/subgraph_manager.py
@@ -29,6 +29,7 @@ class SubgraphEntry(TypedDict):
     """
     Additional info about subgraph; in the case of custom_nodes, will contain nodepack name
     """
+    data: str
 
 class CustomNodeSubgraphEntryInfo(TypedDict):
     node_pack: str
@@ -38,39 +39,49 @@ class SubgraphManager:
     def __init__(self):
         self.cached_custom_node_subgraphs: dict[SubgraphEntry] | None = None
 
-    async def get_custom_node_subgraphs(self, force_reload=False):
+    async def load_entry_data(self, entry: SubgraphEntry):
+        with open(entry['path'], 'r') as f:
+            entry['data'] = f.read()
+        return entry
+
+    async def get_custom_node_subgraphs(self, loadedModules, force_reload=False):
         # if not forced to reload and cached, return cache
         if not force_reload and self.cached_custom_node_subgraphs is not None:
             return self.cached_custom_node_subgraphs
         # Load subgraphs from custom nodes
         subfolder = "subgraphs"
         subgraphs_dict: dict[SubgraphEntry] = {}
-        
+
         for folder in folder_paths.get_folder_paths("custom_nodes"):
             pattern = os.path.join(folder, f"*/{subfolder}/*.json")
             matched_files = glob.glob(pattern)
             for file in matched_files:
+                # replace backslashes with forward slashes
+                file = file.replace('\\', '/')
                 info: CustomNodeSubgraphEntryInfo = {
-                    "node_pack": folder
+                    "node_pack": "custom_nodes." + file.split('/')[-3]
                 }
-                path = f"{folder}/{file.split(folder)[-1]}"
                 source = Source.custom_node
                 # hash source + path to make sure id will be as unique as possible, but
                 # reproducible across backend reloads
-                id = hashlib.sha256(f"{source}{path}".encode()).hexdigest()
+                id = hashlib.sha256(f"{source}{file}".encode()).hexdigest()
                 entry: SubgraphEntry = {
                     "source": Source.custom_node,
                     "name": os.path.splitext(os.path.basename(file))[0],
-                    "path": path,
-                    "info": info
+                    "path": file,
+                    "info": info,
                 }
+                await self.load_entry_data(entry)
                 subgraphs_dict[id] = entry
         self.cached_custom_node_subgraphs = subgraphs_dict
         return subgraphs_dict
-    
+
     async def get_custom_node_subgraph(self, id: str):
         subgraphs = await self.get_custom_node_subgraphs()
-        return subgraphs.get(id, None)
+        entry: SubgraphEntry = subgraphs.get(id, None)
+        if entry is not None:
+            await self.load_entry_data(entry)
+        return entry
 
     def add_routes(self, routes, loadedModules):
         @routes.get("/global_subgraphs")
@@ -79,7 +90,7 @@ class SubgraphManager:
             # NOTE: we may want to include other sources of global subgraphs such as templates in the future;
             # that's the reasoning for the current implementation
             return web.json_response(subgraphs_dict)
-    
+
         @routes.get("/global_subgraphs/{id}")
         async def get_global_subgraph(request):
             id = request.match_info.get("id", None)
diff --git a/server.py b/server.py
index d0871d2a2..aebc4b01c 100644
--- a/server.py
+++ b/server.py
@@ -821,7 +821,7 @@ class PromptServer():
         self.user_manager.add_routes(self.routes)
         self.model_file_manager.add_routes(self.routes)
         self.custom_node_manager.add_routes(self.routes, self.app, nodes.LOADED_MODULE_DIRS.items())
-        self.subgraph_manager.add_routes(self.routes)
+        self.subgraph_manager.add_routes(self.routes, nodes.LOADED_MODULE_DIRS.items())
         self.app.add_subapp('/internal', self.internal_routes.get_app())
 
         # Prefix every route with /api for easier matching for delegation.