/api usage added

app.py

@@ -7,6 +7,7 @@ import numpy as np
 import threading
 import subprocess
 import logging
+from datetime import datetime
 
 import insightface
 from insightface.app import FaceAnalysis
@@ -126,6 +127,18 @@ def verify_token(credentials: HTTPAuthorizationCredentials = Security(security))
         raise HTTPException(status_code=401, detail="Invalid or missing token")
     return credentials.credentials
 
+# --------------------- Logging Helper ---------------------
+async def log_faceswap_hit(token: str, status: str = "success"):
+    if not database:
+        return
+    log_entry = {
+        "endpoint": "/faceswap",
+        "token": token,
+        "status": status,
+        "timestamp": datetime.utcnow()
+    }
+    await database.api_logs.insert_one(log_entry)
+
 # --------------------- Face Swap Pipeline ---------------------
 swap_lock = threading.Lock()
 
@@ -224,14 +237,18 @@ class FaceSwapRequest(BaseModel):
     source_id: str
     target_id: str
 
-@fastapi_app.post("/faceswap", dependencies=[Depends(verify_token)])
-async def perform_faceswap(request: FaceSwapRequest):
+@fastapi_app.post("/faceswap")
+async def perform_faceswap(
+    request: FaceSwapRequest,
+    credentials: HTTPAuthorizationCredentials = Security(security)
+):
     try:
         src_stream = await fs_bucket.open_download_stream(ObjectId(request.source_id))
         src_bytes = await src_stream.read()
         tgt_stream = await fs_bucket.open_download_stream(ObjectId(request.target_id))
         tgt_bytes = await tgt_stream.read()
     except Exception:
+        await log_faceswap_hit(credentials.credentials, status="not_found")
         raise HTTPException(status_code=404, detail="Source or Target not found")
 
     src_array = np.frombuffer(src_bytes, np.uint8)
@@ -240,6 +257,7 @@ async def perform_faceswap(request: FaceSwapRequest):
     tgt_bgr = cv2.imdecode(tgt_array, cv2.IMREAD_COLOR)
 
     if src_bgr is None or tgt_bgr is None:
+        await log_faceswap_hit(credentials.credentials, status="bad_request")
         raise HTTPException(status_code=400, detail="Invalid image data")
 
     src_rgb = cv2.cvtColor(src_bgr, cv2.COLOR_BGR2RGB)
@@ -247,12 +265,17 @@ async def perform_faceswap(request: FaceSwapRequest):
 
     final_img, final_path, err = face_swap_and_enhance(src_rgb, tgt_rgb)
     if err:
+        await log_faceswap_hit(credentials.credentials, status="failed")
         raise HTTPException(status_code=500, detail=err)
 
     with open(final_path, "rb") as f:
         final_bytes = f.read()
 
     result_id = await fs_bucket.upload_from_stream("enhanced.png", final_bytes, metadata={"type": "result"})
+
+    # ✅ Log successful request
+    await log_faceswap_hit(credentials.credentials, status="success")
+
     return {"result_id": str(result_id)}
 
 @fastapi_app.get("/download/{result_id}", dependencies=[Depends(verify_token)])
@@ -269,6 +292,13 @@ async def download_result(result_id: str):
         headers={"Content-Disposition": "attachment; filename=result.png"}
     )
 
+# --------------------- Stats Endpoint ---------------------
+@fastapi_app.get("/stats", dependencies=[Depends(verify_token)])
+async def get_stats(credentials: HTTPAuthorizationCredentials = Security(security)):
+    token = credentials.credentials
+    count = await database.api_logs.count_documents({"endpoint": "/faceswap", "token": token})
+    return {"token": token, "faceswap_calls": count}
+
 # --------------------- Mount Gradio ---------------------
 fastapi_app = mount_gradio_app(fastapi_app, demo, path="/gradio")