29 lines
819 B
Python
29 lines
819 B
Python
"""
|
|
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"]
|