add security

main.py

@@ -17,6 +17,18 @@ app = FastAPI()
 image_size = 256
 hf_token = os.environ.get("api_read")
 
+VALID_API_KEYS = os.environ.get("api_key")
+
+
+@app.middleware("http")
+async def verify_api_key(request, call_next):
+    api_key = request.headers.get("x-api-key")
+    if api_key not in VALID_API_KEYS:
+        raise HTTPException(status_code=403, detail="Unauthorized")
+    response = await call_next(request)
+    return response
+
+
 print(hf_token)
 
 models_locations = [