# Dumas Long Videos wrapper.
#
# The Long Videos implementation is vendored from:
# https://huggingface.co/Smite79/MiniMax-H3-Longvideos
#
# Keep this wrapper small so the upstream sampler can be refreshed without
# touching the other Dumas helper/reference/prompt nodes.

try:
    from .dumas_h3_longvideos_upstream import H3LongVideos
except ImportError:  # pragma: no cover - direct module import in local tests
    from dumas_h3_longvideos_upstream import H3LongVideos


NODE_CLASS_MAPPINGS = {
    "DumasH3LongVideos": H3LongVideos,
    "H3LongVideos": H3LongVideos,
    "H3LongVideosFL2VA": H3LongVideos,
    "H3LongVideosV1": H3LongVideos,
    "H3LongVideosREF2VA": H3LongVideos,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "DumasH3LongVideos": "Dumas H3 Long Videos",
    "H3LongVideos": "H3 Long Videos",
    "H3LongVideosFL2VA": "H3 Long Videos",
    "H3LongVideosV1": "H3 Long Videos",
    "H3LongVideosREF2VA": "H3 Long Videos",
}


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