fix: 漏掉音频下载

api.py

@@ -33,6 +33,38 @@ class GenerateRequest(BaseModel):
     do_normalize: bool = True
     denoise: bool = True
 
+def download_voices(bForce=False):
+    # 检查 /workspace/voices/ 目录中是否有 .pmt 文件
+    voices_dir = "/workspace/voices"
+    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 文件，尝试从远程下载
+        voice_download_url = os.getenv("VOICE_DOWNLOAD_URL")
+        
+        if voice_download_url:
+            try:
+                response = requests.get(voice_download_url)
+                response.raise_for_status()
+                
+                # 保存下载的zip文件
+                zip_path = f"{voices_dir}/voices.zip"
+                with open(zip_path, "wb") as f:
+                    f.write(response.content)
+                
+                # 解压zip文件
+                with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+                    zip_ref.extractall(voices_dir)
+                
+                # 删除临时zip文件
+                os.remove(zip_path)
+                
+            except Exception as e:
+                print_with_time(f"Failed to download and extract voices: {e}")
+                raise HTTPException(status_code=500, detail="Failed to download voice files")
+
 # 队列相关变量
 task_queue = queue.Queue()
 output_dir = "./output"
@@ -70,7 +102,8 @@ async def process_queue():
 
             if model is None:
                 raise RuntimeError("Failed to initialize model")
-
+            
+            download_voices()
             print_with_time(f"Generating audio for : '{text[:60]}...'")
             with open(f"./voices/{request.voice}.pmt", 'r', encoding='utf-8') as f:
                 wav = model.generate(
@@ -120,39 +153,6 @@ async def get_generate_result(task_id: str, token: str = Depends(verify_token)):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to read result file: {str(e)}")
 
-
-def download_voices():
-    # 检查 /workspace/voices/ 目录中是否有 .pmt 文件
-    voices_dir = "/workspace/voices"
-    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 not pmt_files:
-        # 如果没有 .pmt 文件，尝试从远程下载
-        voice_download_url = os.getenv("VOICE_DOWNLOAD_URL")
-        
-        if voice_download_url:
-            try:
-                response = requests.get(voice_download_url)
-                response.raise_for_status()
-                
-                # 保存下载的zip文件
-                zip_path = f"{voices_dir}/voices.zip"
-                with open(zip_path, "wb") as f:
-                    f.write(response.content)
-                
-                # 解压zip文件
-                with zipfile.ZipFile(zip_path, 'r') as zip_ref:
-                    zip_ref.extractall(voices_dir)
-                
-                # 删除临时zip文件
-                os.remove(zip_path)
-                
-            except Exception as e:
-                print_with_time(f"Failed to download and extract voices: {e}")
-                raise HTTPException(status_code=500, detail="Failed to download voice files")
-
 @app.post("/upload_voice")
 def upload_voice(name: str = Form(...), wav: bytes = File(...), prompt: str = Form(...), token: str = Depends(verify_token)):
     # 保存wav文件