Update app.py

app.py

@@ -15,7 +15,7 @@ os.makedirs(BASE_DIR, exist_ok=True)
 
 def add_status(log_list, message):
     log_list.append(message)
-    print(message)
+    print(message)  # Also log to console
 
 @app.post("/process")
 def process_audio(payload: dict):
@@ -70,7 +70,7 @@ def process_audio(payload: dict):
         add_status(status_log, f"Error downloading file: {e}")
         return JSONResponse(status_code=400, content={"status": status_log})
     
-    # Remove query parameters for a clean filename
+    # Clean filename: remove query parameters using urlparse
     parsed = urlparse(audio_url)
     path = parsed.path
     ext = os.path.splitext(path)[1] or ".mp3"
@@ -89,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 updated syntax:
+    # Run Spleeter using the updated syntax:
     # spleeter separate -p spleeter:2stems -o <task_dir> <input_filepath>
     spleeter_cmd = [
         "spleeter", "separate",
@@ -106,22 +106,9 @@ def process_audio(payload: dict):
         add_status(status_log, "Spleeter processing failed: " + e.stderr)
         return JSONResponse(status_code=500, content={"status": status_log})
     
-    # Determine expected output folder.
-    # By default, Spleeter creates a folder with the base name of the input file.
+    # Spleeter creates an output folder with the base name of the input file.
     base_name = os.path.splitext(input_filename)[0]
     output_folder = os.path.join(task_dir, base_name)
-    
-    # If the expected folder doesn't exist, attempt to auto-detect the correct folder.
-    if not os.path.exists(output_folder):
-        add_status(status_log, f"Expected output folder '{output_folder}' not found. Searching for subdirectories.")
-        subdirs = [d for d in os.listdir(task_dir) if os.path.isdir(os.path.join(task_dir, d))]
-        if len(subdirs) == 1:
-            output_folder = os.path.join(task_dir, subdirs[0])
-            add_status(status_log, f"Detected output folder as: {output_folder}")
-        else:
-            add_status(status_log, "Error: Unable to detect unique output folder.")
-            return JSONResponse(status_code=500, content={"status": status_log})
-    
     vocals_file = os.path.join(output_folder, "vocals.wav")
     accompaniment_file = os.path.join(output_folder, "accompaniment.wav")
     
@@ -131,7 +118,7 @@ def process_audio(payload: dict):
     
     add_status(status_log, "Spleeter processing completed successfully.")
     
-    # Read output files and encode in base64
+    # Read output files and encode them in base64
     try:
         with open(vocals_file, "rb") as f:
             vocals_data = f.read()