Update app.py

app.py

@@ -57,13 +57,16 @@ def analyze(path):
     keep_byproducts=True,  # TODO: remove this
   )
 
-
+  for root, dirs, files in os.walk(f"./struct"):
+    for file_path in files:
+      file_path = os.path.join(root, file_path)
+      print(file_path)
   
-  #fig = allin1.visualize(
-  #  result,
-  #  multiprocess=False,
-  #)
-  #fig.set_dpi(300)
+  fig = allin1.visualize(
+    result,
+    multiprocess=False,
+  )
+  fig.set_dpi(300)
 
   #allin1.sonify(
   #  result,
@@ -95,9 +98,9 @@ def analyze(path):
         other_path = file_path
       if "drums.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
+  return result.bpm, fig, elapsed_time, bass_path, drums_path, other_path, vocals_path
 
 
 with gr.Blocks() as demo:
@@ -111,7 +114,7 @@ with gr.Blocks() as demo:
     show_download_button=False,
   )
   button = gr.Button('Analyze', variant='primary')
-  #output_viz = gr.Plot(label='Visualization')
+  output_viz = gr.Plot(label='Visualization')
   with gr.Row():
     output_bpm = gr.Textbox(label='BPM', scale=1)
     #output_sonif = gr.Audio(
@@ -146,7 +149,7 @@ with gr.Blocks() as demo:
     fn=analyze,
     inputs=input_audio_path,
     #outputs=[output_bpm, output_viz, output_sonif, elapsed_time],
-    outputs=[output_bpm, elapsed_time, bass, drums, other, vocals],
+    outputs=[output_bpm, output_viz, elapsed_time, bass, drums, other, vocals],
     api_name='analyze',
   )