From 5d2ade4fd4864c04ff967625bfb4905156593b8b Mon Sep 17 00:00:00 2001
From: KarrixLee <52952301+KarrixLee@users.noreply.github.com>
Date: Mon, 22 Sep 2025 18:18:40 +0800
Subject: [PATCH] Karrix/add remove metadata boolean in save image (#109)

* fix: remove metadata in output node

* set to true

* refactor
---
 comfy-nodes/output_image.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/comfy-nodes/output_image.py b/comfy-nodes/output_image.py
index 24bb197..f7f3e57 100644
--- a/comfy-nodes/output_image.py
+++ b/comfy-nodes/output_image.py
@@ -33,6 +33,7 @@ class ComfyDeployOutputImage:
                     "STRING",
                     {"multiline": False, "default": "output_images"},
                 ),
+                "remove_metadata": ("BOOLEAN", {"default": True}),
             },
             "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
         }
@@ -53,6 +54,7 @@ class ComfyDeployOutputImage:
         output_id="output_images",
         prompt=None,
         extra_pnginfo=None,
+        remove_metadata=True,
     ):
         filename_prefix += self.prefix_append
         full_output_folder, filename, counter, subfolder, filename_prefix = (
@@ -65,11 +67,12 @@ class ComfyDeployOutputImage:
             i = 255.0 * image.cpu().numpy()
             img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
             metadata = PngInfo()
-            if prompt is not None:
-                metadata.add_text("prompt", json.dumps(prompt))
-            if extra_pnginfo is not None:
-                for x in extra_pnginfo:
-                    metadata.add_text(x, json.dumps(extra_pnginfo[x]))
+            if not remove_metadata:
+                if prompt is not None:
+                    metadata.add_text("prompt", json.dumps(prompt))
+                if extra_pnginfo is not None:
+                    for x in extra_pnginfo:
+                        metadata.add_text(x, json.dumps(extra_pnginfo[x]))
 
             filename_with_batch_num = filename.replace("%batch_num%", str(batch_number))
             file = f"{filename_with_batch_num}_{counter:05}_.{file_type}"