Sucial
a9cefc5f73
- Removed the unsafe `PymssLoadAudioBatch` node and its related registrations, locale entries, and example workflow. - Kept the `Upload Audio` button, but restricted `/comfy-mss/upload-audio` to local clients only (`comfy_mss/services/routes.py:30-36`). - Uploads are always saved under ComfyUI's input directory, with filename validation and `realpath`/`commonpath` checks to prevent path traversal. - Migrated node and button translations to ComfyUI's official locale system using `locales/<language>/main.json` and `/i18n`.
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
from .comfy_mss.nodes.audio import (
|
|
PymssAudioEnsemble,
|
|
PymssAudioInvertPhase,
|
|
PymssAudioNormalize,
|
|
)
|
|
from .comfy_mss.nodes.io import (
|
|
PymssLoadAudio,
|
|
PymssSaveAudio,
|
|
)
|
|
from .comfy_mss.nodes.params import PymssMssParams, PymssVrParams
|
|
from .comfy_mss.nodes.separate import (
|
|
PymssCustomMssSeparate,
|
|
PymssCustomMssSeparateList,
|
|
PymssMssSeparate,
|
|
PymssMssSeparateList,
|
|
PymssVrSeparate,
|
|
PymssVrSeparateList,
|
|
)
|
|
from .comfy_mss.services.routes import register_routes
|
|
|
|
|
|
register_routes()
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"pymss_mss_separate": PymssMssSeparate,
|
|
"pymss_mss_separate_list": PymssMssSeparateList,
|
|
"pymss_custom_mss_separate": PymssCustomMssSeparate,
|
|
"pymss_custom_mss_separate_list": PymssCustomMssSeparateList,
|
|
"pymss_vr_separate": PymssVrSeparate,
|
|
"pymss_vr_separate_list": PymssVrSeparateList,
|
|
"pymss_mss_params": PymssMssParams,
|
|
"pymss_vr_params": PymssVrParams,
|
|
"pymss_load_audio": PymssLoadAudio,
|
|
"pymss_audio_invert_phase": PymssAudioInvertPhase,
|
|
"pymss_audio_normalize": PymssAudioNormalize,
|
|
"pymss_audio_ensemble": PymssAudioEnsemble,
|
|
"pymss_save_audio": PymssSaveAudio,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"pymss_mss_separate": "MSS Separate",
|
|
"pymss_mss_separate_list": "MSS Separate List",
|
|
"pymss_custom_mss_separate": "Custom MSS Separate",
|
|
"pymss_custom_mss_separate_list": "Custom MSS Separate List",
|
|
"pymss_vr_separate": "VR Separate",
|
|
"pymss_vr_separate_list": "VR Separate List",
|
|
"pymss_mss_params": "MSS Params",
|
|
"pymss_vr_params": "VR Params",
|
|
"pymss_load_audio": "Load Audio",
|
|
"pymss_audio_invert_phase": "Audio Invert Phase",
|
|
"pymss_audio_normalize": "Audio Normalize",
|
|
"pymss_audio_ensemble": "Audio Ensemble",
|
|
"pymss_save_audio": "Save Audio",
|
|
}
|