Comfy-Org · huchenlei · Jul 6, 2024 · Jul 5, 2024 · Jul 5, 2024 · Jul 5, 2024
@@ -11,10 +11,10 @@ Front-end of [ComfyUI](https://github.com/comfyanonymous/ComfyUI) modernized. Th
 - Added a shim layer to be backward compatible with the existing extension system. https://github.com/huchenlei/ComfyUI_frontend/pull/15
 - Front-end dev server.
 - Zod schema for input validation on ComfyUI workflow.
+- Make litegraph a npm dependency.
 
 ### What to be done
 
-- Make litegraph a npm dependency.
 - Replace the existing ComfyUI front-end impl.
 - Turn on `strict` on `tsconfig.json`.
 - Introduce react to start managing part of the UI.

@@ -38,6 +38,8 @@ export class ComfyPage {
 
   async resetView() {
     await this.resetViewButton.click();
+    // Avoid "Reset View" button highlight.
+    await this.page.mouse.move(10, 10);
     await this.nextFrame();
   }
 

@@ -12,12 +12,6 @@
 				font-family: 'Roboto Mono', 'Noto Color Emoji';
 			}
 		</style> -->
-		<link rel="stylesheet" type="text/css" href="/lib/litegraph.css" />
-		<link rel="stylesheet" type="text/css" href="/style.css" />
-		<link rel="stylesheet" type="text/css" href="/user.css" />
-		<link rel="stylesheet" type="text/css" href="/materialdesignicons.min.css" />
-		<script type="text/javascript" src="/lib/litegraph.core.js"></script>
-		<script type="text/javascript" src="/lib/litegraph.extensions.js" defer></script>
 		<script type="module">
 			console.log("ComfyUI Front-end version:", __COMFYUI_FRONTEND_VERSION__);
 			import { app } from "./src/scripts/app";
@@ -27,6 +21,8 @@
 				window.graph = app.graph;
 			})();
 		</script>
+		<link rel="stylesheet" type="text/css" href="/user.css" />
+		<link rel="stylesheet" type="text/css" href="/materialdesignicons.min.css" />
 	</head>
 	<body class="litegraph">
 		<div id="comfy-user-selection" class="comfy-user-selection" style="display: none;">

@@ -37,6 +37,7 @@
     "zip-dir": "^2.0.0"
   },
   "dependencies": {
+    "@comfyorg/litegraph": "^0.7.20",
     "dotenv": "^16.4.5",
     "zod": "^3.23.8",
     "zod-validation-error": "^3.3.0"
@@ -47,4 +48,4 @@
       "git add"
     ]
   }
