From 5129cfa5a788b36b50fc0f4c61fcc6c6041d150f 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 d8cb5642c..2006d88a5 100644
--- a/src/scripts/app.ts
+++ b/src/scripts/app.ts
@@ -718,7 +718,7 @@ export class ComfyApp {
   }
 
   #addAfterConfigureHandler() {
-    const { canvas, graph } = this
+    const { graph } = this
     const { onConfigure } = graph
     graph.onConfigure = function (...args) {
       fixLinkInputSlots(this)
@@ -735,8 +735,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
     }