36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
# MBM's Music Visualizer
|
|
|
|
# Imports
|
|
from .mbmAudioFeatureCalculator import AudioFeatureCalculator
|
|
from .mbmAudioLoader import AudioLoader
|
|
from .mbmMusicVisualizer import MusicVisualizer
|
|
from .mbmPromptSequence import PromptSequenceBuilder
|
|
from .mbmPromptSequenceInterpolator import PromptSequenceInterpolator
|
|
from .mbmImageConcat import ImageConcatenator
|
|
|
|
# A dictionary that contains all nodes you want to export with their names
|
|
# NOTE: names should be globally unique
|
|
NODE_CLASS_MAPPINGS = {
|
|
# "id": Class
|
|
"mbmAudioFeatureCalculator": AudioFeatureCalculator,
|
|
"mbmAudioLoader": AudioLoader,
|
|
"mbmMusicVisuzalizer": MusicVisualizer,
|
|
"mbmPromptSequenceBuilder": PromptSequenceBuilder,
|
|
"mbmPromptSequenceInterpolator": PromptSequenceInterpolator,
|
|
"mbmImageConcat": ImageConcatenator
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
# "id": "readable"
|
|
"mbmAudioFeatureCalculator": "Audio Feature Calculator",
|
|
"mbmAudioLoader": "Audio Loader",
|
|
"mbmMusicVisuzalizer": "Music Visualizer",
|
|
"mbmPromptSequenceBuilder": "Prompt Sequence Builder",
|
|
"mbmPromptSequenceInterpolator": "Prompt Sequence Interpolator",
|
|
"mbmImageConcat": "Image Concatenator"
|
|
}
|
|
|
|
# Export
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|