From 6c6dcd5647affcd40b1843081324e8cdef04b094 Mon Sep 17 00:00:00 2001
From: solidlime <solidlime@gmail.com>
Date: Wed, 7 Jan 2026 21:53:40 +0900
Subject: [PATCH] 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
---
 random_image_picker.py | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/random_image_picker.py b/random_image_picker.py
index 94c5b51..8e2c01e 100644
--- a/random_image_picker.py
+++ b/random_image_picker.py
@@ -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)}")