From ab7d4f784892c275e888d71aa80a3a2ed59d9b83 Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Wed, 22 Nov 2023 13:53:30 +0000
Subject: [PATCH] Handle collapsing to hide element

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

diff --git a/web/scripts/domWidget.js b/web/scripts/domWidget.js
index 16f4e192e..0f8a2eb01 100644
--- a/web/scripts/domWidget.js
+++ b/web/scripts/domWidget.js
@@ -233,6 +233,7 @@ LGraphNode.prototype.addDOMWidget = function (name, type, element, options) {
 			}
 
 			const hidden =
+				node.flags?.collapsed ||
 				(!!options.hideOnZoom && app.canvas.ds.scale < 0.5) ||
 				widget.computedHeight <= 0 ||
 				widget.type === "converted-widget";
@@ -290,6 +291,15 @@ LGraphNode.prototype.addDOMWidget = function (name, type, element, options) {
 	this.addCustomWidget(widget);
 	elementWidgets.add(this);
 
+	const collapse = this.collapse;
+	this.collapse = function() {
+		collapse.apply(this, arguments);
+		if(this.flags?.collapsed) {
+			element.hidden = true;
+			element.style.display = "none";
+		}
+	}
+
 	const onRemoved = this.onRemoved;
 	this.onRemoved = function () {
 		element.remove();