From e61ae413af061aba9d6237530a29c26cd7385849 Mon Sep 17 00:00:00 2001
From: Joel Mathew Thomas <90510078+joelmathewthomas@users.noreply.github.com>
Date: Tue, 25 Feb 2025 03:23:59 +0530
Subject: [PATCH] return audio_class with response for api/upload endpoint

---
 api/api/views.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/api/api/views.py b/api/api/views.py
index 27ca803..ea7ea36 100644
--- a/api/api/views.py
+++ b/api/api/views.py
@@ -2,6 +2,7 @@ import os
 from rest_framework.decorators import api_view
 from rest_framework.response import Response
 from rest_framework import status
+from freqsplit.preprocessing.classify import classify_audio
 
 UPLOAD_DIR = "/tmp/freq-split-enhance"
 
@@ -21,5 +22,13 @@ def upload_audio(request):
     with open(file_path, 'wb') as destination:
         for chunk in audio_file.chunks():
             destination.write(chunk)
+
+    audio_class = classify_audio(file_path)
             
-    return Response({"Status": "File uploaded successfully", "file_path": file_path}, status=status.HTTP_201_CREATED)
+    return Response(
+        {
+            "Status": "File uploaded successfully",
+            "file_path": file_path,
+            "audio_class": audio_class
+            }, status=status.HTTP_201_CREATED,
+        )