Update main.py

main.py

@@ -6,17 +6,22 @@ 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.environ.get('hf_key')
+
+# Charger les clés valides depuis l'environnement
+VALID_HF_KEYS = os.environ.get('VALID_HF_KEYS', '').split(',')
+logger.info(f"✅ Loaded {len(VALID_HF_KEYS)} valid API key(s)")
+logger.info(f"🔑 First key preview: {VALID_HF_KEYS[0][:10] + '...' if VALID_HF_KEYS and VALID_HF_KEYS[0] else 'NONE'}")
+
 # Initialiser le pipeline au démarrage
 pipeline = KPipeline(lang_code='a', device='cpu')
 logger.info("✅ KPipeline loaded successfully.")
-logger.info(f"✅ Loaded {len(valid_hf_keys)} valid API keys")
-print(f"cle{valid_hf_keys}")
+
 class TTSRequest(BaseModel):
     text: str
     voice: str = 'af_heart'
@@ -24,21 +29,27 @@ class TTSRequest(BaseModel):
 
 @app.post("/tts/stream")
 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:
+    logger.info(f"🚀 Request: text='{request.text[:30]}...', voice='{request.voice}'")
+    logger.info(f"🔑 Received key: {hf_key[:10] + '...' if hf_key else 'NONE'}")
+    
+    # Vérifier le token
+    if not hf_key or hf_key not in VALID_HF_KEYS:
+        logger.error(f"❌ Unauthorized: key not in valid list")
         raise HTTPException(status_code=401, detail="Unauthorized: invalid or missing hf_key")
+    
+    logger.info(f"✅ Key validated, generating audio...")
+    
     def generate():
         chunk_index = 0
         try:
             for result in pipeline(request.text, voice=request.voice, speed=request.speed):
                 chunk_index += 1
-                # Convertir en PCM float32
                 audio_bytes = result.audio.numpy().astype(np.float32).tobytes()
                 logger.info(f"✅ Chunk {chunk_index} ready, size={len(audio_bytes)} bytes")
                 yield audio_bytes
-            logger.info(f"🏁 Streaming finished: total chunks={chunk_index}")
+            logger.info(f"🏁 Streaming finished: {chunk_index} chunks")
         except Exception as e:
-            logger.error(f"❌ Streaming error at chunk {chunk_index}: {e}")
+            logger.error(f"❌ Error at chunk {chunk_index}: {e}")
             raise
 
     return StreamingResponse(
@@ -49,4 +60,4 @@ async def stream_speech(request: TTSRequest, hf_key: str = Header(None)):
             "X-Channels": "1",
             "X-Bit-Depth": "32"
         }
-    )
+    )