From 7672817ff3b2c87a5d62b9b2de33a057511b8d8c Mon Sep 17 00:00:00 2001
From: Joel Mathew Thomas <90510078+joelmathewthomas@users.noreply.github.com>
Date: Wed, 26 Feb 2025 13:37:08 +0530
Subject: [PATCH] return sr with response of /api/upload/

---
 api/api/tasks.py | 2 +-
 api/api/views.py | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

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,
             )