Update app.py

app.py

@@ -43,19 +43,21 @@ HEADER = """
 
 CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
 
+base_dir = "/tmp/gradio/"
 
 def analyze(path):
   #Measure time for inference
   start = time.time()
     
   path = Path(path)
-  result, bass_path, drums_path, other_path, vocals_path = allin1.analyze(
+  result= allin1.analyze(
     path,
     out_dir='./struct',
     multiprocess=False,
     keep_byproducts=True,  # TODO: remove this
   )
 
+
   
   #fig = allin1.visualize(
   #  result,
@@ -73,7 +75,20 @@ def analyze(path):
   #Measure time for inference
   end = time.time()
   elapsed_time = end-start
-
+    
+  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:
+      file_path = os.path.join(root, file_name)
+      if "bass.wav" in file_path:
+        bass_path = file_path
+      if "vocals.wav" in file_path:
+        vocals_path = file_path
+      if "bass.wav" in file_path:
+        other_path = file_path
+      if "bass.wav" in file_path:
+        drums_path = file_path
+  
   #return result.bpm, fig, sonif_path, elapsed_time
   return result.bpm, elapsed_time, bass_path, drums_path, other_path, vocals_path