82 lines
5.9 KiB
Python
82 lines
5.9 KiB
Python
# By Daxton Caylor
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to
|
|
# deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
# and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
# THE SOFTWARE.
|
|
"""
|
|
@author: Daxton Caylor
|
|
@title: ComfyUI-DataSet
|
|
@nickname: ComfyUI-DataSet
|
|
@description: Data Research, Preparation, and Manipulation Nodes for Model Trainers, Artists, Designers, and Animators.
|
|
"""
|
|
|
|
from .classes.DataSet_Visualizer import N_CLASS_MAPPINGS as VisualizerMappings, N_DISPLAY_NAME_MAPPINGS as VisualizerNameMappings
|
|
from .classes.DataSet_TextFilesLoad import N_CLASS_MAPPINGS as TextFilesLoadMappings, N_DISPLAY_NAME_MAPPINGS as TextFilesLoadNameMappings
|
|
from .classes.DataSet_TextFilesLoadFromList import N_CLASS_MAPPINGS as TextFilesLoadFromListMappings, N_DISPLAY_NAME_MAPPINGS as TextFilesLoadFromListNameMappings
|
|
from .classes.DataSet_CopyFiles import N_CLASS_MAPPINGS as CopyFilesMappings, N_DISPLAY_NAME_MAPPINGS as CopyFilesNameMappings
|
|
from .classes.DataSet_TriggerWords import N_CLASS_MAPPINGS as TriggerWordsMappings, N_DISPLAY_NAME_MAPPINGS as TriggerWordsNameMappings
|
|
from .classes.DataSet_TextFilesSave import N_CLASS_MAPPINGS as TextFilesSaveMappings, N_DISPLAY_NAME_MAPPINGS as TextFilesSaveNameMappings
|
|
from .classes.DataSet_ConceptManager import N_CLASS_MAPPINGS as ConceptManagerMappings, N_DISPLAY_NAME_MAPPINGS as ConceptManagerNameMappings
|
|
from .classes.DataSet_FindAndReplace import N_CLASS_MAPPINGS as FindAndReplaceMappings, N_DISPLAY_NAME_MAPPINGS as FindAndReplaceNameMappings
|
|
from .classes.DataSet_PathSelector import N_CLASS_MAPPINGS as PathSelectorMappings, N_DISPLAY_NAME_MAPPINGS as PathSelectorNameMappings
|
|
from .classes.DataSet_LoadImage import N_CLASS_MAPPINGS as LoadImageMappings, N_DISPLAY_NAME_MAPPINGS as LoadImageNameMappings
|
|
from .classes.DataSet_SaveImage import N_CLASS_MAPPINGS as SaveImageMappings, N_DISPLAY_NAME_MAPPINGS as SaveImageNameMappings
|
|
from .classes.DataSet_SaveImagePro import N_CLASS_MAPPINGS as SaveImageProMappings, N_DISPLAY_NAME_MAPPINGS as SaveImageProNameMappings
|
|
from .classes.DataSet_OpenAIChat import N_CLASS_MAPPINGS as OpenAIChatMappings, N_DISPLAY_NAME_MAPPINGS as OpenAIChatNameMappings
|
|
from .classes.DataSet_OpenAIChatImage import N_CLASS_MAPPINGS as OpenAIChatImageMappings, N_DISPLAY_NAME_MAPPINGS as OpenAIChatImageNameMappings
|
|
from .classes.DataSet_OpenAIChatImageBatch import N_CLASS_MAPPINGS as OpenAIChatImageBatchMappings, N_DISPLAY_NAME_MAPPINGS as OpenAIChatImageBatchNameMappings
|
|
from .classes.DataSet_GroqChat import N_CLASS_MAPPINGS as GroqChatMappings, N_DISPLAY_NAME_MAPPINGS as GroqChatNameMappings
|
|
from .classes.DataSet_GroqChatImage import N_CLASS_MAPPINGS as GroqChatImageMappings, N_DISPLAY_NAME_MAPPINGS as GroqChatImageNameMappings
|
|
from .classes.DataSet_ClaudeAIChat import N_CLASS_MAPPINGS as ClaudeChatMappings, N_DISPLAY_NAME_MAPPINGS as ClaudeChatNameMappings
|
|
from .classes.DataSet_ClaudeAIChatImage import N_CLASS_MAPPINGS as ClaudeChatImageMappings, N_DISPLAY_NAME_MAPPINGS as ClaudeChatImageNameMappings
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_CLASS_MAPPINGS.update(VisualizerMappings)
|
|
NODE_CLASS_MAPPINGS.update(CopyFilesMappings)
|
|
NODE_CLASS_MAPPINGS.update(TriggerWordsMappings)
|
|
NODE_CLASS_MAPPINGS.update(TextFilesLoadFromListMappings)
|
|
NODE_CLASS_MAPPINGS.update(TextFilesLoadMappings)
|
|
NODE_CLASS_MAPPINGS.update(TextFilesSaveMappings)
|
|
NODE_CLASS_MAPPINGS.update(FindAndReplaceMappings)
|
|
NODE_CLASS_MAPPINGS.update(PathSelectorMappings)
|
|
NODE_CLASS_MAPPINGS.update(ConceptManagerMappings)
|
|
NODE_CLASS_MAPPINGS.update(LoadImageMappings)
|
|
NODE_CLASS_MAPPINGS.update(SaveImageMappings)
|
|
NODE_CLASS_MAPPINGS.update(SaveImageProMappings)
|
|
NODE_CLASS_MAPPINGS.update(OpenAIChatMappings)
|
|
NODE_CLASS_MAPPINGS.update(OpenAIChatImageMappings)
|
|
NODE_CLASS_MAPPINGS.update(OpenAIChatImageBatchMappings)
|
|
NODE_CLASS_MAPPINGS.update(GroqChatMappings)
|
|
NODE_CLASS_MAPPINGS.update(GroqChatImageMappings)
|
|
NODE_CLASS_MAPPINGS.update(ClaudeChatMappings)
|
|
NODE_CLASS_MAPPINGS.update(ClaudeChatImageMappings)
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(VisualizerNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(CopyFilesNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(TriggerWordsNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(TextFilesLoadNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(TextFilesLoadFromListNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(TextFilesSaveNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(FindAndReplaceNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(PathSelectorNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ConceptManagerNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(LoadImageNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(SaveImageNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(SaveImageProNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(OpenAIChatNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(OpenAIChatImageNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(OpenAIChatImageBatchNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(GroqChatNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(GroqChatImageNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ClaudeChatNameMappings)
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(ClaudeChatImageNameMappings)
|
|
|
|
WEB_DIRECTORY = "./web" |