Refactor token auth

app.py

@@ -11,6 +11,7 @@ from fastapi import (
     UploadFile,
     status,
 )
+from fastapi.security import APIKeyQuery
 
 from model import Model
 
@@ -18,6 +19,8 @@ app = FastAPI(
     title="Facial Expression Embedding Service",
 )
 
+api_key = APIKeyQuery(name="token", auto_error=False)
+
 model = Model(
     os.getenv("MODEL_REPO_ID", ""),
     os.getenv("MODEL_FILENAME", ""),
@@ -26,7 +29,7 @@ model = Model(
 
 
 async def validate_token(
-    token: Union[str, None] = Query(default=None),
+    token: Union[str, None] = Depends(api_key),
 ):
     if token is None:
         raise HTTPException(status.HTTP_401_UNAUTHORIZED, "No token provided")
@@ -35,9 +38,13 @@ async def validate_token(
     return token
 
 
-@app.post("/embed", status_code=status.HTTP_200_OK)
+@app.post(
+    "/embed",
+    status_code=status.HTTP_200_OK,
+    dependencies=[Depends(validate_token)],
+)
 async def calculate_embedding(
-    image: UploadFile = File(...), _: str = Depends(validate_token)
+    image: UploadFile = File(...),
 ):
     try:
         image_content = await image.read()
@@ -49,11 +56,14 @@ async def calculate_embedding(
         )
 
 
-@app.post("/similarity", status_code=status.HTTP_200_OK)
+@app.post(
+    "/similarity",
+    status_code=status.HTTP_200_OK,
+    dependencies=[Depends(validate_token)],
+)
 async def calculate_similarity_score(
     image1: UploadFile = File(...),
     image2: UploadFile = File(...),
-    _: str = Depends(validate_token),
 ):
     try:
         image1_content = await image1.read()