From 019bd519cb12b0be8dc5d24bb14dc95712350729 Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Tue, 14 Mar 2023 21:21:50 +0000
Subject: [PATCH] Add dragleave handler to remove stuck highlight

---
 web/scripts/app.js | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/web/scripts/app.js b/web/scripts/app.js
index 80a4eb3..257d852 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -289,13 +289,21 @@ class ComfyApp {
 			this.dragOverNode = null;
 			// Node handles file drop, we dont use the built in onDropFile handler as its buggy
 			// If you drag multiple files it will call it multiple times with the same file
-			if (n && n.onDragDrop && await n.onDragDrop(event)) {
+			if (n && n.onDragDrop && (await n.onDragDrop(event))) {
 				return;
 			}
 
 			await this.handleFile(event.dataTransfer.files[0]);
 		});
 
+		// Always clear over node on drag leave
+		this.canvasEl.addEventListener("dragleave", async () => {
+			if (this.dragOverNode) {
+				this.dragOverNode = null;
+				this.graph.setDirtyCanvas(false, true);
+			}
+		});
+
 		// Add handler for dropping onto a specific node
 		this.canvasEl.addEventListener(
 			"dragover",