Update app.py

app.py

@@ -188,24 +188,22 @@ def inference(audio):
     os.makedirs("out", exist_ok=True)
     result = os.system(new_command)
     print(f"Demucs script result: {result}")
-    common_dir = os.path.dirname(audio)
+    directory_name = os.path.splitext(os.path.basename(audio))[0]
 
     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"
+        f"./out/mdx_extra_q/{directory_name}/vocals.wav",
+        f"./out/mdx_extra_q/{directory_name}/bass.wav",
+        f"./out/mdx_extra_q/{directory_name}/drums.wav",
+        f"./out/mdx_extra_q/{directory_name}/other.wav"
     ]
 
-    modified_files = [os.path.join(common_dir, os.path.basename(file)) for file in files]
-
-    for file in modified_files:
+    for file in files:
         if not os.path.isfile(file):
             print(f"File not found: {file}")
         else:
             print(f"File exists: {file}")
 
-    return modified_files
+    return files
 
 
 def toggle_audio_src(choice):