Files
relhamdi-ComfyUI-RE-CustomU…/__init__.py
T

63 lines
2.1 KiB
Python

"""Top-level package for the node pack."""
from src.nodes.prompt_layout_filler import (
NODE_CLASS_MAPPINGS as PromptLayoutFillerMappings,
)
from src.nodes.prompt_layout_filler import (
NODE_DISPLAY_NAME_MAPPINGS as PromptLayoutFillerDisplayMappings,
)
from src.nodes.prompt_option_picker import (
NODE_CLASS_MAPPINGS as PromptOptionPickerMappings,
)
from src.nodes.prompt_option_picker import (
NODE_DISPLAY_NAME_MAPPINGS as PromptOptionPickerDisplayMappings,
)
from src.nodes.prompt_preset_selector import (
NODE_CLASS_MAPPINGS as PromptPresetSelectorMappings,
)
from src.nodes.prompt_preset_selector import (
NODE_DISPLAY_NAME_MAPPINGS as PromptPresetSelectorDisplayMappings,
)
from src.nodes.prompt_router import (
NODE_CLASS_MAPPINGS as PromptRouterMappings,
)
from src.nodes.prompt_router import (
NODE_DISPLAY_NAME_MAPPINGS as PromptRouterDisplayMappings,
)
from src.nodes.prompt_switch import (
NODE_CLASS_MAPPINGS as PromptSwitchMappings,
)
from src.nodes.prompt_switch import (
NODE_DISPLAY_NAME_MAPPINGS as PromptSwitchDisplayMappings,
)
from src.nodes.quick_combo import (
NODE_CLASS_MAPPINGS as QuickComboMappings,
)
from src.nodes.quick_combo import (
NODE_DISPLAY_NAME_MAPPINGS as QuickComboDisplayMappings,
)
NODE_CLASS_MAPPINGS = {}
NODE_CLASS_MAPPINGS.update(PromptPresetSelectorMappings)
NODE_CLASS_MAPPINGS.update(PromptOptionPickerMappings)
NODE_CLASS_MAPPINGS.update(PromptLayoutFillerMappings)
NODE_CLASS_MAPPINGS.update(PromptSwitchMappings)
NODE_CLASS_MAPPINGS.update(PromptRouterMappings)
NODE_CLASS_MAPPINGS.update(QuickComboMappings)
NODE_DISPLAY_NAME_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS.update(PromptPresetSelectorDisplayMappings)
NODE_DISPLAY_NAME_MAPPINGS.update(PromptOptionPickerDisplayMappings)
NODE_DISPLAY_NAME_MAPPINGS.update(PromptLayoutFillerDisplayMappings)
NODE_DISPLAY_NAME_MAPPINGS.update(PromptSwitchDisplayMappings)
NODE_DISPLAY_NAME_MAPPINGS.update(PromptRouterDisplayMappings)
NODE_DISPLAY_NAME_MAPPINGS.update(QuickComboDisplayMappings)
# Node registration
__all__ = [
"NODE_CLASS_MAPPINGS",
"NODE_DISPLAY_NAME_MAPPINGS",
]
WEB_DIRECTORY = "./web"