Update app.py

app.py

@@ -13,38 +13,31 @@ def separate_audio(audio_path, stem_count):
     print(f"{audio_filename=}")
     print(f"{stem_count=}")
 
-    # 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 = f"spleeter separate -p spleeter:{stem_count}stems {audio_path}"
     command = command.split()
     print(f"{command=}")
 
-    result = subprocess.run(command, capture_output=True)
+    result = subprocess.run(command)
     print(result)
-    if result.returncode != 0:
-        print(f"Error: {result.stderr.decode('utf-8')}")
-        return []
-
-    # Check if the separated files exist and return their paths
+    randomnumber = str(random.randint(111111111, 999999999))
     paths = []
-    stems = ['vocals', 'drums', 'bass', 'other', 'piano']
-    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(gr.Audio(file=stem_path, label=stem.capitalize()))
-        else:
-            print(f"File not found: {stem_path}")
+    if stem_count == 2:
+        paths.append(('Accompaniment', f"{gradio_temp_path}/separated_audio/{randomnumber}/accompaniment.wav"))
+        paths.append(('Vocals', f"{gradio_temp_path}/separated_audio/{randomnumber}/vocals.wav"))
+    elif stem_count == 4 or stem_count == 5:
+        paths.append(('Vocals', f"{gradio_temp_path}/separated_audio/{randomnumber}/vocals.wav"))
+        paths.append(('Drums', f"{gradio_temp_path}/separated_audio/{randomnumber}/drums.wav"))
+        paths.append(('Bass', f"{gradio_temp_path}/separated_audio/{randomnumber}/bass.wav"))
+        paths.append(('Other', f"{gradio_temp_path}/separated_audio/{randomnumber}/other.wav"))
+        if stem_count == 5:
+            paths.append(('Piano', f"{gradio_temp_path}/separated_audio/{randomnumber}/piano.wav"))
 
     return paths
 
 iface = gr.Interface(
     fn=separate_audio,
     inputs=[gr.Audio(type="filepath"), gr.Radio([2, 4, 5])],
-    outputs=[gr.Audio(label="Accompaniment"), gr.Audio(label="Vocals"), gr.Audio(label="Drums"), gr.Audio(label="Bass"), gr.Audio(label="Other"), gr.Audio(label="Piano")]
+    outputs="text",
 )
 
 iface.launch()