Update app.py

app.py

@@ -971,59 +971,146 @@ def _verify_bearer_token(credentials: Optional[HTTPAuthorizationCredentials] = D
         raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid token")
     return True
 def sync_log_media_click(user_id_str: str, category_id_str: str):
-        """
-        Synchronously logs a click event to the media_clicks collection.
-        This function should be run using asyncio.to_thread().
-        """
-        if _media_clicks_col is None:
-            return
-    
-        try:
-            user_oid = ObjectId(user_id_str.strip())
-            category_oid = ObjectId(category_id_str.strip())
-            now = datetime.utcnow()
-            logger.info(f"Attempting background write for User:{user_id_str}, Category:{category_id_str}") # <--- NEW LOG
-    
-            # 1. Try updating an existing category entry within the user's document
-            update_result = _media_clicks_col.update_one(
-                {
-                    "userId": user_oid,
-                    "categories.categoryId": category_oid
+    """
+    Synchronously logs a click event to the media_clicks collection.
+    This function should be run using asyncio.to_thread().
+    """
+
+    if _media_clicks_col is None:
+        return
+
+    try:
+        user_oid = ObjectId(user_id_str.strip())
+        category_oid = ObjectId(category_id_str.strip())
+        now = datetime.utcnow()
+
+        logger.info(
+            f"Attempting background write for User:{user_id_str}, Category:{category_id_str}"
+        )
+
+        # --------------------------------------------------
+        # 1. AI EDIT USAGE TRACKING (GLOBAL PER USER)
+        # --------------------------------------------------
+        _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. CATEGORY CLICK LOGIC
+        # --------------------------------------------------
+        update_result = _media_clicks_col.update_one(
+            {
+                "userId": user_oid,
+                "categories.categoryId": category_oid
+            },
+            {
+                "$set": {
+                    "updatedAt": now,
+                    "categories.$.lastClickedAt": now
                 },
+                "$inc": {
+                    "categories.$.click_count": 1
+                }
+            }
+        )
+
+        # --------------------------------------------------
+        # 3. IF CATEGORY DOES NOT EXIST → PUSH NEW
+        # --------------------------------------------------
+        if update_result.matched_count == 0:
+            _media_clicks_col.update_one(
+                {"userId": user_oid},
                 {
                     "$set": {
-                        "updatedAt": now,
-                        "categories.$.lastClickedAt": now
+                        "updatedAt": now
                     },
-                    "$inc": {
-                        "categories.$.click_count": 1
+                    "$push": {
+                        "categories": {
+                            "categoryId": category_oid,
+                            "click_count": 1,
+                            "lastClickedAt": now
+                        }
                     }
-                }
+                },
+                upsert=True
             )
+
+        logger.info(
+            f"Media click logged for User {user_id_str} on Category {category_id_str}"
+        )
+
+    except Exception as media_err:
+        logger.error(f"MEDIA_CLICK LOGGING ERROR: {media_err}")
+        # swallow error – do not affect main flow
+        pass
+
+# def sync_log_media_click(user_id_str: str, category_id_str: str):
+#         """
+#         Synchronously logs a click event to the media_clicks collection.
+#         This function should be run using asyncio.to_thread().
+#         """
+#         if _media_clicks_col is None:
+#             return
     
-            # 2. If no category entry exists (matched_count == 0) -> push a new one (or create user doc)
-            if update_result.matched_count == 0:
-                _media_clicks_col.update_one(
-                    { "userId": user_oid },
-                    {
-                        "$setOnInsert": { "createdAt": now,"ai_edit_complete": 0 },
-                        "$set": { "updatedAt": now},
-                        "$push": {
-                            "categories": {
-                                "categoryId": category_oid,
-                                "click_count": 1,
-                                "lastClickedAt": now
-                            }
-                        }
-                    },
-                    upsert=True
-                )
+#         try:
+#             user_oid = ObjectId(user_id_str.strip())
+#             category_oid = ObjectId(category_id_str.strip())
+#             now = datetime.utcnow()
+#             logger.info(f"Attempting background write for User:{user_id_str}, Category:{category_id_str}") # <--- NEW LOG
+    
+#             # 1. Try updating an existing category entry within the user's document
+#             update_result = _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 (matched_count == 0) -> push a new one (or create user doc)
+#             if update_result.matched_count == 0:
+#                 _media_clicks_col.update_one(
+#                     { "userId": user_oid },
+#                     {
+#                         "$setOnInsert": { "createdAt": now,"ai_edit_complete": 0 },
+#                         "$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_str} on Category {category_id_str}")
+#             logger.info(f"Media click logged for User {user_id_str} on Category {category_id_str}")
             
-        except Exception as media_err:
-            logger.error(f"MEDIA_CLICK LOGGING ERROR: {media_err}")
-            pass # Handle logging error gracefully
+#         except Exception as media_err:
+#             logger.error(f"MEDIA_CLICK LOGGING ERROR: {media_err}")
+#             pass # Handle logging error gracefully
 
 DEFAULT_FACE_MODEL = 'GFPGANv1.4.pth'
 DEFAULT_UPSCALE_MODEL = 'SRVGG, realesr-general-x4v3.pth'