Add error boundary on node.DrawBackground (#311)

Comfy-Org · Aug 5, 2024 · 50c4c87 · 50c4c87
1 parent 1b96c16
commit 50c4c87
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/scripts/app.ts b/src/scripts/app.ts
@@ -657,7 +657,7 @@ export class ComfyApp {
       }
     }
 
-    node.prototype.onDrawBackground = function (ctx) {
+    function unsafeDrawBackground(ctx) {
       if (!this.flags.collapsed) {
         let imgURLs = []
         let imagesChanged = false
@@ -992,6 +992,14 @@ export class ComfyApp {
         }
       }
     }
+
+    node.prototype.onDrawBackground = function (ctx) {
+      try {
+        unsafeDrawBackground.call(this, ctx)
+      } catch (error) {
+        console.error('Error drawing node background', error)
+      }
+    }
   }
 
   /**