from .character_reader import NODE_CLASS_MAPPINGS as CR_CLASS, NODE_DISPLAY_NAME_MAPPINGS as CR_DISPLAY
from .resolution_reader import NODE_CLASS_MAPPINGS as RR_CLASS, NODE_DISPLAY_NAME_MAPPINGS as RR_DISPLAY
from .reenforcer import NODE_CLASS_MAPPINGS as RE_CLASS, NODE_DISPLAY_NAME_MAPPINGS as RE_DISPLAY
from .lora_from_string import NODE_CLASS_MAPPINGS as LFS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as LFS_DISPLAY
from .lora_selector import NODE_CLASS_MAPPINGS as LS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as LS_DISPLAY
from .checkpoint_selector import NODE_CLASS_MAPPINGS as CS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as CS_DISPLAY
from .checkpoint_from_string import NODE_CLASS_MAPPINGS as CFS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as CFS_DISPLAY
from .metadata_saver import NODE_CLASS_MAPPINGS as MS_CLASS, NODE_DISPLAY_NAME_MAPPINGS as MS_DISPLAY

NODE_CLASS_MAPPINGS = {
    **CR_CLASS,
    **RR_CLASS,
    **RE_CLASS,
    **LFS_CLASS,
    **LS_CLASS,
    **CS_CLASS,
    **CFS_CLASS,
    **MS_CLASS
}

NODE_DISPLAY_NAME_MAPPINGS = {
    **CR_DISPLAY,
    **RR_DISPLAY,
    **RE_DISPLAY,
    **LFS_DISPLAY,
    **LS_DISPLAY,
    **CS_DISPLAY,
    **CFS_DISPLAY,
    **MS_DISPLAY
}

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']