import os

import folder_paths

# models/minimax_h3_adapter/  -> adapter_stage2.safetensors
folder_paths.add_model_folder_path("minimax_h3_adapter",
                                   os.path.join(folder_paths.models_dir, "minimax_h3_adapter"))
# models/text_encoders/       -> Qwen3.5-4B 文件夹（与 minimax CLIP 同目录）
folder_paths.add_model_folder_path("minimax_h3_student",
                                   os.path.join(folder_paths.models_dir, "text_encoders"))

from .nodes import MiniMaxH3AdapterLoader, MiniMaxH3AdapterFromCLIPLoader  # noqa: E402
from .export_cond import MiniMaxH3ExportCond  # noqa: E402

NODE_CLASS_MAPPINGS = {
    "MiniMaxH3AdapterLoader": MiniMaxH3AdapterLoader,
    "MiniMaxH3AdapterFromCLIPLoader": MiniMaxH3AdapterFromCLIPLoader,
    "MiniMaxH3ExportCond": MiniMaxH3ExportCond,
}
NODE_DISPLAY_NAME_MAPPINGS = {
    "MiniMaxH3AdapterLoader": "MiniMax H3 Adapter Loader",
    "MiniMaxH3AdapterFromCLIPLoader": "MiniMax H3 Adapter (From CLIP)",
    "MiniMaxH3ExportCond": "MiniMax H3 Export Cond (Debug)",
}

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]