Update app.py

app.py

@@ -4,69 +4,49 @@ import tempfile
 import subprocess
 
 def separate_audio(audio_path, stem_count):
-
+    print(f"{audio_path=}")
     head, tail = os.path.split(audio_path)
+
     gradio_temp_path = head
     audio_filename = tail.split('.')[0]
+    print(f"{gradio_temp_path=}")
+    print(f"{audio_filename=}")
+    print(f"{stem_count=}")
 
     command = f"spleeter separate -p spleeter:{stem_count}stems {audio_path}"
     command = command.split()
+    print(f"{command=}")
 
     result = subprocess.run(command)
+    print(result)
 
-    outputs = []
+    paths = []
     if stem_count == 2:
-        accompaniment_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/accompaniment.wav"
-        vocals_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"
-        outputs.append({'description': 'Accompaniment', 'path': accompaniment_path})
-        outputs.append({'description': 'Vocals', 'path': vocals_path})
-    elif stem_count == 4:
-        vocals_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"
-        drums_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/drums.wav"
-        bass_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/bass.wav"
-        other_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/other.wav"
-        outputs.extend([
-            {'description': 'Vocals', 'path': vocals_path},
-            {'description': 'Drums', 'path': drums_path},
-            {'description': 'Bass', 'path': bass_path},
-            {'description': 'Other', 'path': other_path},
-        ])
-    elif stem_count == 5:
-        piano_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/piano.wav"
-        vocals_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"
-        drums_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/drums.wav"
-        bass_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/bass.wav"
-        other_path = f"{gradio_temp_path}/separated_audio/{audio_filename}/other.wav"
-        outputs.extend([
-            {'description': 'Vocals', 'path': vocals_path},
-            {'description': 'Piano', 'path': piano_path},
-            {'description': 'Drums', 'path': drums_path},
-            {'description': 'Bass', 'path': bass_path},
-            {'description': 'Other', 'path': other_path},
-        ])
-    return outputs
-
-def separate_audio_gradio(audio_file, stem_count):
-    with tempfile.NamedTemporaryFile(delete=False) as temp_file:        
-        temp_file.write(audio_file.read())
+        paths.append(('Accompaniment', f"{gradio_temp_path}/separated_audio/{audio_filename}/accompaniment.wav"))
+        paths.append(('Vocals', f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"))
+    elif stem_count == 4 or stem_count == 5:
+        paths.append(('Vocals', f"{gradio_temp_path}/separated_audio/{audio_filename}/vocals.wav"))
+        paths.append(('Drums', f"{gradio_temp_path}/separated_audio/{audio_filename}/drums.wav"))
+        paths.append(('Bass', f"{gradio_temp_path}/separated_audio/{audio_filename}/bass.wav"))
+        paths.append(('Other', f"{gradio_temp_path}/separated_audio/{audio_filename}/other.wav"))
+        if stem_count == 5:
+            paths.append(('Piano', f"{gradio_temp_path}/separated_audio/{audio_filename}/piano.wav"))
+
+    return paths
+
+def gradio_interface(audio_file, stem_count):
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+        temp_file.write(audio_file)
         temp_file_path = temp_file.name
-        
-    separate_audios = separate_audio(temp_file_path, stem_count)
-    
-    outputs = []
-    for audio in separate_audios:
-        outputs.append((audio['description'], gr.outputs.Audio(audio['path'])))
-    
-    return outputs
 
-input_audio = gr.components.File(label="Upload Audio File", type=["filepath",])
-input_stem_count = gr.components.Radio([2, 4, 5], label="Stem count")
+    separated_audios = separate_audio(temp_file_path, stem_count)
+    return [gr.Audio(file_path, label=description) for description, file_path in separated_audios]
+
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs=[gr.Audio(type="filepath"), gr.Radio([2, 4, 5])],
+    outputs=[gr.Audio(label="Output") for _ in range(5)],
+    live=True
+)
 
-gr.Interface(
-    fn=separate_audio_gradio,
-    inputs=[input_audio, input_stem_count],
-    outputs="label",
-    title="Audio Separator",
-    description="Separate audio into vocals and accompaniment using Spleeter",
-    capture_session=True
-).launch()
+iface.launch()