"""
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

# 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
}

# Mapping for display name
NODE_DISPLAY_NAME_MAPPINGS = {
    "AnimaginePrompt": "Animagine Prompt",
    "MultilineTextInput": "Multiline Text Input",
    "TextFileLoader": "Wildcards Text File Loader"
}


__version__ = "1.4.3"