Update app.py

app.py

@@ -3,7 +3,6 @@ import zipfile
 import os
 import tempfile
 import shutil
-from glob import glob
 from infer.modules.train.preprocess import PreProcess, preprocess_trainset
 from infer.modules.train.extract.extract_f0_rmvpe import FeatureInput
 from zero import zero
@@ -13,9 +12,13 @@ def extract_audio_files(zip_file: str, target_dir: str) -> list[str]:
     with zipfile.ZipFile(zip_file, "r") as zip_ref:
         zip_ref.extractall(target_dir)
 
-    audio_files = glob(f"{target_dir}/**/*.wav", recursive=True)
+    audio_files = [
+        os.path.join(target_dir, f)
+        for f in os.listdir(target_dir)
+        if f.endswith((".wav", ".mp3", ".ogg"))
+    ]
     if not audio_files:
-        raise gr.Error("No audio files found in the zip archive.")
+        raise gr.Error("No audio files found at the top level of the zip file")
 
     return audio_files