40 lines
1.7 KiB
Python
40 lines
1.7 KiB
Python
from .components.audio_sink_node import AudioSinkNode
|
|
from .components.load_whisper_model import LoadWhisperModelNode
|
|
from .components.whisper_transcription import WhisperTranscriptionNode
|
|
from .components.whisper_transcription_batch import WhisperTranscriptionBatchNode
|
|
from .components.load_video_audio import LoadVideoAudioNode
|
|
from .components.load_audio import LoadAudioNode
|
|
from .components.load_batch import LoadBatchNode
|
|
|
|
from .components.convert.vhs_audio_to_audio import VHSAudioToWavBytesNode
|
|
|
|
# Set the web directory, any .js file in that directory will be loaded by the frontend as a frontend extension
|
|
# WEB_DIRECTORY = "./somejs"
|
|
|
|
# A dictionary that contains all nodes you want to export with their names
|
|
# NOTE: names should be globally unique
|
|
NODE_CLASS_MAPPINGS = {
|
|
"TT-AudioSink": AudioSinkNode,
|
|
"TT-LoadWhisperModel": LoadWhisperModelNode,
|
|
"TT-WhisperTranscription": WhisperTranscriptionNode,
|
|
"TT-WhisperTranscriptionBatch": WhisperTranscriptionBatchNode,
|
|
"TT-ConvertVhsAudioToAudio": VHSAudioToWavBytesNode,
|
|
"TT-LoadVideoAudio": LoadVideoAudioNode,
|
|
"TT-LoadAudio": LoadAudioNode,
|
|
"TT-LoadBatch": LoadBatchNode,
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"TT-AudioSink": "Audio Sink",
|
|
"TT-LoadWhisperModel": "Load Whisper Transcription Model",
|
|
"TT-WhisperTranscription": "Whisper Transcribe",
|
|
"TT-WhisperTranscriptionBatch": "Whisper Transcribe Batch",
|
|
"TT-ConvertVhsAudioToAudio": "Convert VHS Audio to WAV bytes",
|
|
"TT-LoadVideoAudio": "Load Audio from Video",
|
|
"TT-LoadAudio": "Load Audio",
|
|
"TT-LoadBatch": "Load Audio from Batch",
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|