Update main.py

main.py

@@ -1,17 +1,17 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException, Header
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 from kokoro import KPipeline
 import numpy as np
 import io
 import logging
-
+import os
 # --- Logging ---
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("tts_stream")
 
 app = FastAPI()
-
+valid_hf_keys = os.getenv("hf_key")
 # Initialiser le pipeline au démarrage
 pipeline = KPipeline(lang_code='a', device='cpu')
 logger.info("✅ KPipeline loaded successfully.")
@@ -22,9 +22,10 @@ class TTSRequest(BaseModel):
     speed: float = 1.0
 
 @app.post("/tts/stream")
-async def stream_speech(request: TTSRequest):
+async def stream_speech(request: TTSRequest, hf_key: str = Header(None)):
     logger.info(f"🚀 Streaming request received: text='{request.text[:50]}...', voice='{request.voice}', speed={request.speed}")
-
+    if hf_key is None or hf_key not in VALID_HF_KEYS:
+        raise HTTPException(status_code=401, detail="Unauthorized: invalid or missing hf_key")
     def generate():
         chunk_index = 0
         try: