Upload 3 files

ComfyUI-Metadata-Saver/__init__.py

@@ -0,0 +1,11 @@
+from .metadata_saver import SaveImageCustomMetadata
+
+NODE_CLASS_MAPPINGS = {
+    "SaveImageCustomMetadata": SaveImageCustomMetadata
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "SaveImageCustomMetadata": "Save Image (Custom Metadata)"
+}
+
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']

ComfyUI-Metadata-Saver/metadata_saver.py

@@ -0,0 +1,139 @@
+import os
+import json
+import numpy as np
+from PIL import Image, ExifTags
+from PIL.PngImagePlugin import PngInfo
+import folder_paths
+
+class SaveImageCustomMetadata:
+    def __init__(self):
+        self.output_dir = folder_paths.get_output_directory()
+        self.type = "output"
+        self.prefix_append = ""
+        self.compress_level = 4
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "images": ("IMAGE", ),
+                "filename_prefix": ("STRING", {"default": "ComfyUI"}),
+                "strip_workflow": ("BOOLEAN", {"default": True, "label": "Remove Original Workflow"}),
+                
+                # PNG Text Info
+                "custom_author": ("STRING", {"default": "Me", "multiline": False}),
+                "copyright_info": ("STRING", {"default": "Copyright 2024", "multiline": False}),
+                
+                # EXIF Data (Camera & GPS)
+                "camera_make": ("STRING", {"default": "Apple"}),
+                "camera_model": ("STRING", {"default": "iPhone 16 Pro Max"}),
+                
+                # Default: Miami, FL (25.7617, -80.1918)
+                "latitude": ("FLOAT", {"default": 25.7617, "min": -90.0, "max": 90.0, "step": 0.0001}),
+                "longitude": ("FLOAT", {"default": -80.1918, "min": -180.0, "max": 180.0, "step": 0.0001}),
+            },
+            "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
+        }
+
+    RETURN_TYPES = ()
+    FUNCTION = "save_images"
+    OUTPUT_NODE = True
+    CATEGORY = "image"
+
+    def _convert_to_degrees(self, value):
+        """
+        Helper to convert decimal coordinates to EXIF (Degrees, Minutes, Seconds) format.
+        EXIF expects rational numbers (numerator, denominator).
+        """
+        d = int(value)
+        m = int((value - d) * 60)
+        s = (value - d - m/60) * 3600.00
+        # precision of 100 for seconds
+        return ((d, 1), (m, 1), (int(s * 100), 100))
+
+    def save_images(self, images, filename_prefix="ComfyUI", strip_workflow=True, 
+                    custom_author="", copyright_info="", 
+                    camera_make="", camera_model="", latitude=0.0, longitude=0.0,
+                    prompt=None, extra_pnginfo=None):
+        
+        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()
+        
+        # EXIF Standard Tag IDs
+        TAG_MAKE = 271
+        TAG_MODEL = 272
+        TAG_SOFTWARE = 305
+        TAG_ARTIST = 315
+        TAG_COPYRIGHT = 33432
+        TAG_GPS_INFO = 34853
+        
+        # GPS Sub-IFD Tags
+        GPS_LAT_REF = 1
+        GPS_LAT = 2
+        GPS_LONG_REF = 3
+        GPS_LONG = 4
+
+        for image in images:
+            i = 255. * image.cpu().numpy()
+            img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
+            
+            # --- 1. Handle PNG Text Metadata (Workflow, Author) ---
+            metadata = PngInfo()
+            if not strip_workflow:
+                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 custom_author:
+                metadata.add_text("Author", custom_author)
+            if copyright_info:
+                metadata.add_text("Copyright", copyright_info)
+
+            # --- 2. Handle EXIF Data (Camera & GPS) ---
+            exif_data = img.getexif()
+
+            # Set Camera Details
+            if camera_make:
+                exif_data[TAG_MAKE] = camera_make
+            if camera_model:
+                exif_data[TAG_MODEL] = camera_model
+            
+            exif_data[TAG_SOFTWARE] = "ComfyUI Custom Node"
+            if custom_author:
+                exif_data[TAG_ARTIST] = custom_author
+            if copyright_info:
+                exif_data[TAG_COPYRIGHT] = copyright_info
+
+            # Set GPS Data
+            # We must access the GPS IFD (Sub-dictionary) specifically
+            gps_ifd = exif_data.get_ifd(TAG_GPS_INFO)
+            
+            # Latitude
+            gps_ifd[GPS_LAT_REF] = 'N' if latitude >= 0 else 'S'
+            gps_ifd[GPS_LAT] = self._convert_to_degrees(abs(latitude))
+
+            # Longitude
+            gps_ifd[GPS_LONG_REF] = 'E' if longitude >= 0 else 'W'
+            gps_ifd[GPS_LONG] = self._convert_to_degrees(abs(longitude))
+
+            # Save the image with both PNG Info and EXIF binary data
+            file = f"{filename}_{counter:05}_.png"
+            img.save(
+                os.path.join(full_output_folder, file), 
+                pnginfo=metadata, 
+                exif=exif_data, 
+                compress_level=self.compress_level
+            )
+            
+            results.append({
+                "filename": file,
+                "subfolder": subfolder,
+                "type": self.type
+            })
+            counter += 1
+
+        return { "ui": { "images": results } }

ComfyUI-Metadata-Saver/📸 ComfyUI Custom Metadata Saver.txt

@@ -0,0 +1,25 @@
+📸 ComfyUI Custom Metadata Saver
+This custom node allows you to save images while stripping the internal ComfyUI workflow data and injecting custom EXIF (Camera & GPS) and Copyright information.
+📥 Installation
+Navigate to your ComfyUI/custom_nodes/ folder.
+Create a new folder named ComfyUI-Metadata-Saver.
+Create two files inside: __init__.py and metadata_saver.py (paste the provided code).
+Restart ComfyUI.
+⚙️ How to Use
+Add Node: Double-click the canvas and search for "Save Image (Custom Metadata)".
+Connect: Link your final image output to this node (replace the standard "Save Image" node).
+Configure Settings:
+Setting	Description
+strip_workflow	True: Removes the ComfyUI node graph/prompt (privacy mode).<br>False: Keeps the workflow hidden in the file.
+filename_prefix	Standard file naming (e.g., "MyRender").
+custom_author	Sets the "Artist/Author" tag.
+camera_make	e.g., Apple, Sony, Canon.
+camera_model	e.g., iPhone 16 Pro Max.
+latitude / longitude	Sets GPS location.
+📍 Example Presets (Miami / iPhone)
+To make your render look like a photo taken in Miami on a new iPhone:
+Camera Make: Apple
+Camera Model: iPhone 16 Pro Max
+Latitude: 25.7617
+Longitude: -80.1918
+⚠️ Note: If you set strip_workflow to True, you will not be able to drag and drop the resulting image back into ComfyUI to load the workflow. The image will be "clean."