From 4f63ee99f185db316a017520570310d094efffba Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 10 May 2024 17:30:52 -0400
Subject: [PATCH] Add a button to reset the view.

---
 web/scripts/app.js | 6 ++++++
 web/scripts/ui.js  | 7 +++++++
 2 files changed, 13 insertions(+)

diff --git a/web/scripts/app.js b/web/scripts/app.js
index a0fe8b29..7ed262cb 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -2313,6 +2313,12 @@ export class ComfyApp {
 		await this.#invokeExtensionsAsync("refreshComboInNodes", defs);
 	}
 
+	resetView() {
+		app.canvas.ds.scale = 1;
+		app.canvas.ds.offset = [0, 0]
+		app.graph.setDirtyCanvas(true, true);
+	}
+
 	/**
 	 * Clean current state
 	 */
diff --git a/web/scripts/ui.js b/web/scripts/ui.js
index d0fa46ef..36fed323 100644
--- a/web/scripts/ui.js
+++ b/web/scripts/ui.js
@@ -597,16 +597,23 @@ export class ComfyUI {
 					if (!confirmClear.value || confirm("Clear workflow?")) {
 						app.clean();
 						app.graph.clear();
+						app.resetView();
 					}
 				}
 			}),
 			$el("button", {
 				id: "comfy-load-default-button", textContent: "Load Default", onclick: async () => {
 					if (!confirmClear.value || confirm("Load default workflow?")) {
+						app.resetView();
 						await app.loadGraphData()
 					}
 				}
 			}),
+			$el("button", {
+				id: "comfy-reset-view-button", textContent: "Reset View", onclick: async () => {
+					app.resetView();
+				}
+			}),
 		]);
 
 		const devMode = this.settings.addSetting({