Fix: Add __init__ for SaveImage compatibility

Added __init__ method to properly initialize SaveImage parent class.
This should enable preview display through save_images() method.

Settings:
- output_dir: temp directory
- type: temp
- prefix_append: random temp prefix
- compress_level: 4
This commit is contained in:
solidlime
2026-01-07 21:53:40 +09:00
parent 073b1c159b
commit 6c6dcd5647
+12 -5
View File
@@ -10,14 +10,21 @@ import folder_paths
from PIL import Image
import numpy as np
import torch
from nodes import SaveImage
class RandomImagePicker:
class RandomImagePicker(SaveImage):
"""
ComfyUI node that loads either a single image or a random image from a folder.
Supports subfolder scanning and outputs image with resolution information.
"""
def __init__(self):
self.output_dir = folder_paths.get_temp_directory()
self.type = "temp"
self.prefix_append = "_temp_" + ''.join(random.choice("abcdefghijklmnopqrstupvxyz") for x in range(5))
self.compress_level = 4
@classmethod
def INPUT_TYPES(cls):
# Get default input directory from ComfyUI
@@ -161,8 +168,8 @@ class RandomImagePicker:
img_tensor, width, height = self.load_image_file(selected_image)
print(f"[Random Image Picker] Resolution: {width}x{height}")
# Return just the image tensor
return (img_tensor,)
# Save and return for preview
return self.save_images(img_tensor, filename_prefix=f"RandomPicker")
else:
# Single file mode: use file picker (image_data)
if not image_data:
@@ -199,8 +206,8 @@ class RandomImagePicker:
print(f"[Random Image Picker] Mode: Single (File Picker)")
print(f"[Random Image Picker] Resolution: {width}x{height}")
# Return just the image tensor
return (img_tensor,)
# Save and return for preview
return self.save_images(img_tensor, filename_prefix=f"RandomPicker")
except Exception as e:
raise ValueError(f"Failed to load image from file picker: {str(e)}")