Issue re-using ImageComparer

[erik-saltwell]

I am trying to use the code in ImageComparer to implement my own comparer with some changes. I included the code below, The only changes i think i have made is to use constants for name/category, and put a printf in the compare_images function.

The node loads, and i can see from the printf that it gets called. I checked the temp directory and see the two temp files, but i do not see a preview image or comparison ability in comfy. from inside comfy the node acts like it doesn't do anything (but the printf).

If anyone knows what I am doing wrong, i would love some advice.

Code in my node class:

from nodes import PreviewImage

class SaltyImageCompareeClass(PreviewImage):
  """A node that compares two images in the UI."""

  NAME = "SaltyImageCompare"
  CATEGORY = "Saltworks"
  FUNCTION = "compare_images"

  @classmethod
  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):
    print("**IN COMPARE IMAGES")
    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

---

and the code in my init file:

from .image_sizer import *;
from .compare_to_prior import *
NODE_CLASS_MAPPINGS = {
    "SaltyImageSizer": ImageSizeClass,
    "SaltyImageCompare": SaltyImageCompareeClass,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "SaltyImageSizer": "Salty Image Sizer",
    "SaltyImageCompare": "Salty Image Compare",
}

[erik-saltwell]

sorry, i screwed up the formatting, just to be clear, i did include the following line at the top of my class:
from nodes import PreviewImage