Image output fixes

src/lib/nodes/ComfyActionNodes.ts

@@ -2,6 +2,7 @@ import { LiteGraph, type ContextMenuItem, type LGraphNode, type Vector2, LConnec
 import ComfyGraphNode from "./ComfyGraphNode";
 import { Watch } from "@litegraph-ts/nodes-basic";
 import type { SerializedPrompt } from "$lib/components/ComfyApp";
+import { toast } from '@zerodevx/svelte-toast'
 
 export interface ComfyAfterQueuedEventProperties extends Record<any, any> {
     prompt: SerializedPrompt
@@ -47,6 +48,40 @@ LiteGraph.registerNodeType({
     type: "actions/after_queued"
 })
 
+export interface ComfyOnExecutedEventProperties extends Record<any, any> {
+}
+
+export class ComfyOnExecutedEvent extends ComfyGraphNode {
+    override properties: ComfyOnExecutedEventProperties = {
+    }
+
+    static slotLayout: SlotLayout = {
+        inputs: [
+            { name: "images", type: "IMAGE" }
+        ],
+        outputs: [
+            { name: "onExecuted", type: BuiltInSlotType.EVENT },
+        ],
+    }
+
+    private _output: any = null;
+
+    override receiveOutput(output: any) {
+        if (this._output !== output) {
+            console.error(output)
+            this.triggerSlot(0, "bang")
+            this._output = output
+        }
+    }
+}
+
+LiteGraph.registerNodeType({
+    class: ComfyOnExecutedEvent,
+    title: "Comfy.OnExecutedEvent",
+    desc: "Triggers a 'bang' event when a prompt output is received.",
+    type: "actions/on_executed"
+})
+
 export interface ComfyCopyActionProperties extends Record<any, any> {
     value: any
 }
@@ -130,3 +165,34 @@ LiteGraph.registerNodeType({
     desc: "Swaps two inputs when triggered",
     type: "actions/swap"
 })
+
+export interface ComfyNotifyActionProperties extends Record<any, any> {
+    message: string
+}
+
+export class ComfyNotifyAction extends ComfyGraphNode {
+    override properties: ComfyNotifyActionProperties = {
+        message: "Nya."
+    }
+
+    static slotLayout: SlotLayout = {
+        inputs: [
+            { name: "message", type: "string" },
+            { name: "trigger", type: BuiltInSlotType.ACTION }
+        ],
+    }
+
+    override onAction(action: any, param: any) {
+        const message = this.getInputData(0);
+        if (message) {
+            toast.push(message);
+        }
+    };
+}
+
+LiteGraph.registerNodeType({
+    class: ComfyNotifyAction,
+    title: "Comfy.NotifyAction",
+    desc: "Displays a message.",
+    type: "actions/notify"
+})

src/lib/nodes/ComfyBackendNode.ts

@@ -25,7 +25,7 @@ export class ComfyBackendNode extends ComfyGraphNode {
         // It just returns a hash like { "ui": { "images": results } } internally.
         // So this will need to be hardcoded for now.
         if (["PreviewImage", "SaveImage"].indexOf(comfyClass) !== -1) {
-            this.addOutput("output", "OUTPUT");
+            this.addOutput("output", "IMAGE");
         }
     }
 
@@ -78,12 +78,13 @@ export class ComfyBackendNode extends ComfyGraphNode {
         console.warn("onExecuted outputs", outputData)
         for (let index = 0; index < this.outputs.length; index++) {
             const output = this.outputs[index]
-            if (output.type === "OUTPUT") {
+            if (output.type === "IMAGE") {
                 this.setOutputData(index, outputData)
                 for (const node of this.getOutputNodes(index)) {
+                    console.warn(node)
                     if ("receiveOutput" in node) {
-                        const widgetNode = node as ComfyWidgetNode;
-                        widgetNode.receiveOutput();
+                        const widgetNode = node as ComfyGraphNode;
+                        widgetNode.receiveOutput(outputData);
                     }
                 }
             }

src/lib/nodes/ComfyGraphNode.ts

@@ -23,6 +23,10 @@ export default class ComfyGraphNode extends LGraphNode {
 
     defaultWidgets?: DefaultWidgetLayout
 
+    /** Called when a backend node sends a ComfyUI output over a link */
+    receiveOutput(output: any) {
+    }
+
     override onSerialize(o: SerializedLGraphNode) {
         for (let index = 0; index < this.inputs.length; index++) {
             const input = this.inputs[index]

src/lib/nodes/ComfyWidgetNodes.ts

@@ -135,10 +135,6 @@ export abstract class ComfyWidgetNode<T = any> extends ComfyGraphNode {
         }
     }
 
-    /** Called when a backend node sends a ComfyUI output over a link */
-    receiveOutput() {
-    }
-
     onConnectOutput(
         outputIndex: number,
         inputType: INodeInputSlot["type"],
@@ -411,7 +407,7 @@ export class ComfyGalleryNode extends ComfyWidgetNode<GradioFileData[]> {
 
     static slotLayout: SlotLayout = {
         inputs: [
-            { name: "images", type: "OUTPUT" }
+            { name: "images", type: "IMAGE" }
         ]
     }
 

src/lib/nodes/index.ts

@@ -1,5 +1,5 @@
 export { default as ComfyReroute } from "./ComfyReroute"
 export { ComfyWidgetNode, ComfySliderNode, ComfyComboNode, ComfyTextNode } from "./ComfyWidgetNodes"
-export { ComfyCopyAction, ComfySwapAction } from "./ComfyActionNodes"
+export { ComfyAfterQueuedEvent, ComfyCopyAction, ComfySwapAction, ComfyNotifyAction, ComfyOnExecutedEvent } from "./ComfyActionNodes"
 export { default as ComfyValueControl } from "./ComfyValueControl"
 export { default as ComfySelector } from "./ComfySelector"