Update app/database.py

app/database.py

@@ -246,28 +246,29 @@ def _update_daily_count(collection, user_object_id: ObjectId, today: datetime) -
         "count": 1
     })
     
-    # Push all missing dates (including today)
+    # Merge existing entries with the new ones, sort by date (oldest first),
+    # and keep only the most recent 32 dates (drop the oldest beyond 32).
     if dates_to_add:
-        # Ensure ai_edit_daily_count field exists
-        if "ai_edit_daily_count" not in user_doc:
-            collection.update_one(
-                {"userId": user_object_id},
-                {"$set": {"ai_edit_daily_count": []}}
-            )
-        
+        all_entries = list(existing_counts) + dates_to_add
+
+        def _entry_sort_key(entry: Dict[str, Any]) -> datetime:
+            dt = entry.get("date")
+            if isinstance(dt, datetime):
+                return dt.replace(hour=0, minute=0, second=0, microsecond=0)
+            return datetime.min
+
+        all_entries.sort(key=_entry_sort_key)
+        if len(all_entries) > 32:
+            all_entries = all_entries[-32:]
+
         collection.update_one(
             {"userId": user_object_id},
-            {
-                # Append new dates and keep only the most recent 32 entries
-                "$push": {
-                    "ai_edit_daily_count": {
-                        "$each": dates_to_add,
-                        "$slice": -32,
-                    }
-                }
-            }
+            {"$set": {"ai_edit_daily_count": all_entries}},
+        )
+        logger.debug(
+            "Updated ai_edit_daily_count with %d entries (oldest first, max 32)",
+            len(all_entries),
         )
-        logger.debug("Added %d daily count entries (including today with count 1)", len(dates_to_add)) 
 
 def log_api_call(
     endpoint: str,
@@ -567,6 +568,7 @@ def close_connection():
 
 
 
+
 # """ 
 # MongoDB database connection and logging utilities, including admin media click logging. 
 # """ 
@@ -827,9 +829,11 @@ def close_connection():
 #         collection.update_one(
 #             {"userId": user_object_id},
 #             {
+#                 # Append new dates and keep only the most recent 32 entries
 #                 "$push": {
 #                     "ai_edit_daily_count": {
-#                         "$each": dates_to_add
+#                         "$each": dates_to_add,
+#                         "$slice": -32,
 #                     }
 #                 }
 #             }
@@ -1130,5 +1134,4 @@ def close_connection():
 #         _admin_client.close() 
 #         _admin_client = None 
 #         _admin_db = None 
-#         logger.info("Admin MongoDB connection closed") 
-
+#         logger.info("Admin MongoDB connection closed")