33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import os
|
|
|
|
class BatchImageSaveNode:
|
|
RETURN_TYPES = ("IMAGE",)
|
|
FUNCTION = "execute"
|
|
CATEGORY = "Image Processing"
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"images": ("IMAGE",),
|
|
"extension": ([".png", ".ico", ".jpeg", ".gif"],),
|
|
}
|
|
}
|
|
|
|
def execute(self, images, save_path=None, extension=".png"):
|
|
# Set default save path
|
|
default_save_path = os.path.join("C:", "ComfyUI_windows_portable", "ComfyUI", "output")
|
|
|
|
# Check if the default path exists and is writable
|
|
if not os.path.exists(default_save_path):
|
|
os.makedirs(default_save_path, exist_ok=True) # Create directory if it doesn't exist
|
|
|
|
saved_files = []
|
|
for i, image in enumerate(images):
|
|
filename = f"image_{i}.{extension.strip('.')}"
|
|
full_path = os.path.join(default_save_path, filename)
|
|
# Example save operation
|
|
image.save(full_path) # Replace with actual saving logic
|
|
saved_files.append(full_path)
|
|
return (saved_files,)
|