drago87
392b45c5a2
Added a node that lets the user connect multiple character loras but only uses one of them
36 lines
966 B
Python
36 lines
966 B
Python
from .nodes import (
|
|
DragosVariableNode,
|
|
DragosObjectNode,
|
|
DragosSceneCompiler,
|
|
DragosStructuredBuilderNode,
|
|
DragosPromptLoaderNode,
|
|
DragosCharacterSelectorNode
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
|
|
"DragosVariable": DragosVariableNode,
|
|
"DragosObject": DragosObjectNode,
|
|
"DragosSceneCompiler": DragosSceneCompiler,
|
|
"DragosStructuredBuilder": DragosStructuredBuilderNode,
|
|
"DragosPromptLoader": DragosPromptLoaderNode,
|
|
"DragosCharacterSelector": DragosCharacterSelectorNode
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
|
|
"DragosVariable": "Dragos Variable",
|
|
"DragosObject": "Dragos Object",
|
|
"DragosSceneCompiler": "Dragos Scene Compiler",
|
|
"DragosStructuredBuilder": "Dragos Structured Builder",
|
|
"DragosPromptLoader": "Dragos Prompt Loader",
|
|
"DragosCharacterSelector": "Dragos Character Selector"
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
"WEB_DIRECTORY"
|
|
] |