Update app.py

app.py

@@ -1,9 +1,9 @@
-from typing import Tuple
 import gradio as gr
 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,11 +13,7 @@ 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 = [
-        os.path.join(target_dir, f)
-        for f in os.listdir(target_dir)
-        if f.endswith((".wav", ".mp3", ".ogg"))
-    ]
+    audio_files = glob(f"{target_dir}/**/*.wav", recursive=True)
     if not audio_files:
         raise gr.Error("No audio files found in the zip archive.")
 
@@ -35,9 +31,6 @@ def preprocess(zip_file: str) -> str:
     data_dir = os.path.join(temp_dir, "_data")
     os.makedirs(data_dir)
     audio_files = extract_audio_files(zip_file, data_dir)
-    if not audio_files:
-        shutil.rmtree(temp_dir)
-        raise gr.Error("No audio files found in the zip archive.")
 
     pp = PreProcess(48000, temp_dir, 3.0, False)
     pp.pipeline_mp_inp_dir(data_dir, 4)