Update app.py

app.py

@@ -13,33 +13,33 @@ def separate_audio(audio_path, stem_count):
     print(f"{audio_filename=}")
     print(f"{stem_count=}")
 
-    command = f"spleeter separate -p spleeter:{stem_count}stems {audio_path}"
+    # Ensure the output directory exists
+    randomnumber = str(random.randint(111111111, 999999999))
+    output_folder = f"{gradio_temp_path}/separated_audio/{randomnumber}"
+    os.makedirs(output_folder, exist_ok=True)
+
+    command = f"spleeter separate -p spleeter:{stem_count}stems -o {output_folder} {audio_path}"
     command = command.split()
     print(f"{command=}")
 
     result = subprocess.run(command, capture_output=True)
     print(result)
     if result.returncode != 0:
-        print(f"Error: {result.stderr}")
+        print(f"Error: {result.stderr.decode('utf-8')}")
         return []
 
-    randomnumber = str(random.randint(111111111, 999999999))
-    output_folder = f"{gradio_temp_path}/separated_audio/{randomnumber}"
-    os.makedirs(output_folder, exist_ok=True)  # Ensure the output directory exists
-
+    # Check if the separated files exist and return their paths
     paths = []
     stems = ['vocals', 'drums', 'bass', 'other', 'piano']
-    for i, stem in enumerate(stems[:stem_count]):
-        stem_path = os.path.join(output_folder, f"{stem}.wav")
+    for stem in stems[:stem_count]:
+        stem_path = os.path.join(output_folder, stem, f"{audio_filename}_{stem}.wav")
+        print(f"Checking {stem_path}")
         if os.path.exists(stem_path):
-            paths.append((stem.capitalize(), stem_path))
-
-    # Return audio components for Gradio interface
-    audio_outputs = []
-    for label, path in paths:
-        audio_outputs.append(gr.Audio(file=path, label=label))
+            paths.append(gr.Audio(file=stem_path, label=stem.capitalize()))
+        else:
+            print(f"File not found: {stem_path}")
 
-    return audio_outputs
+    return paths
 
 iface = gr.Interface(
     fn=separate_audio,