41 lines
1.8 KiB
Python
41 lines
1.8 KiB
Python
# /ComfyUI-Prompt-Formatter/__init__.py
|
|
|
|
# Ensure utils are loaded first to check dependencies
|
|
from . import prompt_formatter_utils
|
|
|
|
# Import node classes
|
|
from .categorized_prompt_formatter import CategorizedPromptFormatter
|
|
from .categorized_random_prompt_formatter import CategorizedRandomPromptFormatter
|
|
from .categorized_prompt_analyzer import CategorizedPromptAnalyzer
|
|
from .wildcard_importer import WildcardImporter
|
|
from .prompt_normalizer import PromptNormalizer
|
|
|
|
# Standardize NODE_NAME attribute in WildcardImporter for consistency
|
|
if not hasattr(WildcardImporter, 'NODE_NAME'):
|
|
WildcardImporter.NODE_NAME = "Wildcard Importer"
|
|
|
|
# Mapping node implementations
|
|
NODE_CLASS_MAPPINGS = {
|
|
CategorizedPromptFormatter.NODE_NAME: CategorizedPromptFormatter,
|
|
CategorizedRandomPromptFormatter.NODE_NAME: CategorizedRandomPromptFormatter,
|
|
CategorizedPromptAnalyzer.NODE_NAME: CategorizedPromptAnalyzer,
|
|
WildcardImporter.NODE_NAME: WildcardImporter,
|
|
PromptNormalizer.NODE_NAME: PromptNormalizer,
|
|
}
|
|
|
|
# Mapping display names for the UI
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
CategorizedPromptFormatter.NODE_NAME: "📝 Categorized Prompt Formatter",
|
|
CategorizedRandomPromptFormatter.NODE_NAME: "🎲 Categorized Random Prompt Formatter",
|
|
CategorizedPromptAnalyzer.NODE_NAME: "📊 Categorized Prompt Analyzer",
|
|
WildcardImporter.NODE_NAME: "📂 Wildcard Importer",
|
|
PromptNormalizer.NODE_NAME: "✨ Prompt Normalizer",
|
|
}
|
|
|
|
__all__ =['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
|
|
|
# Print loading info to console
|
|
print("\n### Loading: ComfyUI-Prompt-Formatter (Version: 1.5.0) ###")
|
|
for name, display_name in NODE_DISPLAY_NAME_MAPPINGS.items():
|
|
print(f" - {name} -> {display_name}")
|
|
print("###-------------------------------------------###\n") |