From 669dbdbe56a85e47dbd0c50cb79e07b770ea1ca2 Mon Sep 17 00:00:00 2001
From: filtered <176114999+webfiltered@users.noreply.github.com>
Date: Mon, 12 May 2025 18:07:02 +1000
Subject: [PATCH] Prune DOM widgets outside of current subgraph

---
 src/scripts/app.ts | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/scripts/app.ts b/src/scripts/app.ts
index 1f864830b..05d497bd6 100644
--- a/src/scripts/app.ts
+++ b/src/scripts/app.ts
@@ -717,7 +717,7 @@ export class ComfyApp {
   }
 
   #addAfterConfigureHandler() {
-    const { canvas, graph } = this
+    const { graph } = this
     const { onConfigure } = graph
     graph.onConfigure = function (...args) {
       fixLinkInputSlots(this)
@@ -734,8 +734,10 @@ export class ComfyApp {
         node.onAfterGraphConfigured?.()
       }
 
-      const nodes = canvas.subgraph?.nodes ?? graph.nodes
-      pruneWidgets(nodes)
+      graph.canvasAction((c) => {
+        const nodes = c.subgraph?.nodes ?? graph.nodes
+        pruneWidgets(nodes)
+      })
 
       return r
     }