from .lumaai_api_node import (
    LumaAIClient,
    Text2Video,
    Image2Video,
    InterpolateGenerations,
    ExtendGeneration,
    PreviewVideo,
    Reference,
    ConcatReferences,
    CharacterReference,
    ImageGeneration,
    ModifyImage,
)
from .imgbb_node import ImgBBUpload

NODE_CLASS_MAPPINGS = {
    "LumaAIClient": LumaAIClient,
    "ImgBBUpload": ImgBBUpload,
    "LumaText2Video": Text2Video,
    "LumaImage2Video": Image2Video,
    "LumaInterpolateGenerations": InterpolateGenerations,
    "LumaExtendGeneration": ExtendGeneration,
    "LumaPreviewVideo": PreviewVideo,
    "Reference": Reference,
    "ConcatReferences": ConcatReferences,
    "CharacterReference": CharacterReference,
    "LumaImageGeneration": ImageGeneration,
    "LumaModifyImage": ModifyImage,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "LumaAIClient": "LumaAI Client",
    "ImgBBUpload": "ImgBB Upload",
    "LumaText2Video": "Text to Video",
    "LumaImage2Video": "Image to Video",
    "LumaInterpolateGenerations": "Interpolate Generations",
    "LumaExtendGeneration": "Extend Generation",
    "LumaPreviewVideo": "LumaAI Preview Video",
    "LumaReference": "Reference",
    "LumaConcatReferences": "Concat References",
    "LumaCharacterReference": "Character Reference",
    "LumaImageGeneration": "Image Generation",
    "LumaModifyImage": "Modify Image",
}