From 40a377775e7e383c09418297637a0cc261ead96d Mon Sep 17 00:00:00 2001
From: Farid Safi <FaridSafi@noreply.github.com>
Date: Tue, 28 Mar 2023 20:22:49 +0200
Subject: [PATCH] move clean to handleFile and loadGraphData functions

---
 web/scripts/app.js | 2 ++
 web/scripts/ui.js  | 6 +-----
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/web/scripts/app.js b/web/scripts/app.js
index ddb829ab..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;
 		}
diff --git a/web/scripts/ui.js b/web/scripts/ui.js
index 7e73c108..2aabd29e 100644
--- a/web/scripts/ui.js
+++ b/web/scripts/ui.js
@@ -306,7 +306,6 @@ export class ComfyUI {
 			style: { display: "none" },
 			parent: document.body,
 			onchange: () => {
-				app.clean();
 				app.handleFile(fileInput.files[0]);
 			},
 		});
@@ -393,10 +392,7 @@ export class ComfyUI {
 				app.clean();
 				app.graph.clear();
 			}}),
-			$el("button", { textContent: "Load Default", onclick: () => {
-				app.clean();
-				app.loadGraphData();
-			}}),
+			$el("button", { textContent: "Load Default", onclick: () => app.loadGraphData() }),
 		]);
 
 		dragElement(this.menuContainer);