From 073b1c159bc3768dfabca000b7fde7748f574894 Mon Sep 17 00:00:00 2001 From: solidlime Date: Wed, 7 Jan 2026 21:50:14 +0900 Subject: [PATCH] Simplify: Return plain tuple instead of dict for preview Changed return format from dict to simple tuple (img_tensor,) OUTPUT_NODE = True should auto-display preview with IMAGE type. Removed UI dict format that may have been preventing preview. --- random_image_picker.py | 19 +++++++------------ 1 file changed, 7 insertions(+), 12 deletions(-) diff --git a/random_image_picker.py b/random_image_picker.py index df3a009..94c5b51 100644 --- a/random_image_picker.py +++ b/random_image_picker.py @@ -135,7 +135,7 @@ class RandomImagePicker: image_data: Base64 encoded image data from file picker (used when folder_mode=False) Returns: - Tuple of (image_tensor,) with UI info + Tuple of (image_tensor,) """ if folder_mode: # Folder mode: random selection from folder_path @@ -159,12 +159,10 @@ class RandomImagePicker: print(f"[Random Image Picker] Selected: {selected_image}") img_tensor, width, height = self.load_image_file(selected_image) + print(f"[Random Image Picker] Resolution: {width}x{height}") - # Return with UI info - return { - "ui": {"text": [f"📁 Folder Mode | Resolution: {width}x{height} | File: {os.path.basename(selected_image)}"]}, - "result": (img_tensor,) - } + # Return just the image tensor + return (img_tensor,) else: # Single file mode: use file picker (image_data) if not image_data: @@ -199,13 +197,10 @@ class RandomImagePicker: img_tensor = torch.from_numpy(img_array)[None,] print(f"[Random Image Picker] Mode: Single (File Picker)") - print(f"[Random Image Picker] Loaded: {width}x{height}") + print(f"[Random Image Picker] Resolution: {width}x{height}") - # Return with UI info - return { - "ui": {"text": [f"📄 Single Mode | Resolution: {width}x{height}"]}, - "result": (img_tensor,) - } + # Return just the image tensor + return (img_tensor,) except Exception as e: raise ValueError(f"Failed to load image from file picker: {str(e)}")