gmorks
c267edcf13
Update package version to 1.6.1 in pyproject.toml and __init__.py Modify logger.py and state_manager.py to use absolute paths relative to node directory Update example workflow files with new paths and configurations
49 lines
1.5 KiB
Python
49 lines
1.5 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
|
|
from .api_routes import define_routes
|
|
|
|
# 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 within the node directory
|
|
if not os.path.exists(SAMPLE_DATA_DIR):
|
|
os.makedirs(SAMPLE_DATA_DIR)
|
|
|
|
# Node class mappings for registration
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AnimaginePrompt": AnimaginePromptNode,
|
|
"MultilineTextInput": MultilineTextInput,
|
|
"TextFileLoader": TextFileLoader,
|
|
"MultiWildcardLoader": MultiWildcardLoader
|
|
}
|
|
|
|
# Display name mappings for nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"AnimaginePrompt": "Animagine Prompt",
|
|
"MultilineTextInput": "Multiline Text Input",
|
|
"TextFileLoader": "Wildcards Text File Loader",
|
|
"MultiWildcardLoader": "Multi-Wildcard Loader"
|
|
}
|
|
|
|
# Web directory configuration for JavaScript extensions
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
# Register API routes directly
|
|
define_routes()
|
|
logger.info("[Animagine Prompt] API routes initialized")
|
|
|
|
__version__ = "1.6.1"
|