From 416ba2668592412e67a0581ca13b2324bc398f32 Mon Sep 17 00:00:00 2001
From: Yanming <luoyanming99@gmail.com>
Date: Sat, 2 Aug 2025 03:39:50 -0700
Subject: [PATCH] Add files via upload

---
 .../runway_text2img.cpython-311.pyc           | Bin 0 -> 3484 bytes
 .../websocket_image_save.cpython-311.pyc      | Bin 0 -> 2059 bytes
 ComfyUI/custom_nodes/example_node.py.example  | 155 ++++++++++++++++++
 .../custom_nodes/runway_text2img/__init__.py  | 108 ++++++++++++
 .../__pycache__/__init__.cpython-311.pyc      | Bin 0 -> 5306 bytes
 ComfyUI/custom_nodes/websocket_image_save.py  |  44 +++++
 6 files changed, 307 insertions(+)
 create mode 100644 ComfyUI/custom_nodes/__pycache__/runway_text2img.cpython-311.pyc
 create mode 100644 ComfyUI/custom_nodes/__pycache__/websocket_image_save.cpython-311.pyc
 create mode 100644 ComfyUI/custom_nodes/example_node.py.example
 create mode 100644 ComfyUI/custom_nodes/runway_text2img/__init__.py
 create mode 100644 ComfyUI/custom_nodes/runway_text2img/__pycache__/__init__.cpython-311.pyc
 create mode 100644 ComfyUI/custom_nodes/websocket_image_save.py

