Update app.py

app.py

@@ -162,48 +162,108 @@ async def log_faceswap_hit(user_email: str, status: str, start_time: datetime, e
     })
 
 # --------------------- Media Click Logging Helper ---------------------
+async def log_media_click(user_id: str, category_oid_str: str):
+    """
+    Logs a click event to the media_clicks collection against the Category.
+    Safely updates ai_edit_last_date and ai_edit_complete at ROOT level.
+    """
+    try:
+        user_oid = ObjectId(user_id.strip())
+        category_oid = ObjectId(category_oid_str.strip())
+        now = datetime.utcnow()
+
+        # Optional safety check
+        if not await categories_col.find_one({"_id": category_oid}):
+            logger.warning(
+                "Category ID %s not found. Skipping media click logging.",
+                category_oid_str
+            )
+            return
+
+        # -------------------------------------------------
+        # STEP 1: Ensure user document + root fields exist
+        # -------------------------------------------------
+        await media_clicks_col.update_one(
+            {"userId": user_oid},
+            {
+                "$setOnInsert": {
+                    "userId": user_oid,
+                    "createdAt": now,
+                    "ai_edit_complete": 0
+                }
+            },
+            upsert=True
+        )
+
+        # -------------------------------------------------
+        # STEP 2: Try updating existing category
+        # -------------------------------------------------
+        update_result = await media_clicks_col.update_one(
+            {
+                "userId": user_oid,
+                "categories.categoryId": category_oid
+            },
+            {
+                "$inc": {
+                    "categories.$.click_count": 1,
+                    "ai_edit_complete": 1
+                },
+                "$set": {
+                    "categories.$.lastClickedAt": now,
+                    "ai_edit_last_date": now,
+                    "updatedAt": now
+                }
+            }
+        )
+
+        # -------------------------------------------------
+        # STEP 3: Category not present → push new
+        # -------------------------------------------------
+        if update_result.matched_count == 0:
+            await media_clicks_col.update_one(
+                {"userId": user_oid},
+                {
+                    "$inc": {
+                        "ai_edit_complete": 1
+                    },
+                    "$set": {
+                        "ai_edit_last_date": now,
+                        "updatedAt": now
+                    },
+                    "$push": {
+                        "categories": {
+                            "categoryId": category_oid,
+                            "click_count": 1,
+                            "lastClickedAt": now
+                        }
+                    }
+                }
+            )
+
+        logger.info(
+            "[MEDIA_CLICK] user=%s category=%s ai_edit_complete incremented",
+            user_id,
+            category_oid_str
+        )
+
+    except Exception as media_err:
+        logger.error("MEDIA_CLICK LOGGING ERROR: %s", media_err)
+
 # async def log_media_click(user_id: str, category_oid_str: str):
 #     """
-#     Logs a click event to the media_clicks collection against the Category
-#     and tracks AI edit usage.
+#     Logs a click event to the media_clicks collection against the Category.
 #     """
 #     try:
 #         user_oid = ObjectId(user_id.strip())
 #         category_oid = ObjectId(category_oid_str.strip())
 #         now = datetime.utcnow()
 
-#         # --------------------------------------------------
-#         # OPTIONAL: VALIDATE CATEGORY EXISTS
-#         # --------------------------------------------------
+#         # Check if category exists (optional, but prevents logging invalid IDs)
 #         if not await categories_col.find_one({"_id": category_oid}):
-#             logger.warning(
-#                 f"Category ID {category_oid_str} not found. Skipping click logging."
-#             )
+#             logger.warning(f"Category ID {category_oid_str} not found. Skipping click logging.")
 #             return
 
-#         # --------------------------------------------------
-#         # 1. AI EDIT USAGE TRACKING (PER USER)
-#         # --------------------------------------------------
-#         await media_clicks_col.update_one(
-#             {"userId": user_oid},
-#             {
-#                 "$set": {
-#                     "ai_edit_last_date": now,
-#                     "updatedAt": now
-#                 },
-#                 "$inc": {
-#                     "ai_edit_complete": 1
-#                 },
-#                 "$setOnInsert": {
-#                     "createdAt": now
-#                 }
-#             },
-#             upsert=True
-#         )
-
-#         # --------------------------------------------------
-#         # 2. UPDATE EXISTING CATEGORY CLICK
-#         # --------------------------------------------------
+#         # 1) Try updating existing category click entry
 #         update_result = await media_clicks_col.update_one(
 #             {
 #                 "userId": user_oid,
@@ -220,16 +280,13 @@ async def log_faceswap_hit(user_email: str, status: str, start_time: datetime, e
 #             }
 #         )
 
-#         # --------------------------------------------------
-#         # 3. PUSH CATEGORY IF NOT EXISTS
-#         # --------------------------------------------------
+#         # 2) If no category entry exists → push new one (or create doc)
 #         if update_result.matched_count == 0:
 #             await media_clicks_col.update_one(
-#                 {"userId": user_oid},
+#                 { "userId": user_oid },
 #                 {
-#                     "$set": {
-#                         "updatedAt": now
-#                     },
+#                     "$setOnInsert": { "createdAt": now },
+#                     "$set": { "updatedAt": now },
 #                     "$push": {
 #                         "categories": {
 #                             "categoryId": category_oid,
@@ -240,68 +297,12 @@ async def log_faceswap_hit(user_email: str, status: str, start_time: datetime, e
 #                 },
 #                 upsert=True
 #             )
-
-#         logger.info(
-#             f"Media click logged for User {user_id} on Category {category_oid_str}"
-#         )
+        
+#         logger.info(f"Media click logged for User {user_id} on Category {category_oid_str}")
 
 #     except Exception as media_err:
 #         logger.error(f"MEDIA_CLICK LOGGING ERROR: {media_err}")
 
-async def log_media_click(user_id: str, category_oid_str: str):
-    """
-    Logs a click event to the media_clicks collection against the Category.
-    """
-    try:
-        user_oid = ObjectId(user_id.strip())
-        category_oid = ObjectId(category_oid_str.strip())
-        now = datetime.utcnow()
-
-        # Check if category exists (optional, but prevents logging invalid IDs)
-        if not await categories_col.find_one({"_id": category_oid}):
-            logger.warning(f"Category ID {category_oid_str} not found. Skipping click logging.")
-            return
-
-        # 1) Try updating existing category click entry
-        update_result = await media_clicks_col.update_one(
-            {
-                "userId": user_oid,
-                "categories.categoryId": category_oid
-            },
-            {
-                "$set": {
-                    "updatedAt": now,
-                    "categories.$.lastClickedAt": now
-                },
-                "$inc": {
-                    "categories.$.click_count": 1
-                }
-            }
-        )
-
-        # 2) If no category entry exists → push new one (or create doc)
-        if update_result.matched_count == 0:
-            await media_clicks_col.update_one(
-                { "userId": user_oid },
-                {
-                    "$setOnInsert": { "createdAt": now },
-                    "$set": { "updatedAt": now },
-                    "$push": {
-                        "categories": {
-                            "categoryId": category_oid,
-                            "click_count": 1,
-                            "lastClickedAt": now
-                        }
-                    }
-                },
-                upsert=True
-            )
-        
-        logger.info(f"Media click logged for User {user_id} on Category {category_oid_str}")
-
-    except Exception as media_err:
-        logger.error(f"MEDIA_CLICK LOGGING ERROR: {media_err}")
-
 # --------------------- Face Swap Pipeline ---------------------
 swap_lock = threading.Lock()
 def face_swap_and_enhance(src_img, tgt_img, temp_dir="/tmp/faceswap_work"):