From 28220fa8392b6b2f0eabb0d0c8311ff3b07af69a Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Sat, 2 Dec 2023 12:02:03 +0000
Subject: [PATCH] Fix node growing with DOM widgets when adding image even if
 enough space

---
 web/scripts/app.js       | 2 +-
 web/scripts/domWidget.js | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/web/scripts/app.js b/web/scripts/app.js
index 861db16b..8598d447 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -411,7 +411,7 @@ export class ComfyApp {
 		node.prototype.setSizeForImage = function (force) {
 			if(!force && this.animatedImages) return;
 
-			if (this.inputHeight) {
+			if (this.inputHeight || this.freeWidgetSpace > 210) {
 				this.setSize(this.size);
 				return;
 			}
diff --git a/web/scripts/domWidget.js b/web/scripts/domWidget.js
index 37d26f3c..e919428a 100644
--- a/web/scripts/domWidget.js
+++ b/web/scripts/domWidget.js
@@ -120,6 +120,8 @@ function computeSize(size) {
 		freeSpace -= 220;
 	}
 
+	this.freeWidgetSpace = freeSpace;
+
 	if (freeSpace < 0) {
 		// Not enough space for all widgets so we need to grow
 		size[1] -= freeSpace;