diff --git a/api/api/tasks.py b/api/api/tasks.py index 63f0eb8..39380bb 100644 --- a/api/api/tasks.py +++ b/api/api/tasks.py @@ -18,7 +18,7 @@ def save_and_classify(file_path, file_content): # Classify the audio audio_class = classify_audio(waveform, sr) - return audio_class + return audio_class, sr @shared_task def normalize_audio_task(file_path): diff --git a/api/api/views.py b/api/api/views.py index f026ee9..88ef97a 100644 --- a/api/api/views.py +++ b/api/api/views.py @@ -43,12 +43,13 @@ def upload_audio(request): task = save_and_classify.apply(args=(file_path, audio_file.read())) if task.successful(): - audio_class = task.result + audio_class = task.result[0] return Response( { "Status": "File uploaded successfully", "file_uuid": file_uuid, - "audio_class": audio_class + "audio_class": audio_class, + "sr": task.result[1] }, status=status.HTTP_201_CREATED, )