notid/ComfyUI-compass-paths
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Preview only the first image in batch

Browse Source
This commit is contained in:
Bryce
2026-04-22 15:57:33 -07:00
parent 16e05da2b4
commit a6b1d1027c
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
compass_image_loader.py
View File

@@ -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},