nodes/solidlime-Comfyui-Random_Image_Picker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix: Return both preview and IMAGE tensor

Browse Source 
Changed return format to:
{'ui': preview['ui'], 'result': (img_tensor,)}

This allows:
- Preview display through save_images()
- IMAGE output for downstream nodes

Now both preview AND output work correctly!
This commit is contained in:
solidlime
2026-01-07 22:00:38 +09:00
parent 6c6dcd5647
commit 3db299a60a
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
random_image_picker.py
+7 -5
View File
@@ -142,7 +142,7 @@ class RandomImagePicker(SaveImage):
image_data: Base64 encoded image data from file picker (used when folder_mode=False)
Returns:
Tuple of (image_tensor,)
Dict with ui (for preview) and result (IMAGE tensor)
"""
if folder_mode:
# Folder mode: random selection from folder_path
@@ -168,8 +168,9 @@ class RandomImagePicker(SaveImage):
img_tensor, width, height = self.load_image_file(selected_image)
print(f"[Random Image Picker] Resolution: {width}x{height}")
# Save and return for preview
return self.save_images(img_tensor, filename_prefix=f"RandomPicker")
# Save for preview and return with image tensor
preview = self.save_images(img_tensor, filename_prefix="RandomPicker")
return {"ui": preview["ui"], "result": (img_tensor,)}
else:
# Single file mode: use file picker (image_data)
if not image_data:
@@ -206,8 +207,9 @@ class RandomImagePicker(SaveImage):
print(f"[Random Image Picker] Mode: Single (File Picker)")
print(f"[Random Image Picker] Resolution: {width}x{height}")
# Save and return for preview
return self.save_images(img_tensor, filename_prefix=f"RandomPicker")
# Save for preview and return with image tensor
preview = self.save_images(img_tensor, filename_prefix="RandomPicker")
return {"ui": preview["ui"], "result": (img_tensor,)}
except Exception as e:
raise ValueError(f"Failed to load image from file picker: {str(e)}")