from .Iter_TextSwitch import Iter_TextSwitch
from .Iter_CondClipSwitch import Iter_ConditioningSwitch, Iter_CLIPSwitch
from .Iter_ImageSwitch import Iter_ImageSwitch
from .Iter_ModelSwitch import Iter_ModelSwitch
from .Iter_FloatSwitch import Iter_FloatSwitch

NODE_CLASS_MAPPINGS = {
    "Iter_TextSwitch": Iter_TextSwitch,
    "Iter_ConditioningSwitch": Iter_ConditioningSwitch,
    "Iter_CLIPSwitch": Iter_CLIPSwitch,
    "Iter_ImageSwitch": Iter_ImageSwitch,
    "Iter_ModelSwitch": Iter_ModelSwitch,
    "Iter_FloatSwitch": Iter_FloatSwitch
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "Iter_TextSwitch": "Iter_TextSwitch",
    "Iter_ConditioningSwitch": "Iter_ConditioningSwitch",
    "Iter_CLIPSwitch": "Iter_CLIPSwitch",
    "Iter_ImageSwitch": "Iter_ImageSwitch",
    "Iter_ModelSwitch": "Iter_ModelSwitch",
    "Iter_FloatSwitch": "Iter_FloatSwitch"
}

# 確保 JS 資料夾被載入
WEB_DIRECTORY = "./web"

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