api db changed

app.py

@@ -127,17 +127,17 @@ def verify_token(credentials: HTTPAuthorizationCredentials = Security(security))
         raise HTTPException(status_code=401, detail="Invalid or missing token")
     return credentials.credentials
 
-# --------------------- Logging Helper ---------------------
+# --------------------- Logging API Hits ---------------------
 async def log_faceswap_hit(token: str, status: str = "success"):
-    if not database:
+    global database
+    if database is None:  # ✅ FIXED: explicit None check
         return
-    log_entry = {
-        "endpoint": "/faceswap",
+    await database.api_logs.insert_one({
         "token": token,
+        "endpoint": "/faceswap",
         "status": status,
         "timestamp": datetime.utcnow()
-    }
-    await database.api_logs.insert_one(log_entry)
+    })
 
 # --------------------- Face Swap Pipeline ---------------------
 swap_lock = threading.Lock()
@@ -237,18 +237,15 @@ class FaceSwapRequest(BaseModel):
     source_id: str
     target_id: str
 
-@fastapi_app.post("/faceswap")
-async def perform_faceswap(
-    request: FaceSwapRequest,
-    credentials: HTTPAuthorizationCredentials = Security(security)
-):
+@fastapi_app.post("/faceswap", dependencies=[Depends(verify_token)])
+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")
+        await log_faceswap_hit(credentials.credentials, status="error")
         raise HTTPException(status_code=404, detail="Source or Target not found")
 
     src_array = np.frombuffer(src_bytes, np.uint8)
@@ -257,7 +254,7 @@ async def perform_faceswap(
     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")
+        await log_faceswap_hit(credentials.credentials, status="error")
         raise HTTPException(status_code=400, detail="Invalid image data")
 
     src_rgb = cv2.cvtColor(src_bgr, cv2.COLOR_BGR2RGB)
@@ -265,7 +262,7 @@ async def perform_faceswap(
 
     final_img, final_path, err = face_swap_and_enhance(src_rgb, tgt_rgb)
     if err:
-        await log_faceswap_hit(credentials.credentials, status="failed")
+        await log_faceswap_hit(credentials.credentials, status="error")
         raise HTTPException(status_code=500, detail=err)
 
     with open(final_path, "rb") as f:
@@ -273,7 +270,7 @@ async def perform_faceswap(
 
     result_id = await fs_bucket.upload_from_stream("enhanced.png", final_bytes, metadata={"type": "result"})
 
-    # ✅ Log successful request
+    # ✅ Log success
     await log_faceswap_hit(credentials.credentials, status="success")
 
     return {"result_id": str(result_id)}
@@ -292,13 +289,6 @@ 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")