Update app.py

app.py

@@ -970,6 +970,7 @@ def _verify_bearer_token(credentials: Optional[HTTPAuthorizationCredentials] = D
     if not _api_bearer_token or token != _api_bearer_token:
         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.
@@ -984,6 +985,10 @@ def sync_log_media_click(user_id_str: str, category_id_str: str):
         category_oid = ObjectId(category_id_str.strip())
         now = datetime.utcnow()
 
+        # Normalize dates (UTC midnight)
+        today_date = datetime(now.year, now.month, now.day)
+        yesterday_date = today_date - timedelta(days=1)
+
         logger.info(
             f"Attempting background write for User:{user_id_str}, Category:{category_id_str}"
         )
@@ -994,20 +999,49 @@ def sync_log_media_click(user_id_str: str, category_id_str: str):
         _media_clicks_col.update_one(
             {"userId": user_oid},
             {
+                "$setOnInsert": {
+                    "createdAt": now,
+                    "ai_edit_daily_count": []
+                },
                 "$set": {
                     "ai_edit_last_date": now,
                     "updatedAt": now
                 },
                 "$inc": {
                     "ai_edit_complete": 1
-                },
-                "$setOnInsert": {
-                    "createdAt": now
                 }
             },
             upsert=True
         )
 
+        # Fetch current daily count entries
+        doc = _media_clicks_col.find_one(
+            {"userId": user_oid},
+            {"ai_edit_daily_count": 1}
+        )
+        daily_entries = doc.get("ai_edit_daily_count", []) if doc else []
+
+        is_first_time = len(daily_entries) == 0
+        existing_dates = {entry["date"].date(): entry["count"] for entry in daily_entries}
+        daily_updates = []
+
+        if is_first_time:
+            # First ever usage → only today
+            daily_updates.append({"date": today_date, "count": 1})
+        else:
+            # Insert yesterday if missing
+            if yesterday_date.date() not in existing_dates:
+                daily_updates.append({"date": yesterday_date, "count": 0})
+            # Insert today if missing
+            if today_date.date() not in existing_dates:
+                daily_updates.append({"date": today_date, "count": 1})
+
+        if daily_updates:
+            _media_clicks_col.update_one(
+                {"userId": user_oid},
+                {"$push": {"ai_edit_daily_count": {"$each": daily_updates}}}
+            )
+
         # --------------------------------------------------
         # 2. CATEGORY CLICK LOGIC
         # --------------------------------------------------
@@ -1034,9 +1068,7 @@ def sync_log_media_click(user_id_str: str, category_id_str: str):
             _media_clicks_col.update_one(
                 {"userId": user_oid},
                 {
-                    "$set": {
-                        "updatedAt": now
-                    },
+                    "$set": {"updatedAt": now},
                     "$push": {
                         "categories": {
                             "categoryId": category_oid,
@@ -1054,63 +1086,95 @@ def sync_log_media_click(user_id_str: str, category_id_str: 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
-    
-#         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
 #             )
-    
-#             # 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}")
-            
-#         except Exception as media_err:
-#             logger.error(f"MEDIA_CLICK LOGGING ERROR: {media_err}")
-#             pass # Handle logging error gracefully
+
+#         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
+
 
 DEFAULT_FACE_MODEL = 'GFPGANv1.4.pth'
 DEFAULT_UPSCALE_MODEL = 'SRVGG, realesr-general-x4v3.pth'