Update app.py

app.py

@@ -344,10 +344,6 @@ async def face_swap_api(
                         },
                         upsert=True
                     )
-            
-                    # -------------------------------------------------
-                    # STEP 2: Handle DAILY USAGE (binary per day)
-                    # -------------------------------------------------
                     # -------------------------------------------------
                     # STEP 2: Handle DAILY USAGE (BINARY, NO DUPLICATES)
                     # -------------------------------------------------
@@ -361,28 +357,27 @@ async def face_swap_api(
                     # 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
-                    }
+                    # Build normalized date → count map (THIS ENFORCES UNIQUENESS)
+                    daily_map = {}
+                    for entry in daily_entries:
+                        d = entry["date"]
+                        if isinstance(d, datetime):
+                            d = datetime(d.year, d.month, d.day)
+                        daily_map[d] = entry["count"]  # overwrite = no duplicates
                     
-                    if daily_map:
-                        last_date = max(daily_map.keys())
-                    else:
-                        last_date = today_date
+                    # Determine last recorded date
+                    last_date = max(daily_map.keys()) if daily_map else today_date
                     
-                    # Fill missing dates with 0 (from last_date+1 to yesterday)
+                    # Fill ALL missing days with count = 0
                     next_day = last_date + timedelta(days=1)
                     while next_day < today_date:
-                        if next_day not in daily_map:
-                            daily_map[next_day] = 0
+                        daily_map.setdefault(next_day, 0)
                         next_day += timedelta(days=1)
                     
-                    # Mark today as used (binary = 1)
+                    # Mark today as used (binary)
                     daily_map[today_date] = 1
                     
-                    # Rebuild sorted list (oldest → newest)
+                    # Rebuild list: OLDEST → NEWEST
                     final_daily_entries = [
                         {"date": d, "count": daily_map[d]}
                         for d in sorted(daily_map.keys())
@@ -391,7 +386,7 @@ async def face_swap_api(
                     # Keep only last 32 days
                     final_daily_entries = final_daily_entries[-32:]
                     
-                    # Replace array atomically
+                    # Atomic replace
                     await media_clicks_col.update_one(
                         {"userId": user_oid},
                         {
@@ -402,77 +397,6 @@ async def face_swap_api(
                         }
                     )
 
-
-
-                    # 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 []
-                    # 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 = 1
-                    #     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:
-                    #         daily_updates.append({
-                    #             "date": today_date,
-                    #             "count": 1
-                    #         })
-            
-                    # if daily_updates:
-                    #     await media_clicks_col.update_one(
-                    #         {"userId": user_oid},
-                    #         {
-                    #             "$push": {
-                    #                 "ai_edit_daily_count": {
-                    #                     "$each": daily_updates
-                    #                 }
-                    #             }
-                    #         }
-                    #     )
-
-                    # -------------------------------------------------
-                    # STEP 2.5: Keep ONLY last 32 days of ai_edit_daily_count
-                    # -------------------------------------------------
-                    # doc = await media_clicks_col.find_one(
-                    #     {"userId": user_oid},
-                    #     {"ai_edit_daily_count": 1}
-                    # )
-                    
-                    # if daily_entries:
-                    #     # Sort the array by date ASC (oldest → newest)
-                    #     daily_entries.sort(key=lambda x: x["date"])
-                    
-                    #     # Keep only last 32 days
-                    #     if len(daily_entries) > 32:
-                    #         daily_entries = daily_entries[-32:]
-                    
-                    #     # Update the DB with sorted & trimmed array
-                    #     await media_clicks_col.update_one(
-                    #         {"userId": user_oid},
-                    #         {"$set": {"ai_edit_daily_count": daily_entries}}
-                    #     )
                     # -------------------------------------------------
                     # STEP 3: Try updating existing subCategory
                     # -------------------------------------------------