nodes/LevelPixel-ComfyUI-LevelPixel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changed the logic of Preview Image Bridge to speed up the display of images on the ui

Browse Source
This commit is contained in:
Level Pixel Dev
2025-05-30 21:31:59 +06:00
parent 83e4e0fcee
commit 561837524c
1 changed files with 35 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/io/image_outputs_LP.py
+35 -7
View File
@@ -15,13 +15,20 @@ class SaveImage:
self.type = "output"
self.prefix_append = ""
self.compress_level = 4
self.downscale_preview = False
self.size = 512
self.downscale_mode = True
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"images": ("IMAGE", {"tooltip": "The images to save."}),
"filename_prefix": ("STRING", {"default": "ComfyUI", "tooltip": "The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes."})
"filename_prefix": ("STRING", {"default": "ComfyUI", "tooltip": "The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes."}),
"downscale_preview": ("BOOLEAN", {"default": True, "label_on": "On", "label_off": "Off"}),
"size": ("INT", {"default": 512, "min": 1, "max": 8192, "step": 1}),
"downscale_mode": ("BOOLEAN", {"default": True, "label_on": "max", "label_off": "min"}),
"compress_level": ("INT", {"default": 1, "min": 0, "max": 9, "step": 1}),
},
"hidden": {
"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"
@@ -29,6 +36,7 @@ class SaveImage:
}
RETURN_TYPES = ("IMAGE",)
RETURN_NAMES = ("original IMAGE",)
FUNCTION = "save_images"
OUTPUT_NODE = False
@@ -36,13 +44,26 @@ class SaveImage:
CATEGORY = "LevelPixel/IO"
DESCRIPTION = "Saves the input images to your ComfyUI output directory."
def save_images(self, images, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=None):
def save_images(self, images, filename_prefix="ComfyUI", downscale_preview=False, size=512, downscale_mode=True, compress_level=1, prompt=None, extra_pnginfo=None):
self.downscale_preview = downscale_preview
self.size = size
self.downscale_mode = downscale_mode
self.compress_level = compress_level
filename_prefix += self.prefix_append
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir, images[0].shape[1], images[0].shape[0])
results = list()
for (batch_number, image) in enumerate(images):
i = 255. * image.cpu().numpy()
img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
if self.downscale_preview == True:
w, h = img.size
target = max(w, h) if self.downscale_mode else min(w, h)
if target > self.size:
scale = self.size / target
new_size = (round(w * scale), round(h * scale))
img = img.resize(new_size, Image.Resampling.LANCZOS)
metadata = None
if not args.disable_metadata:
metadata = PngInfo()
@@ -69,20 +90,27 @@ class PreviewImageForConditions(SaveImage):
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 = 1
self.compress_level = 0
self.downscale_preview = True
@classmethod
def INPUT_TYPES(s):
return {"required":
{"images": ("IMAGE", ), },
return {"required":{
"images": ("IMAGE", ),
"size": ("INT", {"default": 512, "min": 1, "max": 8192, "step": 1}),
"downscale_preview": ("BOOLEAN", {"default": True, "label_on": "On", "label_off": "Off"}),
"size": ("INT", {"default": 512, "min": 1, "max": 8192, "step": 1}),
"downscale_mode": ("BOOLEAN", {"default": True, "label_on": "max", "label_off": "min"}),
"compress_level": ("INT", {"default": 0, "min": 0, "max": 9, "step": 1}),
},
"hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"},
}
NODE_CLASS_MAPPINGS = {
"PreviewImageForConditions|LP": PreviewImageForConditions,
"PreviewImageForConditions|LP": PreviewImageForConditions
}
NODE_DISPLAY_NAME_MAPPINGS = {
"PreviewImageForConditions|LP": "Preview Image Bridge [LP]",
"PreviewImageForConditions|LP": "Preview Image Bridge [LP]"
}