generate spectrogram at backend and send as response to client

api/api/tasks.py

@@ -1,5 +1,7 @@
 import os
 import shutil
+import json
+import numpy as np
 from pathlib import Path
 from celery import shared_task
 from freqsplit.input.file_reader import read_audio
@@ -10,6 +12,7 @@ from freqsplit.preprocessing.resample import resample
 from freqsplit.postprocessing.audio_writer import export_audio
 from freqsplit.separation.demucs_wrapper import separate_audio_with_demucs
 from freqsplit.refinement.deepfilternet_wrapper import noisereduce
+from freqsplit.spectrogram.generator import generate_spectrogram
 
 
 @shared_task
@@ -24,8 +27,14 @@ def save_and_classify(file_path, file_content):
     
     # Classify the audio
     audio_class = classify_audio(waveform, sr)
-    
-    return audio_class, org_sr
+
+    # Generate spectrogram
+    spec_db, plot_data = generate_spectrogram(file_path)
+    # Convert numpy array to JSON-safe list
+    spec_db = np.nan_to_num(spec_db, nan=-80.0, posinf=-80.0, neginf=-80.0)
+    spec_data_json = json.dumps(spec_db.tolist())
+
+    return audio_class, org_sr, spec_data_json, plot_data['sr']
 
 @shared_task
 def normalize_audio_task(file_path):

api/api/views.py

@@ -47,17 +47,21 @@ def upload_audio(request):
     # Save the uploaded file
     task = save_and_classify.apply(args=(file_path, audio_file.read()))
     
-    if task.successful():
-        audio_class = task.result[0]
+    if task.ready() and task.successful():
+        result = task.result
         return Response(
             {
                 "Status": "File uploaded successfully",
                 "file_uuid": file_uuid,
-                "audio_class": audio_class,
-                "sr": task.result[1]
-                }, 
+                "audio_class": result[0],
+                "sr": result[1],
+                "spectrogram": result[2],
+                "spec_sr": result[3]
+            }, 
             status=status.HTTP_201_CREATED,
-            )
+        )
+    else:
+        return Response({"error": "Processing failed"}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 
 # Endpoint to normalize audio
 @api_view(['POST'])