From 070b5749b3233adab07db6e63a9373be6e371d52 Mon Sep 17 00:00:00 2001
From: Jairo Correa <jn.j41r0@gmail.com>
Date: Sat, 25 Mar 2023 06:28:09 -0300
Subject: [PATCH] Slot colors by type

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

diff --git a/web/scripts/app.js b/web/scripts/app.js
index a9acdce4..7a926865 100644
--- a/web/scripts/app.js
+++ b/web/scripts/app.js
@@ -486,6 +486,27 @@ class ComfyApp {
 		}
 	}
 
+	/**
+	 * Setup slot colors for types
+	 */
+	setupSlotColors() {
+		let colors = {
+			"CLIP": "#556B2F", // Olive green
+			"CLIP_VISION": "#4B0082", // Dark purple
+			"CLIP_VISION_OUTPUT": "#006400", // Dark green
+			"CONDITIONING": "#FF1493", // Deep pink
+			"CONTROL_NET": "#8B4513", // Dark brown
+			"IMAGE": "#8B0000", // Dark red
+			"LATENT": "#00008B", // Dark blue
+			"MASK": "#2F4F4F", // Dark gray
+			"MODEL": "#FF8C00", // Dark orange
+			"STYLE_MODEL": "#004A4A", // Dark teal
+			"VAE": "#4F394F", // Dark plum
+		};
+
+		Object.assign(this.canvas.default_connection_color_byType, colors);
+	}
+
 	/**
 	 * Set up the app on the page
 	 */
@@ -501,6 +522,8 @@ class ComfyApp {
 		const canvas = (this.canvas = new LGraphCanvas(canvasEl, this.graph));
 		this.ctx = canvasEl.getContext("2d");
 
+		this.setupSlotColors();
+
 		this.graph.start();
 
 		function resizeCanvas() {