72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
"""
|
|
FEnodes — ComfyUI custom node pack
|
|
Author: FugitiveExpert01
|
|
"""
|
|
import logging
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
try:
|
|
from .nodes_tiling import NODE_CLASS_MAPPINGS as _TILING_CLASSES
|
|
from .nodes_tiling import NODE_DISPLAY_NAME_MAPPINGS as _TILING_NAMES
|
|
except Exception as e:
|
|
logging.warning(f"[FEnodes] Failed to load Tiling nodes: {e}")
|
|
_TILING_CLASSES = {}
|
|
_TILING_NAMES = {}
|
|
|
|
try:
|
|
from .nodes_text import NODE_CLASS_MAPPINGS as _TEXT_CLASSES
|
|
from .nodes_text import NODE_DISPLAY_NAME_MAPPINGS as _TEXT_NAMES
|
|
except Exception as e:
|
|
logging.warning(f"[FEnodes] Failed to load Text nodes: {e}")
|
|
_TEXT_CLASSES = {}
|
|
_TEXT_NAMES = {}
|
|
|
|
try:
|
|
from .nodes_color import NODE_CLASS_MAPPINGS as _COLOR_CLASSES
|
|
from .nodes_color import NODE_DISPLAY_NAME_MAPPINGS as _COLOR_NAMES
|
|
except Exception as e:
|
|
logging.warning(f"[FEnodes] Failed to load Color nodes: {e}")
|
|
_COLOR_CLASSES = {}
|
|
_COLOR_NAMES = {}
|
|
|
|
try:
|
|
from .nodes_lora import NODE_CLASS_MAPPINGS as _LORA_CLASSES
|
|
from .nodes_lora import NODE_DISPLAY_NAME_MAPPINGS as _LORA_NAMES
|
|
except Exception as e:
|
|
logging.warning(f"[FEnodes] Failed to load LoRA nodes: {e}")
|
|
_LORA_CLASSES = {}
|
|
_LORA_NAMES = {}
|
|
|
|
try:
|
|
from .nodes_raft_flow import NODE_CLASS_MAPPINGS as _RAFT_CLASSES
|
|
from .nodes_raft_flow import NODE_DISPLAY_NAME_MAPPINGS as _RAFT_NAMES
|
|
except Exception as e:
|
|
logging.warning(f"[FEnodes] Failed to load RAFT Flow nodes: {e}")
|
|
_RAFT_CLASSES = {}
|
|
_RAFT_NAMES = {}
|
|
|
|
try:
|
|
from .routes import register_routes
|
|
register_routes()
|
|
except Exception as e:
|
|
logging.warning(f"[FEnodes] Failed to register API routes: {e}")
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
**_TILING_CLASSES,
|
|
**_TEXT_CLASSES,
|
|
**_COLOR_CLASSES,
|
|
**_LORA_CLASSES,
|
|
**_RAFT_CLASSES,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**_TILING_NAMES,
|
|
**_TEXT_NAMES,
|
|
**_COLOR_NAMES,
|
|
**_LORA_NAMES,
|
|
**_RAFT_NAMES,
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|