gmorks
71615236b6
Introduce new node supporting template mode with {filename.txt} syntax and slots mode with individual seed control. Includes real-time preview and error handling. Update version to 1.5.0 and add example wildcard files for expressions, poses, clothing, and scenarios.
Update example workflows to demonstrate new functionality and improve README with node documentation.
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""
|
|
Animagine Prompt Node for ComfyUI
|
|
A node designed to help structure prompts according to specific guidelines
|
|
"""
|
|
import os
|
|
import logging
|
|
from .animagine_node import AnimaginePromptNode
|
|
from .wildcards_node import MultilineTextInput, TextFileLoader, MultiWildcardLoader
|
|
|
|
# Basic logging configuration
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
)
|
|
logger = logging.getLogger('AnimaginePrompt')
|
|
|
|
# Base directory of the node
|
|
NODE_DIR = os.path.dirname(os.path.realpath(__file__))
|
|
SAMPLE_DATA_DIR = os.path.join(NODE_DIR, "sample_data")
|
|
|
|
# Ensure the sample_data directory exists
|
|
if not os.path.exists(SAMPLE_DATA_DIR):
|
|
os.makedirs(SAMPLE_DATA_DIR)
|
|
|
|
# Mapping for node registration
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AnimaginePrompt": AnimaginePromptNode,
|
|
"MultilineTextInput": MultilineTextInput,
|
|
"TextFileLoader": TextFileLoader,
|
|
"MultiWildcardLoader": MultiWildcardLoader
|
|
}
|
|
|
|
# Mapping for display name
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"AnimaginePrompt": "Animagine Prompt",
|
|
"MultilineTextInput": "Multiline Text Input",
|
|
"TextFileLoader": "Wildcards Text File Loader",
|
|
"MultiWildcardLoader": "Multi-Wildcard Loader"
|
|
}
|
|
|
|
|
|
__version__ = "1.5.0"
|