Allow any node to draw images

web/scripts/app.js

@@ -123,8 +123,9 @@ class ComfyApp {
 							}
 						});
 					}
+				}
 
-					if (this.imgs) {
+				if (this.imgs && this.imgs.length) {
 					const canvas = graph.list_of_graphcanvas[0];
 					const mouse = canvas.graph_mouse;
 					if (!canvas.pointer_is_down && this.pointerDown) {
@@ -226,14 +227,7 @@ class ComfyApp {
 						ctx.drawImage(this.imgs[imageIndex], x, y, w, h);
 
 						const drawButton = (x, y, sz, text) => {
-								const hovered = LiteGraph.isInsideRectangle(
-									mouse[0],
-									mouse[1],
-									x + this.pos[0],
-									y + this.pos[1],
-									sz,
-									sz
-								);
+							const hovered = LiteGraph.isInsideRectangle(mouse[0], mouse[1], x + this.pos[0], y + this.pos[1], sz, sz);
 							let fill = "#333";
 							let textFill = "#fff";
 							let isClicking = false;
@@ -279,7 +273,6 @@ class ComfyApp {
 					}
 				}
 			}
-			}
 		};
 	}