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

Emit multriack pipeline events

Browse Source
This commit is contained in:
Sergey Mankovsky
2025-10-21 16:31:31 +02:00
parent acb6e90f28
commit d82abf65ba
1 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -11,13 +11,13 @@ from celery import chain, shared_task
from reflector.asynctask import asynctask from reflector.asynctask import asynctask
from reflector.db.transcripts import ( from reflector.db.transcripts import (
TranscriptStatus, TranscriptStatus,
TranscriptText,
transcripts_controller, transcripts_controller,
) )
from reflector.logger import logger from reflector.logger import logger
from reflector.pipelines.main_file_pipeline import task_send_webhook_if_needed from reflector.pipelines.main_file_pipeline import task_send_webhook_if_needed
from reflector.pipelines.main_live_pipeline import ( from reflector.pipelines.main_live_pipeline import (
PipelineMainBase, PipelineMainBase,
broadcast_to_sockets,
task_cleanup_consent, task_cleanup_consent,
task_pipeline_post_to_zulip, task_pipeline_post_to_zulip,
) )
@@ -225,6 +225,7 @@ class PipelineMainMultitrack(PipelineMainBase):
for c in containers: for c in containers:
c.close() c.close()
@broadcast_to_sockets
async def set_status(self, transcript_id: str, status: TranscriptStatus): async def set_status(self, transcript_id: str, status: TranscriptStatus):
async with self.lock_transaction(): async with self.lock_transaction():
return await transcripts_controller.set_status(transcript_id, status) return await transcripts_controller.set_status(transcript_id, status)
@@ -274,7 +275,8 @@ class PipelineMainMultitrack(PipelineMainBase):
# Mixdown all available tracks into transcript.audio_mp3_filename, preserving sample rate # Mixdown all available tracks into transcript.audio_mp3_filename, preserving sample rate
try: try:
mp3_writer = AudioFileWriterProcessor( mp3_writer = AudioFileWriterProcessor(
path=str(transcript.audio_mp3_filename) path=str(transcript.audio_mp3_filename),
on_duration=self.on_duration,
) )
await self.mixdown_tracks(track_datas, mp3_writer, offsets_seconds) await self.mixdown_tracks(track_datas, mp3_writer, offsets_seconds)
await mp3_writer.flush() await mp3_writer.flush()
@@ -287,6 +289,7 @@ class PipelineMainMultitrack(PipelineMainBase):
waveform_processor = AudioWaveformProcessor( waveform_processor = AudioWaveformProcessor(
audio_path=transcript.audio_mp3_filename, audio_path=transcript.audio_mp3_filename,
waveform_path=transcript.audio_waveform_filename, waveform_path=transcript.audio_waveform_filename,
on_waveform=self.on_waveform,
) )
waveform_processor.set_pipeline(self.empty_pipeline) waveform_processor.set_pipeline(self.empty_pipeline)
await waveform_processor.flush() await waveform_processor.flush()
@@ -382,13 +385,8 @@ class PipelineMainMultitrack(PipelineMainBase):
merged_transcript = TranscriptType(words=merged_words, translation=None) merged_transcript = TranscriptType(words=merged_words, translation=None)
await transcripts_controller.append_event( # Emit TRANSCRIPT event through the shared handler (persists and broadcasts)
transcript, await self.on_transcript(merged_transcript)
event="TRANSCRIPT",
data=TranscriptText(
text=merged_transcript.text, translation=merged_transcript.translation
),
)
topics = await self.detect_topics(merged_transcript, transcript.target_language) topics = await self.detect_topics(merged_transcript, transcript.target_language)
await asyncio.gather( await asyncio.gather(