Update app.py

app.py

@@ -19,12 +19,19 @@ from audiocraft.data.audio import audio_write
 
 MODEL = None
 
-def split_process(audio):
-  os.makedirs("out", exist_ok=True)
-  write('test.wav', audio[0], audio[1])
-  os.system("python3 -m demucs.separate -n mdx_extra_q -d cpu test.wav -o out")
-  #return "./out/mdx_extra_q/test/vocals.wav","./out/mdx_extra_q/test/bass.wav","./out/mdx_extra_q/test/drums.wav","./out/mdx_extra_q/test/other.wav"
-  return "./out/mdx_extra_q/test/vocals.wav"
+def split_process(audio, chosen_out_track):
+    os.makedirs("out", exist_ok=True)
+    write('test.wav', audio[0], audio[1])
+    os.system("python3 -m demucs.separate -n mdx_extra_q -d cpu test.wav -o out")
+    #return "./out/mdx_extra_q/test/vocals.wav","./out/mdx_extra_q/test/bass.wav","./out/mdx_extra_q/test/drums.wav","./out/mdx_extra_q/test/other.wav"
+    if chosen_out_track == "vocals":
+        return "./out/mdx_extra_q/test/vocals.wav"
+    elif chosen_out_track == "bass":
+        return "./out/mdx_extra_q/test/bass.wav"
+    elif chosen_out_track == "drums":
+        return "./out/mdx_extra_q/test/drums.wav"
+    elif chosen_out_track == "other":
+        return "./out/mdx_extra_q/test/other.wav"
     
 def load_model(version):
     print("Loading model", version)
@@ -88,6 +95,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             with gr.Row():
                 uploaded_sound = gr.Audio(type="numpy", label="Input", source="microphone")
+                chosen_track = gr.Radio(["vocals", "bass", "drums", "other"], label="Track", info="Which track do you want to mashup ?", value="vocals")
                 load_sound_btn = gr.Button('Load sound')
                 #split_vocals = gr.Audio(type="filepath", label="Vocals")
                 #split_bass = gr.Audio(type="filepath", label="Bass")
@@ -109,7 +117,7 @@ with gr.Blocks() as demo:
             #   cfg_coef = gr.Number(label="Classifier Free Guidance", value=3.0, interactive=True)
         with gr.Column():
             output = gr.Audio(label="Generated Music")
-    load_sound_btn.click(split_process, inputs=[uploaded_sound], outputs=[melody])
+    load_sound_btn.click(split_process, inputs=[uploaded_sound, chosen_track], outputs=[melody])
     submit.click(predict, inputs=[music_prompt, melody, duration], outputs=[output])
     
     gr.Markdown(