feat: wav包含子文件夹 feat:新接口-强行重下wav

api.py

@@ -40,8 +40,18 @@ def download_voices(bForce=False):
     if not os.path.exists(voices_dir):
         os.makedirs(voices_dir)
     
-    pmt_files = [f for f in os.listdir(voices_dir) if f.endswith(".pmt")]
-    if bForce or not pmt_files:
+    # 改进：递归查找所有子目录中的.pmt文件，找到一个就返回
+    pmt_file_found = None
+    for root, dirs, files in os.walk(voices_dir):
+        for file in files:
+            if file.endswith(".pmt"):
+                pmt_file_found = file
+                break
+        if pmt_file_found:
+            break
+    
+    # 如果没有找到.pmt文件且需要强制下载
+    if bForce or pmt_file_found is None:
         # 如果没有 .pmt 文件，尝试从远程下载
         voice_download_url = os.getenv("VOICE_DOWNLOAD_URL")
         
@@ -195,17 +205,27 @@ def delete_voice(name: str, token: str = Depends(verify_token)):
 @app.get("/voices")
 def get_voices(token: str = Depends(verify_token)):
     download_voices()
-    # 获取所有 .pmt 文件
-    pmt_files = [f for f in os.listdir("./voices") if f.endswith(".pmt")]
-    # 提取文件名（去掉 .pmt 后缀）
-    voices = [f.split(".")[0] for f in pmt_files]
+    # 获取所有 .pmt 文件（递归搜索子目录）
+    pmt_files = []
+    for root, dirs, files in os.walk("./voices"):
+        for file in files:
+            if file.endswith(".pmt"):
+                # 获取文件相对于当前工作目录的完整相对路径
+                full_path = os.path.join(root, file)
+                relative_path = os.path.relpath(full_path, "./voices")
+                pmt_files.append(relative_path[:-4])
+    
     # 确保对应的 .wav 文件也存在
     valid_voices = []
-    for voice in voices:
+    for voice in pmt_files:
         if os.path.exists(f"./voices/{voice}.wav"):
             valid_voices.append(voice)
     return {"voices": valid_voices}
 
+@app.post("/re_download_voices")
+def download_voices_file(token: str = Depends(verify_token)):
+    download_voices(True)
+
 
 # ↓↓↓↓↓↓↓↓↓无需验证↓↓↓↓↓↓↓↓
 @app.get("/")