41 lines
1.9 KiB
Python
41 lines
1.9 KiB
Python
from .rpg_characters_node import NODE_CLASS_MAPPINGS as CHARACTER_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as CHARACTER_DISPLAY
|
|
from .rpg_art_style_node import NODE_CLASS_MAPPINGS as STYLE_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as STYLE_DISPLAY
|
|
from .prompt_selector_node import NODE_CLASS_MAPPINGS as PROMPT_SELECTOR_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as PROMPT_SELECTOR_DISPLAY
|
|
from .rpg_prompt_concatenator_node import NODE_CLASS_MAPPINGS as CONCATENATOR_NODE_MAPPINGS
|
|
from .rpg_prompt_concatenator_node import NODE_DISPLAY_NAME_MAPPINGS as CONCATENATOR_DISPLAY
|
|
from .prompt_conditioning_converter_node import NODE_CLASS_MAPPINGS as CONDITIONING_NODE_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as CONDITIONING_NODE_DISPLAY
|
|
from .model_likeness_switch import ModelLikenessSwitch
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_CLASS_MAPPINGS.update(CHARACTER_MAPPINGS)
|
|
NODE_CLASS_MAPPINGS.update(STYLE_MAPPINGS)
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CHARACTER_DISPLAY)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(STYLE_DISPLAY)
|
|
|
|
NODE_CLASS_MAPPINGS.update(PROMPT_SELECTOR_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(PROMPT_SELECTOR_DISPLAY)
|
|
|
|
NODE_CLASS_MAPPINGS.update(CONCATENATOR_NODE_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CONCATENATOR_DISPLAY)
|
|
|
|
NODE_CLASS_MAPPINGS.update(CONDITIONING_NODE_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CONDITIONING_NODE_DISPLAY)
|
|
|
|
NODE_CLASS_MAPPINGS["ModelLikenessSwitch"] = ModelLikenessSwitch
|
|
NODE_DISPLAY_NAME_MAPPINGS["ModelLikenessSwitch"] = "Model Selector (Use Likeness)"
|
|
|
|
NODE_CLASS_MAPPINGS["prompt_selector_node"] = prompt_selector_node
|
|
NODE_DISPLAY_NAME_MAPPINGS["prompt_selector_node"] = "Prompt Selector"
|
|
|
|
# Gen 3 structured Character DNA system.
|
|
from .gen3 import (
|
|
NODE_CLASS_MAPPINGS as GEN3_MAPPINGS,
|
|
NODE_DISPLAY_NAME_MAPPINGS as GEN3_DISPLAY,
|
|
)
|
|
NODE_CLASS_MAPPINGS.update(GEN3_MAPPINGS)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(GEN3_DISPLAY)
|