Update app.py

app.py

@@ -25,7 +25,23 @@ def print_directory_contents(path):
 
 # Print the ckpt directory contents
 print_directory_contents('./ckpt')
-            
+
+def get_wav_files(path):
+    wav_files = []  # Initialize an empty list to store the paths of .wav files
+    for root, dirs, files in os.walk(path):
+        level = root.replace(path, '').count(os.sep)
+        indent = ' ' * 4 * (level)
+        print(f"{indent}{os.path.basename(root)}/")
+        subindent = ' ' * 4 * (level + 1)
+        for f in files:
+            file_path = os.path.join(root, f)
+            if f.lower().endswith('.wav'):
+                wav_files.append(file_path)  # Add .wav file paths to the list
+                print(f"{subindent}{file_path}")
+            else:
+                print(f"{subindent}{f}")
+    return wav_files  # Return the list of .wav file paths
+
 def infer(video_in):
 
     # Need to find path to gradio temp vid from video input
@@ -57,6 +73,8 @@ def infer(video_in):
     # Need to find where are the results stored, default should be "./outputs/tmp"
     # Print the outputs directory contents
     print_directory_contents('./outputs/tmp')
+    wave_files = get_wav_files('./outputs/tmp')
+    print(wave_files)
     return "done"
 
 with gr.Blocks() as demo: