Update app.py

app.py

@@ -349,56 +349,61 @@ async def face_swap_api(
                     # STEP 2: Handle DAILY USAGE (binary per day)
                     # -------------------------------------------------
                     # -------------------------------------------------
-                    # STEP 2: Handle DAILY USAGE (binary per day)
+                    # STEP 2: Handle DAILY USAGE (BINARY, NO DUPLICATES)
                     # -------------------------------------------------
                     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_updates = []
+                    # Normalize today to UTC midnight
+                    today_date = datetime(now.year, now.month, now.day)
+                    
+                    # Convert existing entries to map for fast lookup
+                    daily_map = {
+                        entry["date"]: entry["count"]
+                        for entry in daily_entries
+                    }
                     
-                    if not daily_entries:
-                        # FIRST EVER USAGE → only today = 1 (no previous 0s)
-                        daily_updates.append({"date": today_date, "count": 1})
+                    if daily_map:
+                        last_date = max(daily_map.keys())
                     else:
-                        # Build a set of existing dates to avoid duplicates
-                        existing_dates = {entry["date"] for entry in daily_entries}
+                        last_date = today_date
                     
-                        # Find the last recorded date
-                        last_date_in_db = max(entry["date"] for entry in daily_entries)
+                    # Fill missing dates with 0 (from last_date+1 to yesterday)
+                    next_day = last_date + timedelta(days=1)
+                    while next_day < today_date:
+                        if next_day not in daily_map:
+                            daily_map[next_day] = 0
+                        next_day += timedelta(days=1)
                     
-                        # Fill missing days between last date and yesterday with count=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)
+                    # Mark today as used (binary = 1)
+                    daily_map[today_date] = 1
                     
-                        # Add today if not already present
-                        if today_date not in existing_dates:
-                            daily_updates.append({"date": today_date, "count": 1})
+                    # Rebuild sorted list (oldest → newest)
+                    final_daily_entries = [
+                        {"date": d, "count": daily_map[d]}
+                        for d in sorted(daily_map.keys())
+                    ]
                     
-                    # Push updates if any
-                    if daily_updates:
-                        await media_clicks_col.update_one(
-                            {"userId": user_oid},
-                            {"$push": {"ai_edit_daily_count": {"$each": daily_updates}}}
-                        )
+                    # Keep only last 32 days
+                    final_daily_entries = final_daily_entries[-32:]
                     
-                    # Sort and trim to last 32 days
-                    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:]
+                    # Replace array atomically
                     await media_clicks_col.update_one(
                         {"userId": user_oid},
-                        {"$set": {"ai_edit_daily_count": daily_entries}}
+                        {
+                            "$set": {
+                                "ai_edit_daily_count": final_daily_entries,
+                                "updatedAt": now
+                            }
+                        }
                     )
 
 
+
                     # doc = await media_clicks_col.find_one(
                     #     {"userId": user_oid},
                     #     {"ai_edit_daily_count": 1}