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

Restart processing

Browse Source
This commit is contained in:
Sergey Mankovsky
2024-07-17 18:09:36 +02:00
parent 74a1c69e1f
commit 562f2c94f9
8 changed files with 219 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -637,13 +637,21 @@ def pipeline_post(*, transcript_id: str):
@get_transcript
async def pipeline_upload(transcript: Transcript, logger: Logger):
async def pipeline_process(transcript: Transcript, logger: Logger):
import av
try:
# open audio
upload_filename = next(transcript.data_path.glob("upload.*"))
container = av.open(upload_filename.as_posix())
audio_filename = next(transcript.data_path.glob("upload.*"), None)
if audio_filename and transcript.status != "uploaded":
raise Exception("File upload is not completed")
if not audio_filename:
audio_filename = next(transcript.data_path.glob("audio.*"), None)
if not audio_filename:
raise Exception("There is no file to process")
container = av.open(audio_filename.as_posix())
# create pipeline
pipeline = PipelineMainLive(transcript_id=transcript.id)
@@ -676,5 +684,5 @@ async def pipeline_upload(transcript: Transcript, logger: Logger):
@shared_task
@asynctask
async def task_pipeline_upload(*, transcript_id: str):
return await pipeline_upload(transcript_id=transcript_id)
async def task_pipeline_process(*, transcript_id: str):
return await pipeline_process(transcript_id=transcript_id)