Update app.py

app.py

@@ -2,19 +2,20 @@ import os
 import uuid
 import subprocess
 import requests
+import base64
 from urllib.parse import urlparse
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import JSONResponse, HTMLResponse
 
 app = FastAPI()
 
-# Create the base tasks directory if it doesn't exist.
+# Base directory to store task files
 BASE_DIR = "tasks"
 os.makedirs(BASE_DIR, exist_ok=True)
 
 def add_status(log_list, message):
     log_list.append(message)
-    print(message)  # Log to console for debugging
+    print(message)  # Also log to console
 
 @app.post("/process")
 def process_audio(payload: dict):
@@ -69,9 +70,9 @@ def process_audio(payload: dict):
         add_status(status_log, f"Error downloading file: {e}")
         return JSONResponse(status_code=400, content={"status": status_log})
     
-    # Remove any query parameters from the URL for a clean filename
+    # Clean filename: remove query parameters using urlparse
     parsed = urlparse(audio_url)
-    path = parsed.path  # excludes query parameters
+    path = parsed.path
     ext = os.path.splitext(path)[1] or ".mp3"
     input_filename = f"input{ext}"
     input_filepath = os.path.join(task_dir, input_filename)
@@ -88,7 +89,7 @@ def process_audio(payload: dict):
         add_status(status_log, "Error: Input file does not exist after download.")
         return JSONResponse(status_code=500, content={"status": status_log})
     
-    # Run Spleeter using the latest command syntax:
+    # Run Spleeter using the updated syntax:
     # spleeter separate -p spleeter:2stems -o <task_dir> <input_filepath>
     spleeter_cmd = [
         "spleeter", "separate",
@@ -117,9 +118,22 @@ def process_audio(payload: dict):
     
     add_status(status_log, "Spleeter processing completed successfully.")
     
-    # Return task ID and status log (for debugging on the frontend)
+    # Read output files and encode them in base64
+    try:
+        with open(vocals_file, "rb") as f:
+            vocals_data = f.read()
+        with open(accompaniment_file, "rb") as f:
+            accomp_data = f.read()
+        vocals_b64 = base64.b64encode(vocals_data).decode("utf-8")
+        accomp_b64 = base64.b64encode(accomp_data).decode("utf-8")
+    except Exception as e:
+        add_status(status_log, f"Error reading output files: {e}")
+        return JSONResponse(status_code=500, content={"status": status_log})
+    
     return JSONResponse(content={
         "task_id": task_id,
+        "vocals": vocals_b64,
+        "accompaniment": accomp_b64,
         "status": status_log
     })
 
@@ -155,6 +169,14 @@ def index():
                     resultHTML += "<li>" + log + "</li>";
                 }
                 resultHTML += "</ul>";
+                if (data.vocals && data.accompaniment) {
+                    const vocalsDataURI = "data:audio/wav;base64," + data.vocals;
+                    const accompDataURI = "data:audio/wav;base64," + data.accompaniment;
+                    resultHTML += "<h3>Vocals</h3>";
+                    resultHTML += "<audio controls src='" + vocalsDataURI + "'></audio>";
+                    resultHTML += "<h3>Accompaniment</h3>";
+                    resultHTML += "<audio controls src='" + accompDataURI + "'></audio>";
+                }
                 if(data.task_id) {
                     resultHTML += `<p>Task ID: ${data.task_id}</p>`;
                 }