33 lines
1.3 KiB
Python
33 lines
1.3 KiB
Python
from .nodes.toString import *
|
|
from .nodes.files import *
|
|
from .nodes.prompt import *
|
|
from .nodes.log_utils import logger
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
|
|
|
# A dictionary that contains all nodes you want to export with their names
|
|
# NOTE: names should be globally unique
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FormatIntToString": KN_FormatIntToString,
|
|
"RandomIntToString": KN_RandomIntToString,
|
|
"RandomFloatToString": KN_RandomFloatToString,
|
|
"GetFileCountInOutputFolder": KN_GetFileCountInOutputFolder,
|
|
"LoadCSV": KN_CSV_Reader,
|
|
"DynamicPromptNode": KN_DynamicPromptNode,
|
|
"SequentialPromptNode": KN_SequentialPromptNode,
|
|
"GetVideoFileCountInOutputFolder": KN_GetVideoFileCountInOutputFolder,
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FormatIntToString": "Format Int To String",
|
|
"RandomIntToString": "Random Int To String",
|
|
"RandomFloatToString": "Random Float To String",
|
|
"GetFileCountInOutputFolder": "Get Counter In Output Folder",
|
|
"LoadCSV": "Load Prompt from CSV",
|
|
"DynamicPromptNode": "Dynamic Prompt",
|
|
"SequentialPromptNode": "Sequential Prompt",
|
|
"GetVideoFileCountInOutputFolder": "Get Video Counter In Output Folder",
|
|
}
|
|
|
|
print("[Kaleidia Nodes]: \033[92mLoaded\033[0m") |