From d711fa71e94f3a433a779a3913beea85d2a2947e Mon Sep 17 00:00:00 2001
From: brucew4yn3rp <batknights5678@gmail.com>
Date: Sat, 19 Jul 2025 23:09:54 -0400
Subject: [PATCH] Save Selective Data to ComfyUI metadata

---
 __init__.py                          |   7 ++
 __pycache__/__init__.cpython-310.pyc | Bin 0 -> 318 bytes
 __pycache__/node.cpython-310.pyc     | Bin 0 -> 3060 bytes
 node.py                              | 100 +++++++++++++++++++++++++++
 4 files changed, 107 insertions(+)
 create mode 100644 __init__.py
 create mode 100644 __pycache__/__init__.cpython-310.pyc
 create mode 100644 __pycache__/node.cpython-310.pyc
 create mode 100644 node.py

diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..5459f3f
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,7 @@
+from .node import SaveImage
+
+NODE_CLASS_MAPPINGS = {
+    "Save Image (Selective Metadata)": SaveImage,
+}
+
+__all__ = ['NODE_CLASS_MAPPINGS']
\ No newline at end of file
diff --git a/__pycache__/__init__.cpython-310.pyc b/__pycache__/__init__.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..04724bdd3b19f5afa4c9390ac7c6148bfa517668
GIT binary patch
literal 318
zcmYjM%}T>S5Z+Bzt;L?aYmVmP1Bi&B6$#i_Sb7P8amh~O!fsZQtx)kLd<<VHS5Llz
z7u_wiI56M$&Bri5b}|_gA5Y8o=MT?cq4*z>FK&JIXC5SxR1-r>N{Nh+Dq2P|ZfF@x
z#z}hcO?^{Jhhr{YF*l;YxAP#AtxhgDYOI~|#vGlH!ijV^EgolgFn!24hk3SG<i$NN
zQZ}&GO6<b=F7_}S0SK)DK1p{N%m6?2?5fH-r&g*~LMM6aP;1p-TVu7|Sxv23ZI=0-
t>viv}fiS$^nXtcWcUA?UTIB#PUpB+&+E}^PcoV+sM;=ZHi`nm({Q-qmSHb`Q

literal 0
HcmV?d00001

