selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 20:29:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Generate waveforms for the mixed audio

Browse Source
This commit is contained in:
Sergey Mankovsky
2025-10-21 13:33:31 +02:00
parent f844b9fc1f
commit acb6e90f28
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
server/reflector/pipelines/main_multitrack_pipeline.py
View File

@@ -27,6 +27,7 @@ from reflector.processors import (
TranscriptFinalTitleProcessor,
TranscriptTopicDetectorProcessor,
)
from reflector.processors.audio_waveform_processor import AudioWaveformProcessor
from reflector.processors.file_transcript import FileTranscriptInput
from reflector.processors.file_transcript_auto import FileTranscriptAutoProcessor
from reflector.processors.types import TitleSummary
@@ -280,6 +281,23 @@ class PipelineMainMultitrack(PipelineMainBase):
except Exception as e:
self.logger.error("Mixdown failed", error=str(e))
# Generate waveform for the mixed audio if present
try:
if transcript.audio_mp3_filename.exists():
waveform_processor = AudioWaveformProcessor(
audio_path=transcript.audio_mp3_filename,
waveform_path=transcript.audio_waveform_filename,
)
waveform_processor.set_pipeline(self.empty_pipeline)
await waveform_processor.flush()
else:
self.logger.warning(
"Waveform skipped - mixed MP3 not found",
path=str(transcript.audio_mp3_filename),
)
except Exception as e:
self.logger.error("Waveform generation failed", error=str(e))
speaker_transcripts: list[TranscriptType] = []
for idx, key in enumerate(track_keys):
ext = ".mp4"