from .nodes.text_to_speech import TextToSpeech
from .nodes.voice_cloning import VoiceCloning
from .nodes.load_audio import JM_LoadAudio

NODE_CLASS_MAPPINGS = {
    "JM-MiniMax-API/text-to-speech": TextToSpeech,
    "JM-MiniMax-API/voice-cloning": VoiceCloning,
    "JM-MiniMax-API/load-audio": JM_LoadAudio
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "JM-MiniMax-API/text-to-speech": "MiniMax Text to Speech",
    "JM-MiniMax-API/voice-cloning": "MiniMax Voice Cloning",
    "JM-MiniMax-API/load-audio": "Load Audio"
}

# Tell ComfyUI where to find web extensions
WEB_DIRECTORY = "./web"

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]