from .nodes.grayscale_image import GrayscaleImage
from .nodes.brightness_transparency import BrightnessTransparency
from .nodes.binarize_image import BinarizeImage
from .nodes.binarize_image_using_otsu import BinarizeImageUsingOtsu
from .nodes.antialiasing_image import AntialiasingImage
from .nodes.remove_white_background_noise import RemoveWhiteBackgroundNoise
from .nodes.alpha_to_grayscale import AlphaToGrayscale
from .nodes.alpha_flatten import AlphaFlatten

NODE_CLASS_MAPPINGS = {
    "BrightnessTransparency": BrightnessTransparency,
    "BinarizeImage": BinarizeImage,
    "BinarizeImageUsingOtsu": BinarizeImageUsingOtsu,
    "GrayscaleImage": GrayscaleImage,
    "AntialiasingImage": AntialiasingImage,
    "RemoveWhiteBackgroundNoise": RemoveWhiteBackgroundNoise,
    "AlphaToGrayscale": AlphaToGrayscale,
    "AlphaFlatten": AlphaFlatten,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "BrightnessTransparency": "BrightnessTransparency",
    "BinarizeImage": "BinarizeImage",
    "BinarizeImageUsingOtsu": "BinarizeImageUsingOtsu",
    "GrayscaleImage": "GrayscaleImage",
    "AntialiasingImage": "AntialiasingImage",
    "RemoveWhiteBackgroundNoise": "RemoveWhiteBackgroundNoise",
    "AlphaToGrayscale": "AlphaToGrayscale",
    "AlphaFlatten": "AlphaFlatten",
}

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]