Create BatchImageSaveNode.py
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
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,)
|
||||
Reference in New Issue
Block a user