Update app.py

app.py

@@ -317,81 +317,9 @@ async def face_swap_api(
             #category_oid = doc["categoryId"]   # <-- DO NOT CONVERT TO STRING
             subcategory_oid = doc["_id"]
     
-            # ------------------------------------------------------------------
-            # # MEDIA_CLICKS (ONLY IF user_id PRESENT
-            # if user_id:
-            #     try:
-            #         user_oid = ObjectId(user_id.strip())
-            #         now = datetime.utcnow()
-            
-            #         # -------------------------------------------------
-            #         # STEP 1: Ensure root fields exist (NO inc here)
-            #         # -------------------------------------------------
-            #         await media_clicks_col.update_one(
-            #             {"userId": user_oid},
-            #             {
-            #                 "$setOnInsert": {
-            #                     "userId": user_oid,
-            #                     "createdAt": now,
-            #                     "ai_edit_complete": 0
-            #                 }
-            #             },
-            #             upsert=True
-            #         )
-            
-            #         # -------------------------------------------------
-            #         # STEP 2: Try updating existing subCategory
-            #         # -------------------------------------------------
-            #         update_result = await media_clicks_col.update_one(
-            #             {
-            #                 "userId": user_oid,
-            #                 "subCategories.subCategoryId": subcategory_oid
-            #             },
-            #             {
-            #                 "$inc": {
-            #                     "subCategories.$.click_count": 1,
-            #                     "ai_edit_complete": 1
-            #                 },
-            #                 "$set": {
-            #                     "subCategories.$.lastClickedAt": now,
-            #                     "ai_edit_last_date": now,
-            #                     "updatedAt": now
-            #                 }
-            #             }
-            #         )
-            
-            #         # -------------------------------------------------
-            #         # STEP 3: Push subCategory if missing
-            #         # -------------------------------------------------
-            #         if update_result.matched_count == 0:
-            #             await media_clicks_col.update_one(
-            #                 {"userId": user_oid},
-            #                 {
-            #                     "$inc": {
-            #                         "ai_edit_complete": 1
-            #                     },
-            #                     "$set": {
-            #                         "ai_edit_last_date": now,
-            #                         "updatedAt": now
-            #                     },
-            #                     "$push": {
-            #                         "subCategories": {
-            #                             "subCategoryId": subcategory_oid,
-            #                             "click_count": 1,
-            #                             "lastClickedAt": now
-            #                         }
-            #                     }
-            #                 }
-            #             )
-            
-            #         logger.info(
-            #             "[MEDIA_CLICK] user=%s subCategory=%s ai_edit_complete++",
-            #             user_id,
-            #             str(subcategory_oid)
-            #         )
-            
-            #     except Exception as media_err:
-            #         logger.error(f"MEDIA_CLICK ERROR: {media_err}")
+            # ------------------------------------------------------------------#
+            # # MEDIA_CLICKS (ONLY IF user_id PRESENT)
+            # ------------------------------------------------------------------#
             if user_id:
                 try:
                     user_oid = ObjectId(user_id.strip())
@@ -467,7 +395,32 @@ async def face_swap_api(
                                 }
                             }
                         )
-            
+
+                    # -------------------------------------------------
+                    # 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}
+                    )
+                    
+                    daily_entries = doc.get("ai_edit_daily_count", [])
+                    
+                    if len(daily_entries) > 32:
+                        # Sort by date ASC (oldest first)
+                        daily_entries.sort(key=lambda x: x["date"])
+                    
+                        # Keep last 32
+                        trimmed_entries = daily_entries[-32:]
+                    
+                        await media_clicks_col.update_one(
+                            {"userId": user_oid},
+                            {
+                                "$set": {
+                                    "ai_edit_daily_count": trimmed_entries
+                                }
+                            }
+                        )
                     # -------------------------------------------------
                     # STEP 3: Try updating existing subCategory
                     # -------------------------------------------------