clean state when loading another workflow

This commit is contained in:
Farid Safi 2023-03-28 19:45:17 +02:00
parent 31dd6c0531
commit 393084877c
2 changed files with 16 additions and 2 deletions

View File

@ -903,6 +903,13 @@ class ComfyApp {
} }
} }
} }
/**
* Clean current state
*/
clean() {
this.nodeOutputs = {};
}
} }
export const app = new ComfyApp(); export const app = new ComfyApp();

View File

@ -306,6 +306,7 @@ export class ComfyUI {
style: { display: "none" }, style: { display: "none" },
parent: document.body, parent: document.body,
onchange: () => { onchange: () => {
app.clean();
app.handleFile(fileInput.files[0]); app.handleFile(fileInput.files[0]);
}, },
}); });
@ -388,8 +389,14 @@ export class ComfyUI {
}), }),
$el("button", { textContent: "Load", onclick: () => fileInput.click() }), $el("button", { textContent: "Load", onclick: () => fileInput.click() }),
$el("button", { textContent: "Refresh", onclick: () => app.refreshComboInNodes() }), $el("button", { textContent: "Refresh", onclick: () => app.refreshComboInNodes() }),
$el("button", { textContent: "Clear", onclick: () => app.graph.clear() }), $el("button", { textContent: "Clear", onclick: () => {
$el("button", { textContent: "Load Default", onclick: () => app.loadGraphData() }), app.clean();
app.graph.clear();
}}),
$el("button", { textContent: "Load Default", onclick: () => {
app.clean();
app.loadGraphData();
}}),
]); ]);
dragElement(this.menuContainer); dragElement(this.menuContainer);