routers/openface_router.py

from fastapi import APIRouter, Depends, File, UploadFile, status

from auth import validate_token
from schema import EmbeddingResponse, SimilarityResponse

router = APIRouter(
    prefix="/openface",
    tags=["openface"],
    dependencies=[Depends(validate_token)],
)


@router.get(
    "/embed",
    status_code=status.HTTP_200_OK,
    response_model=EmbeddingResponse,
)
async def calculate_embedding(
    image: UploadFile = File(...),
):
    return {"message": "Hello World"}


@router.get(
    "/similarity",
    status_code=status.HTTP_200_OK,
    response_model=SimilarityResponse,
)
async def calculate_similarity_score(
    image1: UploadFile = File(...),
    image2: UploadFile = File(...),
):
    return {"message": "Hello World"}