diff --git a/server/reflector/pipelines/main_live_pipeline.py b/server/reflector/pipelines/main_live_pipeline.py index 8f45cafe..20743391 100644 --- a/server/reflector/pipelines/main_live_pipeline.py +++ b/server/reflector/pipelines/main_live_pipeline.py @@ -474,6 +474,15 @@ class PipelineMainWaveform(PipelineMainFromTopics): ] +@get_transcript +async def pipeline_remove_upload(transcript: Transcript, logger: Logger): + logger.info("Starting remove upload") + uploads = transcript.data_path.glob("upload.*") + for upload in uploads: + upload.unlink(missing_ok=True) + logger.info("Remove upload done") + + @get_transcript async def pipeline_waveform(transcript: Transcript, logger: Logger): logger.info("Starting waveform") @@ -560,6 +569,12 @@ async def pipeline_summaries(transcript: Transcript, logger: Logger): # =================================================================== +@shared_task +@asynctask +async def task_pipeline_remove_upload(*, transcript_id: str): + await pipeline_remove_upload(transcript_id=transcript_id) + + @shared_task @asynctask async def task_pipeline_waveform(*, transcript_id: str): @@ -604,6 +619,7 @@ def pipeline_post(*, transcript_id: str): task_pipeline_waveform.si(transcript_id=transcript_id) | task_pipeline_convert_to_mp3.si(transcript_id=transcript_id) | task_pipeline_upload_mp3.si(transcript_id=transcript_id) + | task_pipeline_remove_upload.si(transcript_id=transcript_id) | task_pipeline_diarization.si(transcript_id=transcript_id) ) chain_title_preview = task_pipeline_title_and_short_summary.si(