From 2ff2ecdd388d7e54abdec8ae3410667d9ea237c9 Mon Sep 17 00:00:00 2001 From: "Reece H. Dunn" Date: Sat, 27 Jul 2024 12:24:02 +0100 Subject: [PATCH] Group and reorder the node mappings. --- __init__.py | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) diff --git a/__init__.py b/__init__.py index 054207b..cb60290 100644 --- a/__init__.py +++ b/__init__.py @@ -9,21 +9,27 @@ from .spect_nodes import ( ) NODE_CLASS_MAPPINGS = { - "ComfyAudio.GriffinLim": GriffinLim, - "ComfyAudio.InverseSpectrogram": InverseSpectrogram, + # I/O "ComfyAudio.LoadAudio": LoadAudio, - "ComfyAudio.MelSpectrogram": MelSpectrogram, + # Plotting "ComfyAudio.PlotWaveform": PlotWaveform, - "ComfyAudio.PowerSpectrogram": PowerSpectrogram, + # Spectrogram "ComfyAudio.Spectrogram": Spectrogram, + "ComfyAudio.InverseSpectrogram": InverseSpectrogram, + "ComfyAudio.PowerSpectrogram": PowerSpectrogram, + "ComfyAudio.GriffinLim": GriffinLim, + "ComfyAudio.MelSpectrogram": MelSpectrogram, } NODE_DISPLAY_NAME_MAPPINGS = { - "ComfyAudio.GriffinLim": "Griffin Lim", - "ComfyAudio.InverseSpectrogram": "Inverse Spectrogram", + # I/O "ComfyAudio.LoadAudio": "Load Audio From Path", - "ComfyAudio.MelSpectrogram": "Mel-scale Spectrogram", + # Plotting "ComfyAudio.PlotWaveform": "Plot Waveform", - "ComfyAudio.PowerSpectrogram": "Power Spectrogram", + # Spectrogram "ComfyAudio.Spectrogram": "Spectrogram", + "ComfyAudio.InverseSpectrogram": "Inverse Spectrogram", + "ComfyAudio.PowerSpectrogram": "Power Spectrogram", + "ComfyAudio.GriffinLim": "Griffin Lim", + "ComfyAudio.MelSpectrogram": "Mel-scale Spectrogram", }