import os
import sys

current_dir = os.path.dirname(os.path.abspath(__file__))
if current_dir not in sys.path:
    sys.path.append(current_dir)

from aspect_outpaint_node import RandomAspectRatioMask
from image_mask_comparer import ImageMaskComparer

NODE_CLASS_MAPPINGS = {
    "RandomAspectRatioMask": RandomAspectRatioMask,
    "ImageMaskComparer": ImageMaskComparer
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "RandomAspectRatioMask": "Pad Image by Aspect for Outpaint",
    "ImageMaskComparer": "Image Mask Comparer",
}

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]