diff --git a/web/scripts/app.js b/web/scripts/app.js index 609cc4cf..cd7fb5d1 100644 --- a/web/scripts/app.js +++ b/web/scripts/app.js @@ -721,6 +721,8 @@ class ComfyApp { * @param {*} graphData A serialized graph object */ loadGraphData(graphData) { + this.clean(); + if (!graphData) { graphData = defaultGraph; } @@ -903,6 +905,13 @@ class ComfyApp { } } } + + /** + * Clean current state + */ + clean() { + this.nodeOutputs = {}; + } } export const app = new ComfyApp(); diff --git a/web/scripts/ui.js b/web/scripts/ui.js index c79caaa9..2aabd29e 100644 --- a/web/scripts/ui.js +++ b/web/scripts/ui.js @@ -388,7 +388,10 @@ export class ComfyUI { }), $el("button", { textContent: "Load", onclick: () => fileInput.click() }), $el("button", { textContent: "Refresh", onclick: () => app.refreshComboInNodes() }), - $el("button", { textContent: "Clear", onclick: () => app.graph.clear() }), + $el("button", { textContent: "Clear", onclick: () => { + app.clean(); + app.graph.clear(); + }}), $el("button", { textContent: "Load Default", onclick: () => app.loadGraphData() }), ]);