From a6b1d1027cc7b3e42c16ce637b15d893e75f7baa Mon Sep 17 00:00:00 2001 From: Bryce Date: Wed, 22 Apr 2026 15:57:33 -0700 Subject: [PATCH] Preview only the first image in batch --- compass_image_loader.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/compass_image_loader.py b/compass_image_loader.py index 570a2df..677489b 100644 --- a/compass_image_loader.py +++ b/compass_image_loader.py @@ -167,13 +167,11 @@ class CompassImageLoader: temp_dir = folder_paths.get_temp_directory() prefix = "_compass_" + "".join(random.choice("abcdefghijklmnopqrstupvxyz") for _ in range(5)) - results = [] - for batch_number, img_tensor in enumerate(image_batch): - i = 255.0 * img_tensor.cpu().numpy() - pil_img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8)) - file = f"{prefix}_{batch_number:05}_.png" - pil_img.save(os.path.join(temp_dir, file), compress_level=1) - results.append({"filename": file, "subfolder": "", "type": "temp"}) + i = 255.0 * image_batch[0].cpu().numpy() + pil_img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8)) + file = f"{prefix}_00000_.png" + pil_img.save(os.path.join(temp_dir, file), compress_level=1) + results = [{"filename": file, "subfolder": "", "type": "temp"}] return { "ui": {"images": results},