add: detail logging and authorization header

app.py

@@ -4,8 +4,10 @@ created @ 2024-10-28
 author  @ github.com/ishworrsubedii
 """
 import time
+from datetime import datetime, timedelta, timezone
 
-from fastapi import FastAPI
+from fastapi import FastAPI, Depends, HTTPException
+from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 from starlette.middleware.cors import CORSMiddleware
 
 from src.api.image_prep_api import preprocessing_router
@@ -13,8 +15,36 @@ from src.api.nto_api import nto_cto_router
 from src.api.image_regeneration_api import image_regeneration_router
 from src.api.batch_api import batch_router
 from src.api.mannequin_to_model_api import mto_router
+from src.api.nto_api import supabase
 
-app = FastAPI()
+security = HTTPBearer()
+
+
+async def verify_login_token(credentials: HTTPAuthorizationCredentials = Depends(security)):
+    try:
+        response = supabase.table("JewelMirrorUserLogins").select("*").eq("LoginToken",
+                                                                          credentials.credentials).execute()
+
+        if not response.data:
+            raise HTTPException(status_code=401, detail="Unauthorized: Token not found")
+
+        token_data = response.data[0]
+
+        created_at = datetime.fromisoformat(token_data["UpdatedAt"].replace("Z", "+00:00"))
+        current_time = datetime.now(timezone.utc)
+        time_difference = current_time - created_at
+
+        if time_difference <= timedelta(minutes=30):
+            return {"status": "Authorized", "message": "Token is valid"}
+
+        raise HTTPException(status_code=401, detail="Unauthorized: Token expired")
+
+    except Exception as e:
+        print(f"Token verification error: {e}")
+        raise HTTPException(status_code=401, detail="Invalid token")
+
+
+app = FastAPI(dependencies=[Depends(verify_login_token)])
 app.include_router(nto_cto_router, tags=["NTO-CTO"])
 app.include_router(preprocessing_router, tags=["Image-Preprocessing"])
 app.include_router(image_regeneration_router, tags=["Image-Regeneration"])
@@ -29,4 +59,4 @@ app.add_middleware(
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
-)
+)

src/api/batch_api.py

@@ -30,6 +30,9 @@ async def rt_cto(
         image: UploadFile = File(...),
         c_list: str = Form(...)
 ):
+    logger.info("-" * 50)
+    logger.info(">>> REAL-TIME CTO STARTED <<<")
+    logger.info(f"Parameters: clothing_list={c_list}")
     try:
         clothing_list = [item.strip() for item in c_list.split(",")]
         logger.info(f">>> CLOTHING LIST: {clothing_list} <<<")

src/api/nto_api.py

@@ -65,6 +65,7 @@ class NecklaceTryOnIDEntity(BaseModel):
 async def clothing_try_on_v2(image: UploadFile = File(...), clothing_type: str = Form(...)):
     logger.info("-" * 50)
     logger.info(">>> CLOTHING TRY ON V2 STARTED <<<")
+    logger.info(f"Parameters: clothing_type={clothing_type}")
     start_time = time.time()
 
     try:
@@ -132,6 +133,7 @@ async def clothing_try_on(image: UploadFile = File(...),
                           mask: UploadFile = File(...), clothing_type: str = Form(...)):
     logger.info("-" * 50)
     logger.info(">>> CLOTHING TRY ON STARTED <<<")
+    logger.info(f"Parameters: clothing_type={clothing_type}")
     start_time = time.time()
 
     try:
@@ -428,7 +430,10 @@ def process_image(image: Image.Image, quality: int) -> bytes:
 async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends(parse_necklace_try_on_id),
                              image: UploadFile = File(...)):
     logger.info("-" * 50)
-    logger.info(f">>> NECKLACE TRY ON ID STARTED :: {necklace_try_on_id.storename} <<<")
+    logger.info(">>> NECKLACE TRY ON ID STARTED <<<")
+    logger.info(f"Parameters: storename={necklace_try_on_id.storename}, "
+                f"necklaceCategory={necklace_try_on_id.necklaceCategory}, "
+                f"necklaceImageId={necklace_try_on_id.necklaceImageId}")
     start_time = time.time()
 
     try:
@@ -493,7 +498,7 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
             return JSONResponse(content={"error": f"Error generating response", "code": 500}, status_code=500)
 
         logger.info(f">>> TOTAL INFERENCE TIME: {total_backend_time}s <<<")
-        logger.info(f">>> NECKLACE TRY ON COMPLETED :: {necklace_try_on_id.storename} <<<")
+        logger.info(f">>> NECKLACE TRY ON COMPLETED <<<")
         logger.info("-" * 50)
 
         return JSONResponse(content=response, status_code=200)
@@ -508,7 +513,10 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
 async def canvas_points(necklace_try_on_id: NecklaceTryOnIDEntity = Depends(parse_necklace_try_on_id),
                         image: UploadFile = File(...)):
     logger.info("-" * 50)
-    logger.info(f">>> CANVAS POINTS STARTED :: {necklace_try_on_id.storename} <<<")
+    logger.info(">>> CANVAS POINTS STARTED <<<")
+    logger.info(f"Parameters: storename={necklace_try_on_id.storename}, "
+                f"necklaceCategory={necklace_try_on_id.necklaceCategory}, "
+                f"necklaceImageId={necklace_try_on_id.necklaceImageId}")
     start_time = time.time()
 
     try:
@@ -543,7 +551,7 @@ async def canvas_points(necklace_try_on_id: NecklaceTryOnIDEntity = Depends(pars
 
     total_inference_time = round((time.time() - start_time), 2)
     logger.info(f">>> TOTAL INFERENCE TIME: {total_inference_time}s <<<")
-    logger.info(f">>> CANVAS POINTS COMPLETED :: {necklace_try_on_id.storename} <<<")
+    logger.info(f">>> CANVAS POINTS COMPLETED <<<")
     logger.info("-" * 50)
 
     return JSONResponse(status_code=200, content=response)
@@ -557,7 +565,11 @@ async def necklace_try_on_with_points(necklace_try_on_id: NecklaceTryOnIDEntity
                                       right_x: int = Form(...),
                                       right_y: int = Form(...)):
     logger.info("-" * 50)
-    logger.info(f">>> NECKLACE TRY ON WITH POINTS STARTED :: {necklace_try_on_id.storename} <<<")
+    logger.info(">>> NECKLACE TRY ON WITH POINTS STARTED <<<")
+    logger.info(f"Parameters: storename={necklace_try_on_id.storename}, "
+                f"necklaceCategory={necklace_try_on_id.necklaceCategory}, "
+                f"necklaceImageId={necklace_try_on_id.necklaceImageId}, "
+                f"left_point=({left_x}, {left_y}), right_point=({right_x}, {right_y})")
     start_time = time.time()
 
     try:
@@ -613,7 +625,7 @@ async def necklace_try_on_with_points(necklace_try_on_id: NecklaceTryOnIDEntity
         return JSONResponse(content={"error": f"Error deducting credits", "code": 500}, status_code=500)
 
     logger.info(f">>> TOTAL INFERENCE TIME: {total_inference_time}s <<<")
-    logger.info(f">>> NECKLACE TRY ON WITH POINTS COMPLETED :: {necklace_try_on_id.storename} <<<")
+    logger.info(f">>> NECKLACE TRY ON WITH POINTS COMPLETED <<<")
     logger.info("-" * 50)
 
     return JSONResponse(content=response, status_code=200)
@@ -629,6 +641,10 @@ async def clothing_and_necklace_try_on(
 ):
     logger.info("-" * 50)
     logger.info(">>> CLOTHING AND NECKLACE TRY ON STARTED <<<")
+    logger.info(f"Parameters: storename={storename}, "
+                f"necklaceCategory={necklaceCategory}, "
+                f"necklaceImageId={necklaceImageId}, "
+                f"clothing_type={clothing_type}")
     start_time = time.time()
 
     def image_to_base64(img: Image.Image) -> str:
@@ -715,7 +731,10 @@ async def clothing_and_necklace_try_on(
 async def mannequin_nto(necklace_try_on_id: NecklaceTryOnIDEntity = Depends(parse_necklace_try_on_id),
                         image: UploadFile = File(...)):
     logger.info("-" * 50)
-    logger.info(f">>> NECKLACE TRY ON ID STARTED :: {necklace_try_on_id.storename} <<<")
+    logger.info(">>> MANNEQUIN NTO STARTED <<<")
+    logger.info(f"Parameters: storename={necklace_try_on_id.storename}, "
+                f"necklaceCategory={necklace_try_on_id.necklaceCategory}, "
+                f"necklaceImageId={necklace_try_on_id.necklaceImageId}")
     start_time = time.time()
 
     try: