Update app.py

app.py

@@ -345,6 +345,9 @@ async def face_swap_api(
                         upsert=True
                     )
             
+                    # -------------------------------------------------
+                    # STEP 2: Handle DAILY USAGE (binary per day)
+                    # -------------------------------------------------
                     # -------------------------------------------------
                     # STEP 2: Handle DAILY USAGE (binary per day)
                     # -------------------------------------------------
@@ -352,71 +355,112 @@ async def face_swap_api(
                         {"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
-                        })
+                    
+                    if not daily_entries:
+                        # FIRST EVER USAGE → 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
-                            })
-            
+                        # Find the latest date in the array
+                        last_date_in_db = max(entry["date"] for entry in daily_entries)
+                        next_day = last_date_in_db + timedelta(days=1)
+                    
+                        # Fill missing days with count = 0 up to yesterday
+                        while next_day < today_date:
+                            daily_updates.append({"date": next_day, "count": 0})
+                            next_day += timedelta(days=1)
+                    
+                        # Today = 1
+                        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
-                                    }
-                                }
-                            }
+                            {"$push": {"ai_edit_daily_count": {"$each": daily_updates}}}
                         )
+                    
+                    # After this, optionally sort & 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:]
+                    await media_clicks_col.update_one(
+                        {"userId": user_oid},
+                        {"$set": {"ai_edit_daily_count": daily_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 []
+                    # 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}
-                    )
+                    # 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"])
+                    # 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:]
+                    #     # 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}}
-                        )
+                    #     # 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
                     # -------------------------------------------------