Hey there
I'm trying to write a custom node for Comfy that:
1.- Receives an image
2.- Receives an optional string text marked as "Author"
3.- Receives an optional string text marked as "Title"
4.- Receives an optional string text marked as "Subject"
5.- Receives an optional string text marked as "Tags"
6.- Have an option for an output subfolder
7.- Saves the image in JPG format (100 quality), filling the right EXIF metadata fields with the text provided in points 2, 3, 4 and 5
8.- The filename should be the day it was created, in the format YYYY/MM/DD, with a four digit numeral, to ensure that every new file has a diferent filename
The problem is, even when the node appears in ComfyUI, it does not save any image nor create any subfolder. I'm not a programmer at all, so maybe I'm doing something completely stupid here. Any clues?
Note: If it's important, I'm working with the portable version of Comfy, on an embedded Python. I also have Pillow installed here, so that shouldn't be a problem
This is the code I have so far:
import os
import datetime
from PIL import Image, TiffImagePlugin
import numpy as np
import folder_paths
import traceback
class SaveImageWithExif:
u/classmethod
def INPUT_TYPES(cls):
return {
"required": {
"image": ("IMAGE",),
},
"optional": {
"author": ("STRING", {"default": "Author"}),
"title": ("STRING", {"default": "Title"}),
"subject": ("STRING", {"default": "Description"}),
"tags": ("STRING", {"default": "Keywords"}),
"subfolder": ("STRING", {"default": "Subfolder"}),
}
}
RETURN_TYPES = ("STRING",) # Must match return type
FUNCTION = "save_image"
CATEGORY = "image/save"
def encode_utf16le(self, text):
return text.encode('utf-16le') + b'\x00\x00'
def save_image(self, image, author="", title="", subject="", tags="", subfolder=""):
print("[SaveImageWithExif] save_image() called")
print(f"Author: {author}, Title: {title}, Subject: {subject}, Tags: {tags}, Subfolder: {subfolder}")
try:
print(f"Image type: {type(image)}, len: {len(image)}")
image = image
img = Image.fromarray(np.clip(255.0 * image, 0, 255).astype(np.uint8))
output_base = folder_paths.get_output_directory()
print(f"Output directory base: {output_base}")
today = datetime.datetime.now()
base_path = os.path.join(output_base, subfolder)
dated_folder = os.path.join(base_path, today.strftime("%Y/%m/%d"))
os.makedirs(dated_folder, exist_ok=True)
counter = 1
while True:
filename = f"{counter:04d}.jpg"
filepath = os.path.join(dated_folder, filename)
if not os.path.exists(filepath):
break
counter += 1
exif_dict = TiffImagePlugin.ImageFileDirectory_v2()
if author:
exif_dict[315] = author
if title:
exif_dict[270] = title
if subject:
exif_dict[40091] = self.encode_utf16le(subject)
if tags:
exif_dict[40094] = self.encode_utf16le(tags)
img.save(filepath, "JPEG", quality=100, exif=exif_dict.tobytes())
print(f"[SaveImageWithExif] Image saved to: {filepath}")
return (f"Saved to {filepath}",)
except Exception as e:
print("[SaveImageWithExif] Error:")
traceback.print_exc()
return ("Error saving image",)
NODE_CLASS_MAPPINGS = {
"SaveImageWithExif": SaveImageWithExif
}
NODE_DISPLAY_NAME_MAPPINGS = {
"SaveImageWithExif": "Save Image with EXIF Metadata"
}