diff --git a/ComfyUI/custom_nodes/__pycache__/runway_text2img.cpython-311.pyc b/ComfyUI/custom_nodes/__pycache__/runway_text2img.cpython-311.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..65c64904e94f4f2850d7c22583382d7a7755e59e
GIT binary patch
literal 3484
zcma(TU2GG{dDiRy-^2+a7>+D~kTZ~k<Epk$2spvXxtBOm;t1hJ-R^lN$p+hNX4gPm
zr;1KI<RmIl35n|!ba^SI;vRUZgep}+s7O4nk)^X%igapq4|y}<PD1z8Z`O{JrBuD~
z_}lMq=KGnMZ{{z7fEPjg`#*labl-!}zZtRJ>`mnLZveTAFv46Gr5j%^%>ix8+H!oF
z=NdeF+73KE>&Q9NP7c}7IKuXu2s<9w5PAgvMoYV}gScca?Zyu5gs%s4vncA?V&Y_!
zGh8Dp8c`EdfI1R6X^~`_BhAnDJg?6Hc^47HJW1Oym*$Bb+is!<z<<;lk#=A{?ZkFw
z6)ZY$BCzKIdv2qBrkKAjt)$2eEv1--N{77yv<6>bf&on#FhxRA!NinixYpzdvF!2i
zf1LyXIpVMl^VlwP3pN%Vc40U6+~eOtd9e@sTe1L9!6p^D=a5-O+LDHwlB-3T5^OkU
zQ!|O=xZ%kavznZh^CV^PiDXKKJfp1(U|mLTgrdCR!el`LC4<k&c>vOlty3W0Vs39?
z2F^0=6F^5DhO<DGTtPD&*JZ3N8O|jlFD?P8YNVhV{(_Rtikh4wN>TeOf)(L2Syg3I
z2|vU}&&<!MgsSt|qOu~*U!+9JO-L(*&W|d&g_XI)e5R;sN>0o}?$mi|rd!k?t*7MN
z;$UIL@FkMdb15-(c{)B@dhfs1V{n%^2sYIMfHhQiqQKtTq3FYppLFWShjrn6?a<N3
z>eEv%H2rvrbz$z^9QyQw7pi`ImUSV;bgZ1Z&$%yHM-1@pHg6rdJ>J(~lSS@GOvc9J
zhI^6Z36(Tb@;3Y(icKdnPNvjs3^(^g3gBI&?b<KPEgW+X0Dsh?SDj_28qyrwlOQ-<
zR*g?NwzJC4KcV|JnOk+0xw5Uym+jbb&%WyRAkDQic6s~kBJ3=?%5LoX%KZR%k6Qdy
zPua7v%i7LZ>-pgj=<SZKrD)aL*k!$?g0ROL!>Tub<S^3Ob~KJ4x_8xQiQ8NDu^6o*
zA$qjx`<?GPcOBi}FQe;R+51+$d~c7?I(F<VLv?F3tKX~sAEQ<Fm*AwA9aqdL=wnN+
zTalN&kY~TO+N}o4fn~GCvY|L|#jIacg7w<l@Q1-=|2D4e-^RdRV43ZDvsq)aQX8Rd
zt=(T7v}&ye%fSVxEB`nDp&yrjOWs;5ircJKIb_lMAFE)^@8ieX;hjiuyLHA-xo9|P
z1ZU=wmtvPiIDF#A@yn$XR$3tWYcf^xIg;0eYZ8^EPqIYFE1IAZZO{nBirSJw<&vby
zO1|V8Ard82h}vl<7^Zs|=mKDL<cm2`&My#3@);tURk1X%q-h29%+QcjkOwijHrS|a
zx$Iy@$qilmV2G7CQBy>j-CkR4^;unBA`*sbY1o%lC2zQzRn_pAT4IrAOWlo%B0x~W
z^wexhSdio_!Dochk*0(awV>oxBIIOMmGg^2e`|FAV3ec%U~B6I^PHa<A1Uo=++CPt
zpaQpM+hTjL)Z3iWoIq$+HOPj*9K!+>{K070uqmqHTqGLg)9}LxCX7?6(8k?FR1E?u
zD5_?(QAt*bxS&u-lcW_@derm-1{<FEjSMNUTpM;agW+qah|Dj_ijrYh3M5Zi-85Vo
zB@YHQ6S+nxyN<Gi6byf=aTA#iL_>yMCE10ymW;AvMZ3*;V12|ZWQF#bR6$zFDiSt$
zNFsfou{gpmItdK8;6f6CZ}5Pq>^!lDz^+EwxuiV+N|8q5cD)-0x9^05AXondMSTrz
zhCA;>^#dm=;el#+VBJ*<b=48)nc_CX2lbxGN;p{!Cv|`Fn~q-n$j=@w0$@G-^L5uZ
zT?cA=A~k@A->-Wd9dHi;Sa;QfUeEc>aOBQFt?hiRyQjA2K&?B%G=i@{BlsNy>!Wo)
z3bcLg7hd{>iodt&@74XiweJ02y1#J$(R<JPW$=sO#&0X#A6C0RTu-bg-fV`tnP=yB
zfa&77Cx`T&bI>c{;c9qT_Yc2;R_8&JeVwRn9~&{=2q<0&k5$8Cx_|83w$3}AKltwW
z)*YL{@GbeXE4QwE9<KxsZA`ojp45XUH@gnr$!?r}{E6Omy3%#J+I4z;9I{gl^=@Qd
zhN5~XTDRHTFLIlao_ojseCgq(uf$5<xoY3JN@Tbi8U8vl_A)Z|v{;EuRwI+QM{5Uq
zKL6?M$v2yO1@H&P?S`9?qk8|yldAxrS0ba;$f({q+KdF00sGe9cI%h7uYNXpYjS<^
z`*sxSc!$&9ccRdKegFAN@Ip0sL3dwJPr_RNE!HvOLC-wikz@R`j_^ny|E$je_&)=`
zh<ng;k2ijdf8G&}_wmpB9DpaIUc)Vlc_~Lk(eR35PQk@2!~LRowJ2qq5|=1qB?D|<
zCM&6GjzG1-^Z;0dLl{rZ%_JLdCc`~8mmEzcrr@bK8cW5;r)Dmjo0RSYUHTpa=Eh|=
zIAwqpVD;Ak)>^;1&E{~|5d&`IY_Hoq4%l%3UhzQQ{8fX|g*p$eP1HjB);_9*_OE?Z
zxA!|@kn4BDhG}N7)Y+^vmgjWTOHY81{se$wlNIye(;-Ij(}@YQB$)mh9m%PSadC7a
zHaja$#-^v?5j<-|Oy<SJ?DPcO0m;~8yd`@p@*mu18#x}PQ!qPwZ>jeIz&_?U*u<Q7
z&BJ~*<Xba;HPoeBdkuxw%wG-d(XG91Kg+Qsyc@nU`QPo9Ua%oYx4!RKowx0^>8PXb
QHR&Os4+0%&bTfPZ0Xn~NZU6uP

literal 0
HcmV?d00001

diff --git a/ComfyUI/custom_nodes/__pycache__/websocket_image_save.cpython-311.pyc b/ComfyUI/custom_nodes/__pycache__/websocket_image_save.cpython-311.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..0089171384f7096a5a798806ffed5cbdea8a93bc
GIT binary patch
literal 2059
zcmZt{O>YxNboR@!W2ZJIiA5+?>H`o|!37CaP=#@v#7)?^*inU>YPEPbVN-8xX4X_r
z9My*$jFd{WLJDdxKG2jZ7r1d~C4Pb(MT#{NQlzQ}Zbs>eOZ#S=H8C)|`{vD?_ujnE
z+202S`Vow4&wjkqmJ#}!E!+WXJ`QGJ+C&C2ggml^tROI4%!^rx&vI7bvnT5j(IhgY
zb!5m}B0}5nbw;+&kj*|z$a)PijmDH$j65b#C28x6rkXos<C80Nkb>nVGEr7AP*yZ0
zLs&;!K-=zeXJtc#!=<dpkl_dg)O*%Z)(5<O@A38<UO3uE!)g7V$<O`TyiE$Z?@T8r
zbX6iG0nWaIj{tz%f*~4`C0GJT%PRF^^KQw)ra?UxGoElaZp0=Ne=v^(LYT7AHxu#C
z=jVut$$Y+4DC_g%*wpPAy=>z7c)?yM&n4$`CE^rpZMk5WWd3{C7>%pd2v~W(Sf>7@
znwraKnOmtudgYV%Z9jixpIJ;I0IFy|G}0P6^=P#*_*q*LL*X_Oec=PxLYQTo`kDUD
zQ6#hC(~H&xes0bUKV$X6$3yW5eA+|~qaBW=Zpi9KCERUP5tf)FABJ$kMbQ?>ZFgnY
zL_=7UD#AGW_LntCW<{>MRo9hZNN?KfO3&uuqL*g{c-9nW;4SA;pNl)kTnwQlpNm%%
zNVT*DR&RGLuk-G3Io#O^xfsa%UA)(`-qo6?BGv;g=h3+W-hVh!hFtL&XyF};1XPg>
zWd;0NlQw>}5EP|H#ViGjT<w10+N=Knqf$zpe8mDiHb9jet2nBZ94k+#KZOg6*d*k-
zj;TWK=tYxCxnhYb%O$&5#>{)_TfhZd$5<~@ak)t4T;3{Dk4~I2a4RKi*||)ErJ|ub
zrlvz1<x7q^)<<Px<`*!l8!8uXgMOJfW|1nqwy0oH$+8y-%fx7BU=|e|1n|M)K2oJ8
zfrmT;P(?3;gBz=R!P7gz)Af6gD^FvOe`p3T?FKK^<W?{QVCZBU$-ZF@YKc~8@Ih&_
z{B!xI)rYHlk+VCIv(3o4-N?CS=v*!KG8}n$;kW4HGfzH!^3_wR8NIk0y|@>>vJ<`X
z*Y!WAn$g5=G|>!C?1m?5@m3(T7Z}|MjMfL;lb;X0kX~dOm%nTVrgsC=jleW8J&4^;
z*2Zh&`@v&1A7*{o(^xEDEX)UI0YGDQ8v+UnD4o$;(2!_Bg$qV@EZZCtF{?WsXu_0@
zj`Bd#{$yH<PsP;9#Q4go4_y}B?b9Gdm|<1)Pw@Cg`Tpw0YQsCiZB@to)T?RBx@~G2
z?bkHBV3hI<4`|xmlAb@@@#pe7A-3t<DHxcw8V%e`Wae(Fo$HEvC+5_6CONB8|Lk0b
zU1RF(c!GN4u}ore_U0{~cg$kIFC~6<ryy8|V+NR3VgsmlpOzA8i;@CY8i3al@*G?L
zrbFmQgTvLCR`BEM)P8ulI=vqpsZO<JALH$UVGwAGf&wK1peyeZIJIL5J0E@+QYtya
zn*y_I46{0<e%?{%c|+krMn1y;lc2|$lUjTxmQHIoVyRS8olL*u-y84MPDsDN7eR!j
zpWFk`mIOg)p`k|a+d|=n`}X4gDu3GHlLGkjK6uS&|GwS0ij)(L5%x=nhz)eS^Kx(h
E2Q8Y_v;Y7A

literal 0
HcmV?d00001

diff --git a/ComfyUI/custom_nodes/example_node.py.example b/ComfyUI/custom_nodes/example_node.py.example
new file mode 100644
index 00000000..e86b4f11
--- /dev/null
+++ b/ComfyUI/custom_nodes/example_node.py.example
@@ -0,0 +1,155 @@
+class Example:
+    """
+    A example node
+
+    Class methods
+    -------------
+    INPUT_TYPES (dict):
+        Tell the main program input parameters of nodes.
+    IS_CHANGED:
+        optional method to control when the node is re executed.
+
+    Attributes
+    ----------
+    RETURN_TYPES (`tuple`):
+        The type of each element in the output tuple.
+    RETURN_NAMES (`tuple`):
+        Optional: The name of each output in the output tuple.
+    FUNCTION (`str`):
+        The name of the entry-point method. For example, if `FUNCTION = "execute"` then it will run Example().execute()
+    OUTPUT_NODE ([`bool`]):
+        If this node is an output node that outputs a result/image from the graph. The SaveImage node is an example.
+        The backend iterates on these output nodes and tries to execute all their parents if their parent graph is properly connected.
+        Assumed to be False if not present.
+    CATEGORY (`str`):
+        The category the node should appear in the UI.
+    DEPRECATED (`bool`):
+        Indicates whether the node is deprecated. Deprecated nodes are hidden by default in the UI, but remain
+        functional in existing workflows that use them.
+    EXPERIMENTAL (`bool`):
+        Indicates whether the node is experimental. Experimental nodes are marked as such in the UI and may be subject to
+        significant changes or removal in future versions. Use with caution in production workflows.
+    execute(s) -> tuple || None:
+        The entry point method. The name of this method must be the same as the value of property `FUNCTION`.
+        For example, if `FUNCTION = "execute"` then this method's name must be `execute`, if `FUNCTION = "foo"` then it must be `foo`.
+    """
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(s):
+        """
+            Return a dictionary which contains config for all input fields.
+            Some types (string): "MODEL", "VAE", "CLIP", "CONDITIONING", "LATENT", "IMAGE", "INT", "STRING", "FLOAT".
+            Input types "INT", "STRING" or "FLOAT" are special values for fields on the node.
+            The type can be a list for selection.
+
+            Returns: `dict`:
+                - Key input_fields_group (`string`): Can be either required, hidden or optional. A node class must have property `required`
+                - Value input_fields (`dict`): Contains input fields config:
+                    * Key field_name (`string`): Name of a entry-point method's argument
+                    * Value field_config (`tuple`):
+                        + First value is a string indicate the type of field or a list for selection.
+                        + Second value is a config for type "INT", "STRING" or "FLOAT".
+        """
+        return {
+            "required": {
+                "image": ("IMAGE",),
+                "int_field": ("INT", {
+                    "default": 0, 
+                    "min": 0, #Minimum value
+                    "max": 4096, #Maximum value
+                    "step": 64, #Slider's step
+                    "display": "number", # Cosmetic only: display as "number" or "slider"
+                    "lazy": True # Will only be evaluated if check_lazy_status requires it
+                }),
+                "float_field": ("FLOAT", {
+                    "default": 1.0,
+                    "min": 0.0,
+                    "max": 10.0,
+                    "step": 0.01,
+                    "round": 0.001, #The value representing the precision to round to, will be set to the step value by default. Can be set to False to disable rounding.
+                    "display": "number",
+                    "lazy": True
+                }),
+                "print_to_screen": (["enable", "disable"],),
+                "string_field": ("STRING", {
+                    "multiline": False, #True if you want the field to look like the one on the ClipTextEncode node
+                    "default": "Hello World!",
+                    "lazy": True
+                }),
+            },
+        }
+
+    RETURN_TYPES = ("IMAGE",)
+    #RETURN_NAMES = ("image_output_name",)
+
+    FUNCTION = "test"
+
+    #OUTPUT_NODE = False
+
+    CATEGORY = "Example"
+
+    def check_lazy_status(self, image, string_field, int_field, float_field, print_to_screen):
+        """
+            Return a list of input names that need to be evaluated.
+
+            This function will be called if there are any lazy inputs which have not yet been
+            evaluated. As long as you return at least one field which has not yet been evaluated
+            (and more exist), this function will be called again once the value of the requested
+            field is available.
+
+            Any evaluated inputs will be passed as arguments to this function. Any unevaluated
+            inputs will have the value None.
+        """
+        if print_to_screen == "enable":
+            return ["int_field", "float_field", "string_field"]
+        else:
+            return []
+
+    def test(self, image, string_field, int_field, float_field, print_to_screen):
+        if print_to_screen == "enable":
+            print(f"""Your input contains:
+                string_field aka input text: {string_field}
+                int_field: {int_field}
+                float_field: {float_field}
+            """)
+        #do some processing on the image, in this example I just invert it
+        image = 1.0 - image
+        return (image,)
+
+    """
+        The node will always be re executed if any of the inputs change but
+        this method can be used to force the node to execute again even when the inputs don't change.
+        You can make this node return a number or a string. This value will be compared to the one returned the last time the node was
+        executed, if it is different the node will be executed again.
+        This method is used in the core repo for the LoadImage node where they return the image hash as a string, if the image hash
+        changes between executions the LoadImage node is executed again.
+    """
+    #@classmethod
+    #def IS_CHANGED(s, image, string_field, int_field, float_field, print_to_screen):
+    #    return ""
+
+# Set the web directory, any .js file in that directory will be loaded by the frontend as a frontend extension
+# WEB_DIRECTORY = "./somejs"
+
+
+# Add custom API routes, using router
+from aiohttp import web
+from server import PromptServer
+
+@PromptServer.instance.routes.get("/hello")
+async def get_hello(request):
+    return web.json_response("hello")
+
+
+# A dictionary that contains all nodes you want to export with their names
+# NOTE: names should be globally unique
+NODE_CLASS_MAPPINGS = {
+    "Example": Example
+}
+
+# A dictionary that contains the friendly/humanly readable titles for the nodes
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "Example": "Example Node"
+}
diff --git a/ComfyUI/custom_nodes/runway_text2img/__init__.py b/ComfyUI/custom_nodes/runway_text2img/__init__.py
new file mode 100644
index 00000000..bed0f3ad
--- /dev/null
+++ b/ComfyUI/custom_nodes/runway_text2img/__init__.py
@@ -0,0 +1,108 @@
+import requests
+import os
+import time
+from io import BytesIO
+from PIL import Image
+import numpy as np
+import torch
+from dotenv import load_dotenv
+
+# Load API key
+load_dotenv()
+RUNWAY_API_KEY = os.getenv("RUNWAY_API_KEY")
+RUNWAY_API_URL = "https://api.dev.runwayml.com/v1/text_to_image"
+
+class RunwayTextToImage:
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "prompt": ("STRING", {
+                    "multiline": True,
+                    "default": "a fantasy landscape with mountains and rivers"
+                }),
+                "ratio": ([
+                    "1920:1080", "1080:1920", "1024:1024", "1280:720", "720:1280",
+                    "720:720", "960:720", "720:960", "1360:768", "1168:880",
+                    "1440:1080", "1080:1440", "1808:768", "2112:912", "1680:720"
+                ], {"default": "1024:1024"}),
+                "timeout": ("INT", {"default": 30, "min": 1, "max": 1200}),
+                "seed": ("INT", {"default": 42, "min": 0, "max": 4294967295}),
+                "mock": ("BOOLEAN", {"default": False}),
+            }
+        }
+
+    RETURN_TYPES = ("IMAGE",)
+    RETURN_NAMES = ("images",)
+    FUNCTION = "run"
+    CATEGORY = "Runway"
+
+# ✅ ComfyUI expects this function to exist at module level
+def run(prompt, ratio, timeout, seed, mock):
+    width, height = map(int, ratio.split(":"))
+
+    if mock:
+        print("✅ Mock mode enabled – skipping real API call.")
+
+        dummy_np = (np.random.rand(height, width, 3) * 255).astype(np.uint8)  # [H, W, 3]
+        dummy_tensor = torch.from_numpy(dummy_np).permute(2, 0, 1).float() / 255.0  # [3, H, W]
+        print(f"✅ Dummy tensor created with shape: {dummy_tensor.shape}, dtype: {dummy_tensor.dtype}")
+
+        return (dummy_tensor,)
+
+
+    api_key = os.getenv("RUNWAY_API_KEY")
+    if not api_key:
+        raise RuntimeError("❌ Missing RUNWAY_API_KEY environment variable.")
+
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json",
+        "X-Runway-Version": "2024-11-06"
+    }
+
+    payload = {
+        "model": "gen4_image",
+        "promptText": prompt,
+        "ratio": ratio,
+        "seed": seed
+    }
+
+    print("got prompt")
+    print("=== Runway Payload ===")
+    print(payload)
+
+    response = requests.post(RUNWAY_API_URL, json=payload, headers=headers)
+    print("=== Response Code ===", response.status_code)
+    print("=== Response Body ===", response.text)
+    response.raise_for_status()
+
+    job_id = response.json()["id"]
+
+    # Polling for result
+    image_url = None
+    for _ in range(timeout):
+        poll = requests.get(f"{RUNWAY_API_URL}/{job_id}", headers=headers)
+        poll_data = poll.json()
+        if poll_data.get("status") == "succeeded":
+            image_url = poll_data["outputs"][0]["uri"]
+            break
+        elif poll_data.get("status") == "failed":
+            raise RuntimeError("❌ Runway generation failed.")
+        time.sleep(1)
+
+    if not image_url:
+        raise RuntimeError("⏰ Timed out waiting for Runway result.")
+
+    image_bytes = requests.get(image_url).content
+    image = Image.open(BytesIO(image_bytes)).convert("RGB")
+    image_tensor = torch.from_numpy(np.array(image)).float() / 255.0
+    image_tensor = image_tensor.permute(2, 0, 1)  # [3, H, W]
+    return (image_tensor,)  # no batch dim
+
+RunwayTextToImage.run = staticmethod(run)
+
+NODE_CLASS_MAPPINGS = {
+    "RunwayTextToImage": RunwayTextToImage
+}
diff --git a/ComfyUI/custom_nodes/runway_text2img/__pycache__/__init__.cpython-311.pyc b/ComfyUI/custom_nodes/runway_text2img/__pycache__/__init__.cpython-311.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..b8786474eb6961801da0c46b6cee01ba530091a8
GIT binary patch
literal 5306
zcmb6dTWlN0agRKb$G0d_A|=@pZ6$S>NV2FZvaH5UBwLnc%ao%>TQg1$7w$<qnXlX(
z?T95CCU60zP?5AvVWY88B}h{h>4!e^>5mrZ=bdnXi30=}De~cdu}%BqpLF(4lypv8
zq<hP`+1c6I+1c5-S^nPd_aG?D-+sR02_W=W(r6ZX5BX{bklP3&Ol48pd{b%4Xxq{@
zc-yk}9G#{q(r3>)a?Z5Vz^SY&%@};{v>V=Z)|2z5y%e$`-uo#?{j50`GHBGQuSvj;
zPbuh!pIOp=>`Vu2XbNH1Ergj*ZO{w9z1|>p-$LmYkQ?G#tRwj#+tTF23?CGzbStLj
zQNr_xkW&dtcTKFxyfi%vsAD?EE%ShQvjq;bxFGZS8+u@FG4;dotL*rdY4+0O)pGxe
zEElD*lP9^NFo5|R17azElUvJW2Qr1+$s5U&GXJK`$^}*+(lX8I$-AlZSHsY_jd(;X
znYQtEOx;3Oo1|%MOFOVV?ZkB2g&kl^C+O|cLvzM-3oyw-!B9A}Zzz!JhMygNU(Fa(
zU<%u?9n;ux*S2I^M%ei=g<Y5ts3jYAV-NOXANJq18?68i;+DNu2)J4euJB!lKui`%
z>5Ry~Q4&NR>(2Rwx#`r@Lb;!dFL8O9lh)!{E{~-QSLEY21$iZ&E0lmP<fS;!<Dzhb
z7o~(vcjro3S;z``UUy-BiGx-mSav5*4-Jkb2S*111vp09j{zMT2I|monMn=-_0-T{
znE^0HT4fi(2+4JNWDg+}XaM_n329^$u;j?-*r*|Rc-T4zw4r}=aMa`+N+yTKPA7+e
zHeyaD`oW4K91f)0r&9}$x&eDc%q?ug2}#(Ee%+oE@&LFuA3aOh$u4Kf9(?`Q*KgSo
zc-ZXh%;b3Le8QnSi((;Hly!&5$wEPQ$wH1Vlw_Tjc(5v+D`c+!0f7hFGg(O@6U852
zJUe!6Q3B&#%a#gj+_iHe&*f&gHD0`Swvby|Tb#a@DM@l6$L0%|m#&%a$PzCc5^~Ge
zSXRgjGRqDW*L3f6>dN8*yKwc&<b3()H#}+JP<IF>mu>;5p@tLrqxG)d`mv|$J*kGn
z{+-qaGC^0k!O&gM+y@M5>%Q}2CH!2&O?UP;Jm~20E%EbX_s=Q4)1-D?tao&8O>Dn$
zzf+0NkQ%#O?>N4Bb=#plHKN2$H9Gtqfd(SL9&GV6P|)r93iKdmaZFzxpPB^3X&f#|
zx5H|6r@4?+!mTqbo9A*o%jzDM%@uGdOK=~{zER?`dp+JvmXo9$FRv7^NOn~B%}p*W
z&ZSJd?6*?mmnY|S=KNyn?85Xcc>me)g~_SexvOFq3=(??I7$H11VKQgGXQG)&qGH`
z!`9+}WdQiQ9XUg{UNH$zp-6XarXD6B7^z*T2cxwM<S``^p&18&W^n>yoa|x1ZX@~7
zsh6991z0N7=Al+|FoFPX4UN_ydJV=@9aU$A7Pn>Bp;0#LC<@6$vAwdouiqxo?i;1z
ztUw6+6tw<~Xl@itmtoZAp*fm8$SQk(9;*gbmmEGAv1)w6NW0>)@@;w#xDGJvz)tM4
zYP7lk;W^+#PlRL4ltEsVvDn64VOB|iFv^Drd!S8X#a{a!ZPkcE6=n${sZ2a)udE(?
zp1fA>s=MNrf^u(@6L!#N<#>m}{)+oUbkBy3$bHvd_25tiJfLF7t#|3F*P?CMIvRHJ
z&>rB(|LUzgp&N{S`arMM{r!-iIiOa&70(~rtd_3&EOWM3e5-BE5mtYELc1t%w*NQI
zif=R8>^uCneo(>szRlR-{w8ka!?BKL2derpss`lm?h{buiSuv{z4}o#Xt73Tll!nW
zz&koBe%LEJSf$gt)@n<o<=bdh4OK#$`_h}UkT>qK?jAmJ-G~8VdnJUA9;^pD58>Sh
zcr`$w{F!epy=PCl)jQ)>9p0hjQwKV(9%<rm`XF*0BzF|r__z{y!qbmgc|g`-f7`b~
zC1B-SCE?g8)mE9r;k~lDf8ylvCJiD+t97)swR!A-^8mxgtz6ace~v~n$;0-_>hUI4
zdEyLOW&Iy<C{l@h`}kG`ZW4DAy{X64SSbnGd~x@u@yn2FK$3>}IG^WU&+<6_MeSE{
z>AFxX3i;(YWEt5w<h=0=m(315BJumt*8~vHGIAV|qzKFKL+D}Qr4rd8BKKS=ms^WN
z>MRw+cm~ACAlk?Xr4>jH$KqvoUmVM81Qp3=qcliH-rI;@79@#GZB4jAy&Hm9$mjUH
z9KXSd0#SCL?3&;?kr(6T7Oq&#3K^15<xj3kg?!mF1Q}z0GTA>klCbOk@e+LLMWJkP
zlznFlc~DO7UjPltEkEivlZXBvK!ybpbk8!MA2xIM1g$%WfwH>COxsA(Es|IxlCx9x
zEEnXs*;8&mbLLFk6cxY1t&s~v97vGx(nKfSwZd~4rYN@=jJ#AV<Rw0Sme?OCPmm`H
zc#V*Cn}8ovCli!70u$;^N#^8|q`Re3CIbm0q>L`e9*ZRzGNF<v=*}fhfSoLnb3iP*
zrw}ZOHxkY~B<qA>phP0|7w>Jy7a#$}aS#%}$q6#C{t|54-Z+t$AUz+@?Q>HTiC&W8
z7InLjhxAnflf$bhl9nxB)SV(^`GuTLleg~VB(lM}qXhk<x<f9AnHAl$1j#R(FXbQ;
zbrpFrSCWBc32a3q66eqzhGXcqf}}f_dE@@&17%2>JSmC=(Y%825-gZ579`nnJ6oKa
z(Y>aL*bEp}r%4j5w}_k|@hq&1H9N)Ui7_Ds&oA@39i{<Bm;iLh@_bQuWlRqdX~ISq
zihN!q?r(I!Ee4e3M3Gxdw29=KWI6=-vKU?=lIWy6ZVFgl(VZ*2u)HGc3?_$y1v~jn
z%sd{t3+|BYb$-p<URTj{9I{B$KXm76;dNHPy4%Qi*^-#mDV8ifo7LR}vY3-O-D`Hg
zPOh@L!)WL}lV;8&k?(B$KNT~DB_T>AfZSlD$8g4LXg3nQXTP)f+gE<`%2rrC+OHk$
zS0e*jWZ><~waN97hc0gexg0&aj9&?itIUMPOeoC6myz}lrf;Wync5uLdVc$u8X49i
z!`qS?8G8_(*a=VkdFW50YIsHq&#W`M(IZ;)#1^AQlUg+SAbNTydiwKYpTB&cQlk@E
zbmBqu!cO#p8oj7RFaBjwjn1w6jM<jAv)af7C9q8DJgd#~Cay9o8ndD>D-T<atiPo&
z-T!i;U}!^7V$Z07gceBDrs`g}tWb_wYB$iSbX`^hDJ_svm=xS`I{P<IDII-K>#=9+
zN1kc8=%5pBHbG}?su4h~-Sze(^-z1gCDL#^NDqM8)V~7A;aTrf>8M6W6*{_0yA{uI
zmG0H(UWM+3=K6`ZQ@2vJ)R)2NM)lrn+b8ayQi7M%;3X}1sdmBOKejol^`27cQH>r|
z=+P$alQSxv)aax_C+oED0ey6bKC03^8r`GNJrDhD@4fo7SKnnf*gFfF|9I^@tRd7^
zo8I-ezJKc7*EU{jAjZ*dfOTpYCa>K4iE?BV>bBE--#?)QFRH<dTJYjJU5~_WPycfI
z7neS`wC=8ZBk!I0*_k^r)q7m?9^bsa<4r2wWIc9lGoi)$wqDj^C$}A1Y-D}9?hCC?
z*8S~7nLDGZ|0&J?)E2YjA5{E<yRF?y&-c~V7q!+G72k`yjBms9e)3)ahJPKNhu#R;
ziz#z2Ce^@sEpT38&VSk3`9a^EHMKRNk(#Vs+VzDtF8=!Fogb^w=d|c^s;^J;^*!(n
z@A!td7ggW)G~f4Xle^)V7ViDs#3vWF7C)N(WL6CiYvJJs;jx|Ym>PaT3%~GrNe!PP
zcdniX;S)RI6I;|)a*I>LgIahH$iCS6iwfOo+`kfG@nv!*QUqKD0HL^Gh5`GP=@~ON
zIE+{(o+UgN2#?eJ&q#zD4iaYxARjn!o&a*6F^-t-BVkC$m^V;8oSHp1$)24VpPy$h
zk6*b0f2+(tal?GfykOpCzc0!lh1~q4e*kFMDT=D2V9j{y$Xzp@I`Y<xr;gecYptU;
z#aipARcY2b^3{x|j=B|V1>d54HSgR0TmG8A9_oC*^@9$j>p3;lr-k}zfqLkrTA)G4
zs8|Di6a1af{&~<b@&eV>K>MIVp^m5T_%`R&7&uo#K_Lyb!Gw;O((!D=Zj0I!6gI1|
GH~$7%UI3K<

literal 0
HcmV?d00001

diff --git a/ComfyUI/custom_nodes/websocket_image_save.py b/ComfyUI/custom_nodes/websocket_image_save.py
new file mode 100644
index 00000000..ca56ebc8
--- /dev/null
+++ b/ComfyUI/custom_nodes/websocket_image_save.py
@@ -0,0 +1,44 @@
+from PIL import Image
+import numpy as np
+import comfy.utils
+import time
+
+#You can use this node to save full size images through the websocket, the
+#images will be sent in exactly the same format as the image previews: as
+#binary images on the websocket with a 8 byte header indicating the type
+#of binary message (first 4 bytes) and the image format (next 4 bytes).
+
+#Note that no metadata will be put in the images saved with this node.
+
+class SaveImageWebsocket:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required":
+                    {"images": ("IMAGE", ),}
+                }
+
+    RETURN_TYPES = ()
+    FUNCTION = "save_images"
+
+    OUTPUT_NODE = True
+
+    CATEGORY = "api/image"
+
+    def save_images(self, images):
+        pbar = comfy.utils.ProgressBar(images.shape[0])
+        step = 0
+        for image in images:
+            i = 255. * image.cpu().numpy()
+            img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
+            pbar.update_absolute(step, images.shape[0], ("PNG", img, None))
+            step += 1
+
+        return {}
+
+    @classmethod
+    def IS_CHANGED(s, images):
+        return time.time()
+
+NODE_CLASS_MAPPINGS = {
+    "SaveImageWebsocket": SaveImageWebsocket,
+}