Update app.py

app.py

@@ -1,26 +1,34 @@
-import os
 import builtins
-import shutil
-import io
-import time  # 🔴 新增
-import uvicorn
-from fastapi import FastAPI, UploadFile, File, Form, HTTPException
-from fastapi.responses import StreamingResponse
-import genie_tts
-from huggingface_hub import snapshot_download
+import os
 
-# --- 初期化屏蔽 ---
+# 🔴 核心：在所有 import 之前，必须先劫持 input，否则导入 genie_tts 时会因为 input 阻塞导致崩溃
 builtins.input = lambda prompt="": "y"
 os.environ["GENIE_DATA_DIR"] = "/app/GenieData"
 
+# 🔴 核心：在导入之前，必须先确保数据已下载
+from huggingface_hub import snapshot_download
 if not os.path.exists("/app/GenieData/G2P"):
+    print("📦 Downloading GenieData Assets...")
     snapshot_download(repo_id="High-Logic/Genie", allow_patterns=["GenieData/*"], local_dir="/app", local_dir_use_symlinks=False)
 
+# 现在可以安全地 import 了
+import shutil
+import io
+import time
+import uvicorn
+from fastapi import FastAPI, UploadFile, File, Form, HTTPException
+from fastapi.responses import StreamingResponse
+import genie_tts
+
+# ---------------------------------------------------------
+# 服务器设置
+# ---------------------------------------------------------
 app = FastAPI()
 
 DEFAULT_REF_PATH = "/app/ref.wav"
 DEFAULT_REF_TEXT = "琴是个称职的好团长。看到她认真工作的样子，就连我也忍不住想要多帮她一把。"
 
+print("⚡ Starting Genie Engine...")
 genie_tts.load_character("Default", "/app", "zh")
 
 @app.post("/upload_and_tts")
@@ -31,40 +39,43 @@ async def upload_and_tts(
     file: UploadFile = File(...)
 ):
     try:
-        # 🔴 关键改动：使用时间戳作为文件名，确保路径唯一，强行冲掉缓存
-        timestamp = int(time.time())
-        save_path = f"/app/ref_{timestamp}.wav"
+        # 使用时间戳文件名强行刷新缓存
+        ts = int(time.time() * 1000)
+        save_path = f"/app/ref_{ts}.wav"
         
         with open(save_path, "wb") as buffer:
             shutil.copyfileobj(file.file, buffer)
         
-        print(f"🔥 [Action] Loading NEW unique audio: {save_path}")
-        
-        # 马上设置新参考
+        print(f"🔥 [Custom] Loaded new unique audio: {save_path}")
         genie_tts.set_reference_audio("Default", save_path, prompt_text, language)
         
-        out_path = f"/app/out_{timestamp}.wav"
+        out_path = f"/app/out_{ts}.wav"
         genie_tts.tts("Default", text, save_path=out_path, play=False)
         
-        # 合成完后清理掉参考音频，防止塞满硬盘
-        # 我们返回后再删
+        # 定义生成器，在发送完后尝试清理临时文件（可选）
         def iterfile():
             with open(out_path, "rb") as f:
                 yield from f
-            # 💡 提示：这里可以在生成后做一个延时删，但为了稳，先留着不删也可以
-            
+            # 这里的清理可以防止占用过多存储空间
+            try: os.remove(save_path); os.remove(out_path)
+            except: pass
+
         return StreamingResponse(iterfile(), media_type="audio/wav")
     except Exception as e:
-        print(f"❌ Error: {e}")
+        print(f"❌ Error in upload/tts: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/tts")
 async def safe_tts(data: dict):
-    # 标准模式强制回归内置原始音频
-    genie_tts.set_reference_audio("Default", DEFAULT_REF_PATH, DEFAULT_REF_TEXT, "zh")
-    out_path = "/app/out_std.wav"
-    genie_tts.tts("Default", data.get("text", ""), save_path=out_path, play=False)
-    return StreamingResponse(open(out_path, "rb"), media_type="audio/wav")
+    print(f"🛡️ [Reset] Back to standard Qin voice.")
+    try:
+        genie_tts.set_reference_audio("Default", DEFAULT_REF_PATH, DEFAULT_REF_TEXT, "zh")
+        out_path = "/app/out_std.wav"
+        genie_tts.tts("Default", data.get("text", ""), save_path=out_path, play=False)
+        return StreamingResponse(open(out_path, "rb"), media_type="audio/wav")
+    except Exception as e:
+        print(f"❌ Error in std tts: {e}")
+        raise HTTPException(status_code=404, detail=str(e))
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)