35 lines
1.6 KiB
Python
35 lines
1.6 KiB
Python
from .nodes import NODE_CLASS_MAPPINGS as NODES_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as NODES_DISPLAY_MAPPINGS
|
|
|
|
# Import Chinese tools nodes
|
|
try:
|
|
from .chinese_tools import NODE_CLASS_MAPPINGS as CHINESE_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as CHINESE_DISPLAY_MAPPINGS
|
|
# Merge the mappings
|
|
NODE_CLASS_MAPPINGS = {**NODES_MAPPINGS, **CHINESE_MAPPINGS}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {**NODES_DISPLAY_MAPPINGS, **CHINESE_DISPLAY_MAPPINGS}
|
|
except Exception as e:
|
|
print(f"⚠️ Failed to load Chinese tools nodes: {e}")
|
|
NODE_CLASS_MAPPINGS = NODES_MAPPINGS
|
|
NODE_DISPLAY_NAME_MAPPINGS = NODES_DISPLAY_MAPPINGS
|
|
|
|
# Import person feature extractor nodes
|
|
try:
|
|
from .person_feature_extractor import NODE_CLASS_MAPPINGS as FEATURE_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as FEATURE_DISPLAY_MAPPINGS
|
|
# Merge the mappings
|
|
NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **FEATURE_MAPPINGS}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **FEATURE_DISPLAY_MAPPINGS}
|
|
except Exception as e:
|
|
print(f"⚠️ Failed to load person feature extractor nodes: {e}")
|
|
|
|
# Import show text node
|
|
try:
|
|
from .show_text import NODE_CLASS_MAPPINGS as SHOWTEXT_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as SHOWTEXT_DISPLAY_MAPPINGS
|
|
# Merge the mappings
|
|
NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **SHOWTEXT_MAPPINGS}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **SHOWTEXT_DISPLAY_MAPPINGS}
|
|
except Exception as e:
|
|
print(f"⚠️ Failed to load show text node: {e}")
|
|
|
|
# Web directory for frontend extensions
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS', 'WEB_DIRECTORY'] |