Update app.py

app.py

@@ -207,29 +207,46 @@ async def log_media_click(user_id: str, category_oid_str: str):
             {"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
+        
+        if not daily_entries:
+            # First-ever usage → only today
             daily_updates.append({"date": today_date, "count": 1})
         else:
-            # Backfill 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 date map for lookup
+            existing_dates = {entry["date"].date(): entry["count"] for entry in daily_entries}
+            last_date_in_db = max(entry["date"].date() for entry in daily_entries)
+        
+            # Fill missing days between last recorded day and today-1
+            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:
             await 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}}}
             )
+        
+        # Sort oldest → newest and trim to last 32 entries
+        doc = await 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:]
+            await media_clicks_col.update_one(
+                {"userId": user_oid},
+                {"$set": {"ai_edit_daily_count": daily_entries}}
+            )
+
 
         # -------------------------------------------------
         # STEP 2: Try updating existing category