Upload 2 files

Dockerfile

@@ -7,16 +7,21 @@ RUN apt-get update && apt-get install -y \
     git ffmpeg build-essential cmake libmecab-dev mecab-ipadic-utf8 \
     && rm -rf /var/lib/apt/lists/*
 
-# 安装所有必要库
+# 安装 genie_tts 的依赖（不安装 genie_tts 本身，使用本地源码）
 RUN pip install --no-cache-dir \
-    git+https://github.com/High-Logic/Genie-TTS.git \
-    python-multipart uvicorn fastapi
+    onnx onnxruntime==1.22.1 tokenizers numpy soundfile soxr pyyaml \
+    python-multipart uvicorn fastapi huggingface_hub
 
-# 镜像权限处理
+# 复制所有文件（包括本地 genie_tts 源码）
 COPY . /app
+
+# 预下载 GenieData 资源（避免每次启动都下载）
+RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='High-Logic/Genie', allow_patterns=['GenieData/*'], local_dir='/app', local_dir_use_symlinks=False)"
+
+# 镜像权限处理
 RUN chmod -R 777 /app
 
 EXPOSE 7860
 
 # 使用直接启动模式
-CMD ["python", "app.py"]
+CMD ["python", "app.py"]

app.py

@@ -1,25 +1,34 @@
 import builtins
 import os
+import sys
 import shutil
 import io
 import time
 import uvicorn
 from fastapi import FastAPI, UploadFile, File, Form, HTTPException
 from fastapi.responses import StreamingResponse
-from huggingface_hub import snapshot_download
 
 # 🔴 核心：在所有 import 之前，必须先劫持 input
 builtins.input = lambda prompt="": "y"
+
+# 使用本地 genie_tts 源码（而非已安装的包）
+# 将当前目录添加到 sys.path，确保优先加载本地模块
+current_dir = os.path.dirname(os.path.abspath(__file__))
+if current_dir not in sys.path:
+    sys.path.insert(0, current_dir)
+
 # 适配 Space 路径，本地运行时请确保此目录存在
 os.environ["GENIE_DATA_DIR"] = "/app/GenieData"
 
-# 下载环境
-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)
+# 注释掉自动下载逻辑，假设 GenieData 已预装在镜像中
+# if not os.path.exists("/app/GenieData/G2P"):
+#     print("📦 Downloading GenieData Assets...")
+#     from huggingface_hub import snapshot_download
+#     snapshot_download(repo_id="High-Logic/Genie", allow_patterns=["GenieData/*"], local_dir="/app", local_dir_use_symlinks=False)
 
 import genie_tts
 
+
 app = FastAPI()
 
 # 角色模型存放根目录