"""Standalone list-index selector nodes for ComfyUI."""

from .nodes import (
    AudioFromListIdx,
    BytesFromListIdx,
    ConditioningFromListIdx,
    DictFromListIdx,
    FloatFromListIdx,
    ImageFromListIdx,
    IntFromListIdx,
    LatentFromListIdx,
    StrFromListIdx,
)


NODE_CLASS_MAPPINGS = {
    "ImageFromListIdx": ImageFromListIdx,
    "LatentFromListIdx": LatentFromListIdx,
    "ConditioningFromListIdx": ConditioningFromListIdx,
    "AudioFromListIdx": AudioFromListIdx,
    "DictFromListIdx": DictFromListIdx,
    "StrFromListIdx": StrFromListIdx,
    "BytesFromListIdx": BytesFromListIdx,
    "IntFromListIdx": IntFromListIdx,
    "FloatFromListIdx": FloatFromListIdx,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "AudioFromListIdx": "Audio From List Index",
}

__all__ = [
    "AudioFromListIdx",
    "BytesFromListIdx",
    "ConditioningFromListIdx",
    "DictFromListIdx",
    "FloatFromListIdx",
    "ImageFromListIdx",
    "IntFromListIdx",
    "LatentFromListIdx",
    "NODE_CLASS_MAPPINGS",
    "NODE_DISPLAY_NAME_MAPPINGS",
    "StrFromListIdx",
]