Fix Comfy.WidgetControlMode watcher (#2079)

Comfy-Org · Dec 28, 2024 · af1a5d9 · af1a5d9
1 parent feabd3f
commit af1a5d9
Showing 1 changed file with 16 additions and 11 deletions.
diff --git a/src/components/graph/GraphCanvas.vue b/src/components/graph/GraphCanvas.vue
@@ -196,22 +196,27 @@ watchEffect(() => {
   LiteGraph.alwaysSnapToGrid = settingStore.get('pysssss.SnapToGrid')
 })
 
-watch(settingStore.get('Comfy.WidgetControlMode'), () => {
-  for (const n of comfyApp.graph.nodes) {
-    if (!n.widgets) continue
-    for (const w of n.widgets) {
-      if (w[IS_CONTROL_WIDGET]) {
-        updateControlWidgetLabel(w)
-        if (w.linkedWidgets) {
-          for (const l of w.linkedWidgets) {
-            updateControlWidgetLabel(l)
+watch(
+  () => settingStore.get('Comfy.WidgetControlMode'),
+  () => {
+    if (!canvasStore.canvas) return
+
+    for (const n of comfyApp.graph.nodes) {
+      if (!n.widgets) continue
+      for (const w of n.widgets) {
+        if (w[IS_CONTROL_WIDGET]) {
+          updateControlWidgetLabel(w)
+          if (w.linkedWidgets) {
+            for (const l of w.linkedWidgets) {
+              updateControlWidgetLabel(l)
+            }
           }
         }
       }
     }
+    comfyApp.graph.setDirtyCanvas(true)
   }
-  comfyApp.graph.setDirtyCanvas(true)
-})
+)
 
 watchEffect(() => {
   if (!canvasStore.canvas) return