Update app.py

app.py

@@ -12,36 +12,25 @@ def separate_audio(audio_path, stem_count):
     print(f"{gradio_temp_path=}")
     print(f"{audio_filename=}")
     print(f"{stem_count=}")
-
-    # Construct the spleeter command
-    command = f"spleeter separate -p spleeter:{stem_count}stems -o {gradio_temp_path} {audio_path}"
+    os.mkdir(gradio_temp_path + f"/separated_audio/{audio_filename}")
+    command = f"spleeter separate -p spleeter:{stem_count}stems {audio_path}"
     command = command.split()
     print(f"{command=}")
 
-    # Run the spleeter command and capture the output
-    result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    print(f"stdout: {result.stdout.decode('utf-8')}")
-    print(f"stderr: {result.stderr.decode('utf-8')}")
-
-    # Check if the command executed successfully
-    if result.returncode != 0:
-        print(f"Error executing command: {result.returncode}")
-        return []
+    result = subprocess.run(command)
+    print(result)
 
-    # Generate a random number for unique folder naming
-    randomnumber = str(random.randint(111111111, 999999999))
     paths = []
-    # Define the output paths for the separated stems
     if stem_count == 2:
-        paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/accompaniment.wav"))
-        paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/vocals.wav"))
+        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/accompaniment.wav")
+        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/vocals.wav")
     elif stem_count == 4 or stem_count == 5:
-        paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/vocals.wav"))
-        paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/drums.wav"))
-        paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/bass.wav"))
-        paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/other.wav"))
+        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/vocals.wav")
+        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/drums.wav")
+        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/bass.wav")
+        paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/other.wav")
         if stem_count == 5:
-            paths.append(os.path.join(gradio_temp_path, f"separated_audio/{audio_filename}/piano.wav"))
+            paths.append(gradio_temp_path + f"/separated_audio/{audio_filename}/piano.wav")
 
     # Return the audio file paths as a list of Audio components
     return [gr.Audio(path) for path in paths]
@@ -50,7 +39,7 @@ def separate_audio(audio_path, stem_count):
 iface = gr.Interface(
     fn=separate_audio,
     inputs=[gr.Audio(type="filepath"), gr.Radio([2, 4, 5])],
-    outputs=[gr.Audio() for _ in range(2)]  # Adjust the range according to the max number of stems
+    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')]
 )
 
 # Launch the interface