ajout d'une vérification par api

app.py

@@ -1,6 +1,6 @@
 import os
 import litellm
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException, Header, Depends
 from pydantic import BaseModel
 
 # Configuration du modèle
@@ -21,7 +21,14 @@ base_prompt = """
 Tu es un développeur logiciel expérimenté. Tu dois toujours répondre en français d'une manière réfléchie et concise. 
 """
 
-@app.post("/")
+# Clé API pour l'authentification
+USER_API_KEY = "YOUR_API_KEY"  # Remplacez par une clé secrète
+
+async def verify_api_key(x_api_key: str = Header(...)):
+    if x_api_key != API_KEY:
+        raise HTTPException(status_code=401, detail="Clé API invalide")
+
+@app.post("/", dependencies=[Depends(verify_api_key)])
 async def predict(request: PromptRequest):
     try:
         # Ajout du prompt de base à la requête de l'utilisateur