Update app.py

app.py

@@ -1,61 +1,59 @@
 import builtins
 import os
 import shutil
-import logging
 
-# 1. 劫持 input
+# 1. 劫持 input 防止崩溃
 builtins.input = lambda prompt="": "y"
 
 # 2. 设置环境
 os.environ["GENIE_DATA_DIR"] = "/app/GenieData"
 
-# 3. 手动下载数据
+# 3. 手动下载权重
 from huggingface_hub import snapshot_download
 if not os.path.exists("/app/GenieData/G2P"):
-    print("📦 Downloading GenieData...")
+    print("📦 Downloading GenieData Dependencies...")
     snapshot_download(repo_id="High-Logic/Genie", allow_patterns=["GenieData/*"], local_dir="/app", local_dir_use_symlinks=False)
 
-# 4. 导入引擎
+# 4. 导入库
 from fastapi import UploadFile, File, Form
 import genie_tts
 from genie_tts.Server import app as genie_app
-from genie_tts.ModelManager import model_manager
 
-# --- 增强版上传端点 ---
+# --- 核心：上传并设置参考音频 ---
 @genie_app.post("/upload_and_set")
 async def upload_and_set(
-    character_name: str = Form("Default"), # 改为大写 Default
+    character_name: str = Form("Default"),
     prompt_text: str = Form(...),
     language: str = Form("zh"),
     file: UploadFile = File(...)
 ):
-    # 强制将名字改为首字母大写，与引擎保持一致
     char_name = character_name.capitalize()
-    save_path = f"/app/uploaded_custom_ref.wav"
+    # 存放在固定位置
+    save_path = f"/app/uploaded_ref.wav"
     
     with open(save_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
     
+    print(f"🔄 Setting reference audio for {char_name}: {prompt_text}")
     try:
-        # 调用底层 model_manager 确保设置成功
-        model_manager.set_reference_audio(char_name, save_path, prompt_text, language)
-        print(f"✅ Success: Character '{char_name}' updated with text: {prompt_text}")
-        return {"message": f"Success! Character '{char_name}' style updated."}
+        # 🟢 调用官方顶级 API，这是最稳妥的
+        genie_tts.set_reference_audio(char_name, save_path, prompt_text, language)
+        return {"message": f"Successfully updated style for {char_name}"}
     except Exception as e:
-        print(f"❌ Error setting ref audio: {e}")
+        print(f"❌ Error: {e}")
         return {"error": str(e)}, 500
 
-# --- 调试端点：查看当前加载了哪些角色 ---
-@genie_app.get("/debug_chars")
-async def debug_chars():
-    return {
-        "loaded_characters": list(model_manager.loaded_characters.keys()),
-        "detail": {k: v.ref_audio_path for k, v in model_manager.loaded_characters.items()}
-    }
+# --- 极简调试：只看当前角色状态 ---
+@genie_app.get("/status")
+async def status():
+    return {"status": "ok", "engine": "Genie-TTS V2 Pro Plus"}
 
 if __name__ == "__main__":
-    # 加载角色，直接用 Default
-    model_manager.load_character("Default", "/app", "zh")
-    
-    # 启动服务器
+    # 启动前先预载角色模型
+    try:
+        genie_tts.load_character("Default", "/app", "zh")
+    except Exception as e:
+        print(f"Initial load error: {e}")
+
+    # 启动 FastAPI 服务 (HF Space 固定 7860)
     genie_tts.start_server(host="0.0.0.0", port=7860)