Upload 2 files

requirements.txt

@@ -3,4 +3,5 @@ uvicorn
 aiohttp
 websockets==10.4
 pydantic
-aiofiles
+aiofiles
+python-dotenv

tts-server.py

@@ -17,6 +17,10 @@ from fastapi.responses import StreamingResponse, JSONResponse
 from pydantic import BaseModel, Field
 import uvicorn
 from contextlib import asynccontextmanager
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
 
 # ==========================================
 # Configuration & Setup
@@ -35,6 +39,7 @@ HOST = "0.0.0.0"
 MODELS_FILE = "models.json"
 COOKIE_DIR = "cookie"  # Directory to store cookie txt files
 AUDIO_DIR = "saved_audio"
+AUTH_PASSWORD = os.getenv("PASSWORD", "sk-wei123")
 
 # Initialize FastAPI
 @asynccontextmanager
@@ -408,11 +413,16 @@ async def stream_speech_get(
     voice: str,
     speed: float = 1.0,
     pitch: float = 0.0,
-    response_format: str = "aac"
+    response_format: str = "aac",
+    token: Optional[str] = None
 ):
     """
     GET endpoint for direct streaming (e.g. for <audio src="...">).
     """
+    # Verify token if password is set
+    if AUTH_PASSWORD and token != AUTH_PASSWORD:
+        raise HTTPException(status_code=401, detail="Invalid authentication token")
+
     if not input:
         raise HTTPException(status_code=400, detail="Input text is required")