Update app.py

app.py

@@ -161,7 +161,34 @@ def predict_full(model, text, melody, duration, topk, topp, temperature, cfg_coe
     outs = _do_predictions(
         [text], [melody], duration, progress=True,
         top_k=topk, top_p=topp, temperature=temperature, cfg_coef=cfg_coef)
-    return outs[0]
+    
+    return inference(outs[0])
+
+def inference(audio):
+    os.makedirs("out", exist_ok=True)
+    write('test.wav', audio[0], audio[1])
+
+    command = "python3 -m demucs.separate -n mdx_extra_q -d cpu test.wav -o out"
+    process = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    print("Demucs script output:", process.stdout.decode())
+  
+    os.makedirs("out", exist_ok=True)
+    write('test.wav', audio[0], audio[1])
+    result = os.system("python3 -m demucs.separate -n mdx_extra_q -d cpu test.wav -o out")
+    print(f"Demucs script result: {result}")
+
+    # Check if files exist before returning
+    files = ["./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"]
+    for file in files:
+        if not os.path.isfile(file):
+            print(f"File not found: {file}")
+        else:
+            print(f"File exists: {file}")
+
+    return files
 
 
 def toggle_audio_src(choice):
@@ -204,7 +231,12 @@ def ui_full(launch_kwargs):
                     temperature = gr.Number(label="Temperature", value=1.0, interactive=True)
                     cfg_coef = gr.Number(label="Classifier Free Guidance", value=3.0, interactive=True)
             with gr.Column():
-                output = gr.Audio(label="Generated Music")
+                output = [
+    gr.Audio(audio_file1, label="Generated Music 1"),
+    gr.Audio(audio_file2, label="Generated Music 2"),
+    gr.Audio(audio_file3, label="Generated Music 3"),
+    gr.Audio(audio_file4, label="Generated Music 4")
+]
         submit.click(predict_full,
                      inputs=[model, text, melody, duration, topk, topp, temperature, cfg_coef],
                      outputs=[output])