Update app.py

app.py

@@ -75,10 +75,16 @@ def analyze(path):
   #Measure time for inference
   end = time.time()
   elapsed_time = end-start
+
+  # Get the base name of the file
+  file_name = os.path.basename(path)
+    
+  # Remove the extension from the file name
+  file_name_without_extension = os.path.splitext(file_name)[0]
     
   bass_path, drums_path, other_path, vocals_path = None, None, None, None
-  for root, dirs, files in os.walk(base_dir):
-    for file_name in files:
+  for root, dirs, files in os.walk(f"./demix/htdemucs/{file_name_without_extension}/"):
+    for file in files:
       file_path = os.path.join(root, file_name)
       print(file_path)
       if "bass.wav" in file_path:
@@ -117,10 +123,14 @@ with gr.Blocks() as demo:
     #)
     elapsed_time = gr.Textbox(label='Overall inference time', scale=1)
   with gr.Column():
-    bass_path = gr.Textbox(label='bass_path', scale=1)
-    drums_path = gr.Textbox(label='drums_path', scale=1)
-    other_path = gr.Textbox(label='other_path', scale=1)
-    vocals_path = gr.Textbox(label='vocals_path', scale=1)
+    bass = gr.Audio(label='bass', show_share_button=False)
+    vocals =gr.Audio(label='vocals', show_share_button=False)
+    other = gr.Audio(label='other', show_share_button=False)
+    drums =gr.Audio(label='drums', show_share_button=False)
+    #bass_path = gr.Textbox(label='bass_path', scale=1)
+    #drums_path = gr.Textbox(label='drums_path', scale=1)
+    #other_path = gr.Textbox(label='other_path', scale=1)
+    #vocals_path = gr.Textbox(label='vocals_path', scale=1)
   #gr.Examples(
   #  examples=[
   #    './assets/NewJeans - Super Shy.mp3',