"""ComfyUI-VideoToText: Video → Audio → MiMo-V2.5-ASR → JSON pipeline helpers.

This pack provides the *non-ASR* pieces of a video-to-text pipeline. The ASR itself
is handled by ComfyUI-MiMoASR (https://github.com/aadebuger/ComfyUI-MiMoASR), which
must be installed separately.
"""

from .nodes.load_video import LoadVideoForASR
from .nodes.extract_audio import ExtractAudio
from .nodes.audio_from_path import AudioFromPath
from .nodes.build_json import BuildTranscriptJSON
from .nodes.save_transcript import SaveTranscript


NODE_CLASS_MAPPINGS = {
    "VTT_LoadVideo": LoadVideoForASR,
    "VTT_ExtractAudio": ExtractAudio,
    "VTT_AudioFromPath": AudioFromPath,
    "VTT_BuildTranscriptJSON": BuildTranscriptJSON,
    "VTT_SaveTranscript": SaveTranscript,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "VTT_LoadVideo": "🎙️ Load Video (ASR)",
    "VTT_ExtractAudio": "🎙️ Extract Audio (ffmpeg)",
    "VTT_AudioFromPath": "🎙️ Audio From Path",
    "VTT_BuildTranscriptJSON": "🎙️ Build Transcript JSON",
    "VTT_SaveTranscript": "🎙️ Save Transcript JSON",
}

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