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() {