diff --git a/__pycache__/node.cpython-310.pyc b/__pycache__/node.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..a4237c0920ab7471bc5bcd5194d5dadf3d3232ad
GIT binary patch
literal 3060
zcmZuz&2QVt6`$dkD2kG0%SqPBZmI&^7DBs56WeG}EVgUdT0^ZZtCoyy2p0odL)r9F
zqC6aBBY^@6*w<qJfOrAvQlO{)KXWS3Q%-y9p+K6xAuU<8OM=5UZ{B>p`MuwpFrT*&
zJbyol&*>6E|CE)*M}^9l@I|LkP(*Qnob-<!oYWOZg})MLUER^UhGTRyP6kUGH88tb
zC!5GPupDa-QH|=?i0W60lcTzq^RZ)7of`1ZQ`|(A%sZF|)hntnt8e^H%RP&PUhBGN
zo+vbWXNlAZ24{Y+f`#U?v#2dQEI#uke*sz#$z?hU#Z<XQj_PStrP?*R!j4XL$DoFj
zp&7@drjw;v$D##l)BH7t<>g>GOCsBJUwg^4ZM-mxloj}P;ER3%1xFJ+MKK~U6_hGf
zpoGbw)TsugL3JoI6v41YT^Qkj_Xk{{cLaLBGi0GbQ89(x2?Ode(s%iJBv#KnPEun+
zede`!$cDnY(+G_Z`<}@6nb+~(5Vzm=dQ@aC2X5f=p)k50KM$!Ww8Ji#L=g$R*Ippq
z7kNSFZ-_mDY5z@je{-_UxZm;Hz8jQlJ?;g8f9Cbt-bu3klbx{J86MZ>|Mno_VVCqm
z>P0gpX?g*y(0}b6c-*BfcTZ&ddVeTPLj0c33CY5Akqw1~joa74qm{Xz*Jp+g9A1D<
ziNpU2f;>^G`naP=P*H<sZWNHs8(Gg%>$WcC+=6T`$oU1iuploj$i)SDc|l%TkW07n
z>WxJo(Z}=&{pd!eYd0z!_0u}w_h5UkDl$9_1Mc@nrPjGu_9dpHG7rlU0Bl_t&DK$^
zz9%xbZXT7TReGl7PRPm}svSS@=3Od^$WX824g${Bz#&jK(ow~dH0-%ukE8=I6WT|G
z*_o8T@rLCWqvE{&+6@LCX#UsY<pYbC_bfIZSZv(0*nD8IdCy|&fyGwE5;`oAyl^Z4
zZvrFoB2~RyP`z{#tBnQKMylFeP;I8Ftp(MV$R5{ge>kp^+OEh~|JXX(CXM=Dt^U=a
zFd9dP2aOgGP<GJsF9#m+smOb8ICDwA2fWh>h0$jaOI(r^^DYMvlvJ2upZj6Y4TN#-
zQ|k4Su!1O4+d-7bY4I3NLgZ@o#&L_ZUNx%CWl%&vg~HNoto_H<ZG8XL{-xv<tYGcE
z@x$oRU7#$Ck_;|#5=$fnK}+a~Fxr<L0BI6*J)oQ!^xUWnn=S*izV=QV<!^m{UZz7p
zjo&WM+%A#wS&ELn>$7N`J%g3j+mfkf&ngHKU%!VhqCku>P~Qdm6&foXPn4vl#+c$Y
zbfHmY4b5fHOtlkztdEUxW^97Jeuc(aAB`<;$mqUOcF~`nk8`m$$pDGw-lFdns!x@P
zIYnbT(Hi%(`PgRFI3HWF{W-e)8PCS}3d1Pwi83yXA+qD8i8V!&+!U>%agN)QJTFX^
z;vDdaIaMb`UbtAE;wc^%V`H*Hv%C}+uJBYDFUQMo5w+qB&ArvA{k=9`p=ey<s~3-=
z?_!I6#~)9gz`9DYHTh_YU``9>w8#0lG+B%7I6qY|y1b5y7f<=Ki;pLt#Cp6kRpJtO
z>p9K;Ma3x2PJR@x#3pFUv~Z(NG3YW>0bgb=e@B;SG0ytfmx*$@&SVaHCkrNXxX@(C
zB$4i9XXi)Gpar3SgkQe<0RK_$(^HqX&q;64{ZwEglwWR*bg~Y>mTYp1Z&m_R3GNw-
zh-@@CO_LIXO9B!KVEz+n^kY_pACqh@N-|O10We8lOj5ScqjMKBlot3A7dmr$kj+)F
zh6c|3zQ}s89*?;k)b0L2=uqtsnM~v&+ksef8FPn1>Gg%y4xoeKMoG@o2Y!$LT4b2l
z58O8NhVhs#yA`P)xu=0gW}aXYvLYkz5Hf^ypL^X%m@cJ+Lv%wjgeVb87zs`0U7=ls
zelPI^%$B6oWb2hxCh?x^o;{b+Pvp49H{Z&;h*;?if?$@dZ#}k|Oxz9!kWv}jN@|dY
zfwm*3cpGr@WeV&I6%JD9Nz@5@eqN{0`7_yB^}A={$?R;1j9W6d5}Irf=6u^Sx!*Al
zlUJ1u_@Z4XEUe+@xbVSHHMM{r!>eHvD}OF3Mf@=?;uWkZ7HEpN0Fo^k2pjJ;BjF-!
zzPEH+S>&U;!;dOT$`ln_CjKN95n&P14e0<*g$SDvc|%EC*b`Wd$h8AEieTk%U12|$
z8vCf)IzFnWL@CU#j_W(E+F@Pf4v$-sK<kIQRblRIx2k)GN3TR~x7yq}sx_odveO0L
zP5dCqm&vP^<f|wpF4(XMj(DcQ=Hy#Y5{pq;(q_3P({##>ySLnUaGR1U1-362sa-^2
zkNs8-y)El%qqfiF-7o=giaUAv>I~NbyhP$8!DtdS$y}l&H$ryyx0_9Ju-$0DMW893
zCd%Dfv$4PZiqy9cs`Ix0w=ZFSo;vpnNePkMs)<ZEy;aC3CTvUlTKk~s=7;|R!ADe&

literal 0
HcmV?d00001

diff --git a/node.py b/node.py
new file mode 100644
index 0000000..3227d55
--- /dev/null
+++ b/node.py
@@ -0,0 +1,100 @@
+import os
+import json
+import numpy as np
+from PIL import ExifTags, Image, PngImagePlugin
+from comfy.cli_args import args
+import folder_paths
+
+class SaveImage:
+    def __init__(self):
+        self.output_dir = folder_paths.get_output_directory()
+        self.type = "output"
+        self.prefix_append = ""
+        self.quality = 95  # Default quality for WebP
+        self.method = 6    # Compression method for WebP
+        self.compress_level = 4  # Default PNG compression level
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "images": ("IMAGE", {"tooltip": "The images to save."}),
+                "filename_prefix": ("STRING", {"default": "ComfyUI", "tooltip": "The prefix for the file to save."})
+            },
+            "optional": {
+                "key_1": ("STRING", {"default": "", "tooltip": "Metadata key 1"}),
+                "metadata_1": ("STRING", {"default": "", "tooltip": "Metadata value 1"}),
+                "key_2": ("STRING", {"default": "", "tooltip": "Metadata key 2"}),
+                "metadata_2": ("STRING", {"default": "", "tooltip": "Metadata value 2"}),
+                "key_3": ("STRING", {"default": "", "tooltip": "Metadata key 3"}),
+                "metadata_3": ("STRING", {"default": "", "tooltip": "Metadata value 3"}),
+                "key_4": ("STRING", {"default": "", "tooltip": "Metadata key 4"}),
+                "metadata_4": ("STRING", {"default": "", "tooltip": "Metadata value 4"}),
+                "key_5": ("STRING", {"default": "", "tooltip": "Metadata key 5"}),
+                "metadata_5": ("STRING", {"default": "", "tooltip": "Metadata value 5"}),
+            },
+            "hidden": {
+                "unique_id": "UNIQUE_ID",
+                "extra_pnginfo": "EXTRA_PNGINFO",
+                "prompt": "PROMPT"
+            }
+        }
+
+
+
+    RETURN_TYPES = ()
+    FUNCTION = "save_images"
+    OUTPUT_NODE = True
+    CATEGORY = "image"
+    DESCRIPTION = "Saves the input images as PNG or WebP with dynamic metadata key-value pairs."
+
+    def save_images(self, images, filename_prefix="ComfyUI", unique_id=None, **kwargs):
+        filename_prefix += self.prefix_append
+        full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(
+            filename_prefix, self.output_dir, images[0].shape[1], images[0].shape[0]
+        )
+        results = list()
+
+        # Extract dynamic metadata from kwargs
+        dynamic_metadata = {}
+        for i in range(1, 6):
+            key = kwargs.get(f"key_{i}", "").strip()
+            value = kwargs.get(f"metadata_{i}", "").strip()
+            if key:
+                dynamic_metadata[key] = value
+
+        for batch_number, image in enumerate(images):
+            i = 255. * image.cpu().numpy()
+            img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
+            filename_with_batch_num = filename.replace("%batch_num%", str(batch_number))
+            file = f"{filename_with_batch_num}_{counter:05}_.png"
+
+            metadata = None
+            if not args.disable_metadata and dynamic_metadata:
+                metadata = PngImagePlugin.PngInfo()
+                for key, value in dynamic_metadata.items():
+                    metadata.add_text(key, value)
+
+            img.save(
+                os.path.join(full_output_folder, file),
+                pnginfo=metadata,
+                compress_level=self.compress_level
+            )
+
+            results.append({
+                "filename": file,
+                "subfolder": subfolder,
+                "type": self.type
+            })
+            counter += 1
+
+        return {"ui": {"images": results}}
+
+# Node registration
+NODE_CLASS_MAPPINGS = {
+    "SaveImage": SaveImage
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "SaveImage": "Save Image with Dynamic Metadata"
+}
\ No newline at end of file