Update app.py

app.py

@@ -346,49 +346,6 @@ async def face_swap_api(
                 except Exception as media_err:
                     logger.error(f"MEDIA_CLICK ERROR: {media_err}")
 
-            # if user_id:
-            #     try:
-            #         user_oid = ObjectId(user_id.strip())
-            #         now = datetime.utcnow()
-
-            #         update_result = await media_clicks_col.update_one(
-            #             #{"userId": user_oid, "categories.categoryId": category_oid},
-            #             {"userId": user_oid, "categories.subcategoryId": subcategory_oid},
-                        
-
-            #             {
-            #                 "$set": {
-            #                     "updatedAt": now,
-            #                     "categories.$.lastClickedAt": now
-            #                 },
-            #                 "$inc": {"categories.$.click_count": 1}
-            #             }
-            #         )
-
-            #         if update_result.matched_count == 0:
-            #             # category does not exist → push new entry
-            #             await media_clicks_col.update_one(
-            #                 {"userId": user_oid},
-            #                 {
-            #                     "$setOnInsert": {"createdAt": now},
-            #                     "$set": {"updatedAt": now},
-            #                     "$push": {
-            #                         "categories": {
-            #                             #"categoryId": category_oid,
-            #                             #categoryId": subcategory_oid,
-            #                             "subCategoryId": subcategory_oid,
-            #                             "click_count": 1,
-            #                             "lastClickedAt": now
-            #                         }
-            #                     }
-            #                 },
-            #                 upsert=True
-            #             )
-
-            #     except Exception as media_err:
-            #         logger.error(f"MEDIA_CLICK ERROR: {media_err}")
-                    # do NOT interrupt faceswap flow
-
         # ------------------------------------------------------------------
         # CASE 2 : target_category_id → DigitalOcean path (unchanged logic)
         # ------------------------------------------------------------------
@@ -448,18 +405,43 @@ async def face_swap_api(
 
         result_key = f"faceswap/result/{uuid.uuid4().hex}_enhanced.png"
         result_url = upload_to_spaces(result_bytes, result_key)
+        end_time = datetime.utcnow()
+        response_time_ms = (end_time - start_time).total_seconds() * 1000
+
+        if database is not None:
+            await database.api_logs.insert_one({
+                "endpoint": "/face-swap",
+                "status": "success",
+                "response_time_ms": response_time_ms,
+                "timestamp": end_time
+            })
+
 
         return {
             "result_key": result_key,
             "result_url": result_url
         }
+        
 
     except Exception as e:
+        end_time = datetime.utcnow()
+        response_time_ms = (end_time - start_time).total_seconds() * 1000
+
+        if database is not None:
+            await database.api_logs.insert_one({
+                "endpoint": "/face-swap",
+                "status": "fail",
+                "response_time_ms": response_time_ms,
+                "timestamp": end_time,
+                "error": str(e)
+            })
+
         raise HTTPException(500, f"Face swap failed: {str(e)}")
 
 
 
 
+
 # @fastapi_app.post("/face-swap", dependencies=[Depends(verify_token)])
 # async def face_swap_api(
 #     source: UploadFile = File(...),