return audio_class with response for api/upload endpoint
This commit is contained in:
+10
-1
@@ -2,6 +2,7 @@ import os
|
|||||||
from rest_framework.decorators import api_view
|
from rest_framework.decorators import api_view
|
||||||
from rest_framework.response import Response
|
from rest_framework.response import Response
|
||||||
from rest_framework import status
|
from rest_framework import status
|
||||||
|
from freqsplit.preprocessing.classify import classify_audio
|
||||||
|
|
||||||
UPLOAD_DIR = "/tmp/freq-split-enhance"
|
UPLOAD_DIR = "/tmp/freq-split-enhance"
|
||||||
|
|
||||||
@@ -22,4 +23,12 @@ def upload_audio(request):
|
|||||||
for chunk in audio_file.chunks():
|
for chunk in audio_file.chunks():
|
||||||
destination.write(chunk)
|
destination.write(chunk)
|
||||||
|
|
||||||
return Response({"Status": "File uploaded successfully", "file_path": file_path}, status=status.HTTP_201_CREATED)
|
audio_class = classify_audio(file_path)
|
||||||
|
|
||||||
|
return Response(
|
||||||
|
{
|
||||||
|
"Status": "File uploaded successfully",
|
||||||
|
"file_path": file_path,
|
||||||
|
"audio_class": audio_class
|
||||||
|
}, status=status.HTTP_201_CREATED,
|
||||||
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user