Upload 3 files

tts-server.py

@@ -10,8 +10,9 @@ from typing import Optional, List, Dict, Any, AsyncGenerator
 
 import aiohttp
 import websockets
-from fastapi import FastAPI, HTTPException, Header, Request, BackgroundTasks
+from fastapi import FastAPI, HTTPException, Header, Request, BackgroundTasks, Depends
 from fastapi.responses import StreamingResponse, JSONResponse
+from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 from pydantic import BaseModel, Field
 import uvicorn
 from contextlib import asynccontextmanager
@@ -32,6 +33,7 @@ PORT = 1547
 HOST = "0.0.0.0"
 MODELS_FILE = "models.json"
 COOKIE_DIR = "cookie"  # Directory to store cookie txt files
+AUTH_PASSWORD = os.getenv("PASSWORD", "sk-wei123")
 
 # Initialize FastAPI
 @asynccontextmanager
@@ -50,6 +52,12 @@ async def lifespan(app: FastAPI):
     yield
 
 app = FastAPI(title="Doubao TTS OpenAI API Server", lifespan=lifespan)
+security = HTTPBearer()
+
+async def verify_token(credentials: HTTPAuthorizationCredentials = Depends(security)):
+    if credentials.credentials != AUTH_PASSWORD:
+        raise HTTPException(status_code=401, detail="Invalid authentication token")
+    return credentials.credentials
 
 # ==========================================
 # Cookie Manager
@@ -337,7 +345,7 @@ async def check_speech_endpoint():
     return {"status": "ok", "message": "Speech endpoint is ready"}
 
 @app.post("/v1/audio/speech")
-async def create_speech(req: OpenAIRequest):
+async def create_speech(req: OpenAIRequest, token: str = Depends(verify_token)):
     """
     OpenAI-compatible speech generation endpoint.
     """