"""
ComfyUI Custom Node Pack: Prompt Database
==========================================
Allows storing, categorizing, and selecting prompts from a local JSON database.
"""

from .nodes import PromptDBManager, PromptSelector, PromptCombiner, ApplyLoraStack

# Register API routes on import
from . import api_routes  # noqa: F401

NODE_CLASS_MAPPINGS = {
    "PromptDBManager": PromptDBManager,
    "PromptSelector": PromptSelector,
    "PromptCombiner": PromptCombiner,
    "ApplyLoraStack": ApplyLoraStack,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "PromptDBManager": "Prompt DB Manager",
    "PromptSelector": "Prompt Selector",
    "PromptCombiner": "Prompt Combiner",
    "ApplyLoraStack": "Apply Lora Stack",
}

WEB_DIRECTORY = "./web"

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]