Update app.py

app.py

@@ -326,7 +326,24 @@ async def face_swap_api(
                     user_oid = ObjectId(user_id.strip())
                     now = datetime.utcnow()
             
-                    # 1) Try updating existing subCategory
+                    # -------------------------------------------------
+                    # 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,
@@ -341,23 +358,17 @@ async def face_swap_api(
                                 "subCategories.$.lastClickedAt": now,
                                 "ai_edit_last_date": now,
                                 "updatedAt": now
-                            },
-                            "$setOnInsert": {
-                                "createdAt": now
                             }
                         }
                     )
             
-                    # 2) If no subCategory exists → push new one (or create doc)
+                    # -------------------------------------------------
+                    # STEP 3: Push subCategory if missing
+                    # -------------------------------------------------
                     if update_result.matched_count == 0:
                         await media_clicks_col.update_one(
-                            { "userId": user_oid },
+                            {"userId": user_oid},
                             {
-                                "$setOnInsert": {
-                                    "userId": user_oid,
-                                    "createdAt": now,
-                                    "ai_edit_complete": 0   # initialize safely
-                                },
                                 "$inc": {
                                     "ai_edit_complete": 1
                                 },
@@ -372,12 +383,11 @@ async def face_swap_api(
                                         "lastClickedAt": now
                                     }
                                 }
-                            },
-                            upsert=True
+                            }
                         )
             
                     logger.info(
-                        "MEDIA_CLICK | user=%s | subCategory=%s | ai_edit_complete++",
+                        "[MEDIA_CLICK] user=%s subCategory=%s ai_edit_complete++",
                         user_id,
                         str(subcategory_oid)
                     )
@@ -387,6 +397,7 @@ async def face_swap_api(
 
 
 
+
         
             # if user_id:
             #     try: