Upload 3 files

Dockerfile

@@ -0,0 +1,14 @@
+FROM python:3.9-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+RUN mkdir -p audio_files
+
+ENV TTS_HOST=${TTS_HOST}
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,188 @@
+from fastapi import FastAPI, HTTPException, Request, Body
+from fastapi.responses import JSONResponse, FileResponse
+from fastapi.templating import Jinja2Templates
+import httpx
+import os
+import uuid
+from pathlib import Path
+import aiofiles
+from typing import Optional, Dict, Any
+from datetime import datetime, timedelta
+import hashlib
+import json
+
+app = FastAPI()
+
+templates = Jinja2Templates(directory="templates")
+
+# 配置
+HOST = os.getenv("TTS_HOST", "https://99i-tts.hf.space")  # 替换为实际环境变量名
+AUDIO_DIR = "audio_files"
+CACHE_EXPIRE_HOURS = 24  # 缓存过期时间(小时)
+Path(AUDIO_DIR).mkdir(exist_ok=True)
+
+# 内存缓存 (生产环境建议使用Redis等)
+cache_store = {}
+
+def get_cache_key(params: Dict[str, Any]) -> str:
+    """生成缓存键"""
+    param_str = json.dumps(params, sort_keys=True)
+    return hashlib.md5(param_str.encode()).hexdigest()
+
+async def download_audio(params: dict, use_post: bool = False) -> str:
+    """下载音频文件并保存到本地"""
+    cache_key = get_cache_key(params)
+    
+    # 检查缓存
+    if cache_key in cache_store:
+        cached_item = cache_store[cache_key]
+        if datetime.now() < cached_item["expire_time"]:
+            return cached_item["file_name"]
+    
+    async with httpx.AsyncClient() as client:
+        try:
+            if use_post:
+                response = await client.post(f"{HOST}/tts", json=params)
+            else:
+                response = await client.get(f"{HOST}/tts", params=params)
+            
+            response.raise_for_status()
+            
+            # 生成唯一文件名
+            file_name = f"{uuid.uuid4()}.mp3"
+            file_path = Path(AUDIO_DIR) / file_name
+            
+            # 保存音频文件
+            async with aiofiles.open(file_path, "wb") as f:
+                await f.write(response.content)
+            
+            # 更新缓存
+            cache_store[cache_key] = {
+                "file_name": file_name,
+                "expire_time": datetime.now() + timedelta(hours=CACHE_EXPIRE_HOURS)
+            }
+            
+            return file_name
+        except httpx.HTTPStatusError as e:
+            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+
+@app.get("/tts")
+async def text_to_speech_get(
+    request: Request,
+    t: str,
+    v: Optional[str] = "zh-CN-XiaoxiaoMultilingualNeural",
+    r: Optional[int] = 0,
+    p: Optional[int] = 0,
+    o: Optional[str] = "audio-24khz-48kbitrate-mono-mp3"
+):
+    """GET方式转发TTS请求并返回在线播放链接"""
+    params = {
+        "t": t,
+        "v": v,
+        "r": r,
+        "p": p,
+        "o": o
+    }
+    
+    try:
+        # 先尝试GET，失败后尝试POST
+        try:
+            file_name = await download_audio(params, use_post=False)
+        except:
+            file_name = await download_audio(params, use_post=True)
+            
+        base_url = str(request.base_url)
+        audio_url = f"{base_url}audio/{file_name}"
+        return {"status": "success", "audio_url": audio_url, "cached": False}
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={"status": "error", "message": str(e)}
+        )
+
+@app.post("/tts")
+async def text_to_speech_post(
+    request: Request,
+    data: Dict[str, Any] = Body(...)
+):
+    """POST方式转发TTS请求并返回在线播放链接"""
+    params = {
+        "t": data.get("t"),
+        "v": data.get("v", "zh-CN-XiaoxiaoMultilingualNeural"),
+        "r": data.get("r", 0),
+        "p": data.get("p", 0),
+        "o": data.get("o", "audio-24khz-48kbitrate-mono-mp3")
+    }
+    
+    # 检查必填参数
+    if not params["t"]:
+        raise HTTPException(status_code=400, detail="Parameter 't' (text) is required")
+    
+    try:
+        file_name = await download_audio(params, use_post=True)
+        base_url = str(request.base_url)
+        audio_url = f"{base_url}audio/{file_name}"
+        
+        # 检查是否来自缓存
+        cache_key = get_cache_key(params)
+        cached = cache_key in cache_store and datetime.now() < cache_store[cache_key]["expire_time"]
+        
+        return {"status": "success", "audio_url": audio_url, "cached": cached}
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={"status": "error", "message": str(e)}
+        )
+
+@app.get("/audio/{file_name}")
+async def get_audio(file_name: str):
+    """返回音频文件"""
+    file_path = Path(AUDIO_DIR) / file_name
+    if not file_path.exists():
+        raise HTTPException(status_code=404, detail="Audio file not found")
+    return FileResponse(file_path, media_type="audio/mpeg")
+
+@app.get("/voices")
+async def get_voices(
+    l: Optional[str] = "zh",
+    d: Optional[bool] = False
+):
+    """获取语音列��"""
+    params = {}
+    if l:
+        params["l"] = l
+    if d:
+        params["d"] = ""
+    
+    async with httpx.AsyncClient() as client:
+        try:
+            response = await client.get(f"{HOST}/voices", params=params)
+            response.raise_for_status()
+            return response.json()
+        except httpx.HTTPStatusError as e:
+            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+
+@app.get("/cache/status")
+async def cache_status():
+    """获取缓存状态"""
+    return {
+        "cache_count": len(cache_store),
+        "expire_hours": CACHE_EXPIRE_HOURS
+    }
+
+@app.delete("/cache/clear")
+async def clear_cache():
+    """清除所有缓存"""
+    global cache_store
+    cache_store = {}
+    return {"status": "success", "message": "Cache cleared"}
+
+@app.get("/")
+async def read_root(request: Request):
+    """主页，展示语音列表"""
+    return templates.TemplateResponse("index.html", {"request": request})
+
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)

requirements.txt

@@ -0,0 +1,6 @@
+fastapi
+uvicorn
+httpx
+aiofiles
+python-multipart
+jinja2