-}
+}
@@ -1,5 +1,3 @@
-@import url("scripts/ui/menu/menu.css");
-
 :root {
 	--fg-color: #000;
 	--bg-color: #fff;

@@ -1,6 +1,7 @@
 import { app } from "../../scripts/app";
 import { $el } from "../../scripts/ui";
 import type { ColorPalettes } from "/types/colorPalette";
+import { LGraphCanvas, LiteGraph } from "@comfyorg/litegraph";
 
 // Manage color palettes
 
@@ -427,6 +428,32 @@ const els: { select: HTMLSelectElement | null } = {
 // const ctxMenu = LiteGraph.ContextMenu;
 app.registerExtension({
   name: id,
+  init() {
+    /**
+     * Changes the background color of the canvas.
+     *
+     * @method updateBackground
+     * @param {image} String
+     * @param {clearBackgroundColor} String
+     */
+    // @ts-ignore
+    LGraphCanvas.prototype.updateBackground = function (
+      image,
+      clearBackgroundColor
+    ) {
+      this._bg_img = new Image();
+      this._bg_img.name = image;
+      this._bg_img.src = image;
+      this._bg_img.onload = () => {
+        this.draw(true, true);
+      };
+      this.background_image = image;
+
+      this.clear_background = true;
+      this.clear_background_color = clearBackgroundColor;
+      this._pattern = null;
+    };
+  },
   addCustomNodeDefs(node_defs) {
     const sortObjectKeys = (unordered) => {
       return Object.keys(unordered)

@@ -1,3 +1,4 @@
+import { LiteGraph, LGraphCanvas } from "@comfyorg/litegraph";
 import { app } from "../../scripts/app";
 
 // Adds filtering to combo context menus

@@ -2,7 +2,8 @@ import { app } from "../../scripts/app";
 import { api } from "../../scripts/api";
 import { mergeIfValid } from "./widgetInputs";
 import { ManageGroupDialog } from "./groupNodeManage";
-import type { LGraphNode } from "/types/litegraph";
+import type { LGraphNode } from "@comfyorg/litegraph";
+import { LGraphCanvas, LiteGraph } from "@comfyorg/litegraph";
 
 const GROUP = Symbol();
 

@@ -3,7 +3,11 @@ import { DraggableList } from "../../scripts/ui/draggableList";
 import { GroupNodeConfig, GroupNodeHandler } from "./groupNode";
 import "./groupNodeManage.css";
 import { app, type ComfyApp } from "../../scripts/app";
-import type { LGraphNode, LGraphNodeConstructor } from "/types/litegraph";
+import {
+  LiteGraph,
+  type LGraphNode,
+  type LGraphNodeConstructor,
+} from "@comfyorg/litegraph";
 
 const ORDER: symbol = Symbol();
 

@@ -1,4 +1,5 @@
 import { app } from "../../scripts/app";
+import { LGraphCanvas, LiteGraph } from "@comfyorg/litegraph";
 
 function setNodeMode(node, mode) {
   node.mode = mode;

@@ -1,3 +1,4 @@
+import { LiteGraph } from "@comfyorg/litegraph";
 import { app } from "../../scripts/app";
 
 // Inverts the scrolling of context menus

@@ -1,5 +1,5 @@
 import { app } from "../../scripts/app";
-
+import { LiteGraph } from "@comfyorg/litegraph";
 const id = "Comfy.LinkRenderMode";
 const ext = {
   name: id,

@@ -2,6 +2,7 @@ import { app } from "../../scripts/app";
 import { api } from "../../scripts/api";
 import { ComfyDialog, $el } from "../../scripts/ui";
 import { GroupNodeConfig, GroupNodeHandler } from "./groupNode";
+import { LGraphCanvas } from "@comfyorg/litegraph";
 
 // Adds the ability to save and add multiple nodes as a template
 // To save:

@@ -1,3 +1,4 @@
+import { LiteGraph, LGraphCanvas } from "@comfyorg/litegraph";
 import { app } from "../../scripts/app";
 import { ComfyWidgets } from "../../scripts/widgets";
 // Node that add notes to your project
@@ -8,11 +9,8 @@ app.registerExtension({
     class NoteNode {
       static category: string;
 
-      // @ts-ignore
       color = LGraphCanvas.node_colors.yellow.color;
-      // @ts-ignore
       bgcolor = LGraphCanvas.node_colors.yellow.bgcolor;
-      // @ts-ignore
       groupcolor = LGraphCanvas.node_colors.yellow.groupcolor;
       properties: { text: string };
       serialize_widgets: boolean;

@@ -1,5 +1,6 @@
 import { app } from "../../scripts/app";
 import { mergeIfValid, getWidgetConfig, setWidgetConfig } from "./widgetInputs";
+import { LiteGraph, LGraphCanvas } from "@comfyorg/litegraph";
 
 // Node that allows you to redirect connections for cleaner graphs
 

@@ -1,4 +1,5 @@
 import { app } from "../../scripts/app";
+import { LGraphCanvas, LiteGraph } from "@comfyorg/litegraph";
 
 let touchZooming;
 let touchCount = 0;

@@ -1,5 +1,6 @@
 import { app } from "../../scripts/app";
 import { ComfyWidgets } from "../../scripts/widgets";
+import { LiteGraph } from "@comfyorg/litegraph";
 // Adds defaults for quickly adding nodes with middle click on the input/output
 
 app.registerExtension({

@@ -1,4 +1,10 @@
 import { app } from "../../scripts/app";
+import {
+  LGraphCanvas,
+  LGraphNode,
+  LGraphGroup,
+  LiteGraph,
+} from "@comfyorg/litegraph";
 
 // Shift + drag/resize to snap to grid
 
@@ -77,12 +83,14 @@ app.registerExtension({
 
         let w, h;
         if (node.flags.collapsed) {
+          // @ts-ignore
           w = node._collapsed_width;
           h = LiteGraph.NODE_TITLE_HEIGHT;
           shiftY -= LiteGraph.NODE_TITLE_HEIGHT;
         } else {
           w = node.size[0];
           h = node.size[1];
+          // @ts-ignore
           let titleMode = node.constructor.title_mode;
           if (
             titleMode !== LiteGraph.TRANSPARENT_TITLE &&
@@ -105,7 +113,7 @@ app.registerExtension({
      * The currently moving, selected group only. Set after the `selected_group` has actually started
      * moving.
      */
-    let selectedAndMovingGroup = null;
+    let selectedAndMovingGroup: LGraphGroup | null = null;
 
     /**
      * Handles moving a group; tracking when a group has been moved (to show the ghost in `drawGroups`
@@ -154,13 +162,16 @@ app.registerExtension({
     LGraphCanvas.prototype.drawGroups = function (canvas, ctx) {
       if (this.selected_group && app.shiftDown) {
         if (this.selected_group_resizing) {
+          // @ts-ignore
           roundVectorToGrid(this.selected_group.size);
         } else if (selectedAndMovingGroup) {
+          // @ts-ignore
           const [x, y] = roundVectorToGrid([...selectedAndMovingGroup.pos]);
           const f = ctx.fillStyle;
           const s = ctx.strokeStyle;
           ctx.fillStyle = "rgba(100, 100, 100, 0.33)";
           ctx.strokeStyle = "rgba(100, 100, 100, 0.66)";
+          // @ts-ignore
           ctx.rect(x, y, ...selectedAndMovingGroup.size);
           ctx.fill();
           ctx.stroke();

@@ -1,6 +1,6 @@
 import { app } from "../../scripts/app";
 import { api } from "../../scripts/api";
-import type { IWidget } from "/types/litegraph";
+import type { IWidget } from "@comfyorg/litegraph";
 import type { DOMWidget } from "/scripts/domWidget";
 import { ComfyNodeDef } from "/types/apiTypes";
 

@@ -1,6 +1,7 @@
 import { ComfyWidgets, addValueControlWidgets } from "../../scripts/widgets";
 import { app } from "../../scripts/app";
 import { applyTextReplacements } from "../../scripts/utils";
+import { LiteGraph } from "@comfyorg/litegraph";
 
 const CONVERTED_TYPE = "converted-widget";
 const VALID_TYPES = ["STRING", "combo", "number", "BOOLEAN"];