studio/meet
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🩹(summary) remove temporary audio files saved

Browse Source 
I forgot to remove temporary files downloaded from the Minio bucket.
This commit is contained in:
lebaudantoine
2024-12-03 19:09:42 +01:00
committed by aleb_the_flash
parent 459bbf65a8
commit ad0c3eea66
1 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/summary/summary/core/celery_worker.py
View File

@@ -1,6 +1,7 @@
"""Celery workers.""" """Celery workers."""
import json import json
import os
import tempfile import tempfile
from pathlib import Path from pathlib import Path
@@ -28,6 +29,7 @@ celery = Celery(
) )
if settings.sentry_dsn and settings.sentry_is_enabled: if settings.sentry_dsn and settings.sentry_is_enabled:
@signals.celeryd_init.connect @signals.celeryd_init.connect
def init_sentry(**_kwargs): def init_sentry(**_kwargs):
"""Initialize sentry.""" """Initialize sentry."""
@@ -100,15 +102,20 @@ def process_audio_transcribe_summarize(filename: str, email: str, sub: str):
api_key=settings.openai_api_key, base_url=settings.openai_base_url api_key=settings.openai_api_key, base_url=settings.openai_base_url
) )
logger.debug("Querying transcription …") try:
with open(temp_file_path, "rb") as audio_file: logger.debug("Querying transcription …")
transcription = openai_client.audio.transcriptions.create( with open(temp_file_path, "rb") as audio_file:
model=settings.openai_asr_model, file=audio_file transcription = openai_client.audio.transcriptions.create(
) model=settings.openai_asr_model, file=audio_file
)
transcription = transcription.text transcription = transcription.text
logger.debug("Transcription: \n %s", transcription) logger.debug("Transcription: \n %s", transcription)
finally:
if os.path.exists(temp_file_path):
os.remove(temp_file_path)
logger.debug("Temporary file removed: %s", temp_file_path)
instructions = get_instructions(transcription) instructions = get_instructions(transcription)
summary_response = openai_client.chat.completions.create( summary_response = openai_client.chat.completions.create(