from .nodes import (
    ImageToBase64,
    Base64ToImage,
    AudioToBase64,
    FirstFrameToBase64,
    LastFrameToBase64,
    SaveBase64Websocket,
)

NODE_CLASS_MAPPINGS = {
    "ImageToBase64": ImageToBase64,
    "Base64ToImage": Base64ToImage,
    "Base64ToAudio": AudioToBase64,
    "FirstFrameToBase64": FirstFrameToBase64,
    "LastFrameToBase64": LastFrameToBase64,
    "SaveBase64Websocket": SaveBase64Websocket,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "ImageToBase64": "Image => Base64",
    "Base64ToImage": "Base64 => Image",
    "Base64ToAudio": "Base64 => Audio",
    "FirstFrameToBase64": "First Frame => Base64",
    "LastFrameToBase64": "Last Frame => Base64",
    "SaveBase64Websocket": "Save Base64 (Websocket)",
}