Update app.py

app.py

@@ -224,28 +224,45 @@ async def generate(
             # --------------------------------------------------
             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:
-                # 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:
+                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 all missing days between last_date_in_db 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}}}
                 )
-    
+            
+            # 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}}
+                )
+
             # --------------------------------------------------
             # CATEGORY CLICK LOGIC
             # --------------------------------------------------