Update app.py

app.py

@@ -165,17 +165,23 @@ async def shutdown_db():
         logger.info("MongoDB connection closed")
 
 # --------------------- Logging API Hits ---------------------
-async def log_faceswap_hit(user_email: str, status: str = "success"):
+async def log_faceswap_hit(user_email: str, status: str, start_time: datetime, end_time: datetime):
     global database
     if database is None:
         return
+
+    response_time_ms = (end_time - start_time).total_seconds() * 1000
+
     await database.api_logs.insert_one({
         "user": user_email,
         "endpoint": "/face-swap",
         "status": status,
-        "timestamp": datetime.utcnow()
+        "start_time": start_time,
+        "end_time": end_time,
+        "response_time_ms": response_time_ms
     })
 
+
 # --------------------- Face Swap Pipeline ---------------------
 swap_lock = threading.Lock()
 
@@ -280,6 +286,7 @@ async def face_swap_api(
     target_category_id: str = Form(...),
     user_email: str = Depends(verify_firebase_token)
 ):
+    start_time = datetime.utcnow()
     try:
         src_bytes = await source.read()
         src_key = f"bikini-theme/source/{uuid.uuid4().hex}_{source.filename}"
@@ -316,14 +323,16 @@ async def face_swap_api(
             result_bytes = f.read()
         result_key = f"bikini-theme/result/{uuid.uuid4().hex}_enhanced.png"
         result_url = upload_to_spaces(result_bytes, result_key, content_type="image/png")
-
-        await log_faceswap_hit(user_email, status="success")
+        end_time = datetime.utcnow()
+        await log_faceswap_hit(user_email, status="success", start_time=start_time, end_time=end_time)
 
         return {"result_url": result_url}
 
-    except Exception as e:
-        await log_faceswap_hit(user_email, status="error")
-        raise HTTPException(status_code=500, detail=f"Face swap failed: {str(e)}")
+        except Exception as e:
+            end_time = datetime.utcnow()
+            await log_faceswap_hit(user_email, status="error", start_time=start_time, end_time=end_time)
+            raise HTTPException(status_code=500, detail=f"Face swap failed: {str(e)}")
+
 
 @fastapi_app.get("/preview/{result_key:path}")
 async def preview_result(result_key: str):