Update app.py

app.py

@@ -1014,32 +1014,53 @@ def sync_log_media_click(user_id_str: str, category_id_str: str):
             upsert=True
         )
 
+        # Fetch current daily count entries
         # 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:
+        
+        if not daily_entries:
             # 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:
+            # Build existing dates map for quick lookup
+            existing_dates = {entry["date"].date(): entry["count"] for entry in daily_entries}
+        
+            # Find last recorded date
+            last_date_in_db = max(entry["date"].date() for entry in daily_entries)
+        
+            # Fill missing days between last date and today-1 with 0
+            next_day = last_date_in_db + timedelta(days=1)
+            while next_day < today_date:
+                if next_day not in existing_dates:
+                    daily_updates.append({"date": next_day, "count": 0})
+                next_day += timedelta(days=1)
+        
+            # Add today if not already present
+            if today_date not in existing_dates:
                 daily_updates.append({"date": today_date, "count": 1})
-
+        
+        # Push updates if any
         if daily_updates:
             _media_clicks_col.update_one(
                 {"userId": user_oid},
-                {"$push": {"ai_edit_daily_count": {"$each": daily_updates, "$sort": {"date": 1}}}}
+                {"$push": {"ai_edit_daily_count": {"$each": daily_updates}}}
+            )
+        
+        # Fetch again, sort oldest → newest, and trim to last 32 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 []
+        daily_entries.sort(key=lambda x: x["date"])
+        if len(daily_entries) > 32:
+            daily_entries = daily_entries[-32:]
+            _media_clicks_col.update_one(
+                {"userId": user_oid},
+                {"$set": {"ai_edit_daily_count": daily_entries}}
             )
 
         # --------------------------------------------------