17 lines
733 B
Python
17 lines
733 B
Python
"""
|
|
Prompt Factory Generator - Custom ComfyUI nodes for advanced prompt generation
|
|
"""
|
|
|
|
# Display startup message in yellow
|
|
print("\033[93m🏭 The assembly line is live.\033[0m")
|
|
|
|
from .Factory_prompt_generator import NODE_CLASS_MAPPINGS as PROMPT_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as PROMPT_DISPLAY_MAPPINGS
|
|
from .Factory_negative_generator import NODE_CLASS_MAPPINGS as NEGATIVE_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as NEGATIVE_DISPLAY_MAPPINGS
|
|
|
|
# Combine all node mappings
|
|
NODE_CLASS_MAPPINGS = {**PROMPT_MAPPINGS, **NEGATIVE_MAPPINGS}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {**PROMPT_DISPLAY_MAPPINGS, **NEGATIVE_DISPLAY_MAPPINGS}
|
|
|
|
# Export node mappings
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|