Spaces:
Running
Running
from nodes import PreviewImage | |
from .constants import get_category, get_name | |
class RgthreeImageComparer(PreviewImage): | |
"""A node that compares two images in the UI.""" | |
NAME = get_name('Image Comparer') | |
CATEGORY = get_category() | |
FUNCTION = "compare_images" | |
def INPUT_TYPES(cls): # pylint: disable = invalid-name, missing-function-docstring | |
return { | |
"required": {}, | |
"optional": { | |
"image_a": ("IMAGE",), | |
"image_b": ("IMAGE",), | |
}, | |
"hidden": { | |
"prompt": "PROMPT", | |
"extra_pnginfo": "EXTRA_PNGINFO" | |
}, | |
} | |
def compare_images(self, | |
image_a=None, | |
image_b=None, | |
filename_prefix="rgthree.compare.", | |
prompt=None, | |
extra_pnginfo=None): | |
result = { "ui": { "a_images":[], "b_images": [] } } | |
if image_a is not None and len(image_a) > 0: | |
result['ui']['a_images'] = self.save_images(image_a, filename_prefix, prompt, extra_pnginfo)['ui']['images'] | |
if image_b is not None and len(image_b) > 0: | |
result['ui']['b_images'] = self.save_images(image_b, filename_prefix, prompt, extra_pnginfo)['ui']['images'] | |
return result |