From 27c5bc1581afe6a1a75b6d56a11aa22c1c0dcf93 Mon Sep 17 00:00:00 2001
From: bymyself <abolkonsky.rem@gmail.com>
Date: Fri, 5 Jul 2024 10:28:39 -0700
Subject: [PATCH] Update state when graph cleared via UI (#88)

---
 src/scripts/changeTracker.js | 4 ++++
 src/scripts/ui.ts            | 1 +
 src/scripts/ui/menu/index.js | 2 ++
 3 files changed, 7 insertions(+)

diff --git a/src/scripts/changeTracker.js b/src/scripts/changeTracker.js
index adcb8421fa..007ffd5d4e 100644
--- a/src/scripts/changeTracker.js
+++ b/src/scripts/changeTracker.js
@@ -160,6 +160,10 @@ export class ChangeTracker {
       changeTracker().checkState();
     });
 
+    api.addEventListener("graphCleared", () => {
+      changeTracker().checkState();
+    });
+
     // Handle litegraph clicks
     // @ts-ignore
     const processMouseUp = LGraphCanvas.prototype.processMouseUp;
diff --git a/src/scripts/ui.ts b/src/scripts/ui.ts
index 3a5292af76..0893aff52b 100644
--- a/src/scripts/ui.ts
+++ b/src/scripts/ui.ts
@@ -706,6 +706,7 @@ export class ComfyUI {
             app.clean();
             app.graph.clear();
             app.resetView();
+            api.dispatchEvent(new CustomEvent("graphCleared"));
           }
         },
       }),
diff --git a/src/scripts/ui/menu/index.js b/src/scripts/ui/menu/index.js
index e708d08847..1171da67dc 100644
--- a/src/scripts/ui/menu/index.js
+++ b/src/scripts/ui/menu/index.js
@@ -1,5 +1,6 @@
 // @ts-nocheck
 
+import { api } from "../../api";
 import { $el } from "../../ui";
 import { downloadBlob } from "../../utils";
 import { ComfyButton } from "../components/button";
@@ -112,6 +113,7 @@ export class ComfyAppMenu {
           ) {
             app.clean();
             app.graph.clear();
+            api.dispatchEvent(new CustomEvent("graphCleared"));
           }
         },
       })