Update app.py

app.py

@@ -3,13 +3,27 @@ import gradio as gr
 from scipy.io.wavfile import write
 from gradio import components
 
-def inference(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")
-  time.sleep(5)  # wait for 5 seconds
-  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"
+    def inference(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")
+        
+        # Paths to the output files
+        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"]
+    
+        # Check for the existence of the files
+        start_time = time.time()
+        while True:
+            if all(os.path.exists(file) for file in files):
+                break
+            elif time.time() - start_time > 30:  # timeout after 30 seconds
+                raise TimeoutError("Timeout while waiting for output files to be created.")
+            time.sleep(1)  # wait a bit before checking again
+    
+        return tuple(files)
   
 title = "Demucs"
 description = "Forked from here https://huggingface.co/spaces/akhaliq/demucs/ and changed to updated demucs pip source. Gradio demo for Demucs: Music Source Separation in the Waveform Domain. To use it, simply upload your audio, or click one of the examples to load them. Read more at the links below."