Update app.py

app.py

@@ -217,29 +217,53 @@ async def generate(
             # If category or user does not exist
             if update_res.matched_count == 0:
                 media_clicks_col.update_one(
-                    {"userId": user_oid},
                     {
-                        "$setOnInsert": {
-                            "createdAt": now,
-                            "ai_edit_complete": 0
-                        },
-                        "$set": {
-                            "updatedAt": now,
-                            "ai_edit_last_date": now
-                        },
-                        "$inc": {
-                            "ai_edit_complete": 1
-                        },
-                        "$push": {
-                            "categories": {
-                                "categoryId": category_oid,
-                                "click_count": 1,
-                                "lastClickedAt": now
+                        "userId": user_oid,
+                        "categories.categoryId": category_oid
+                    },
+                    [
+                        {
+                            "$set": {
+                                "updatedAt": now,
+                                "ai_edit_last_date": now,
+                                "ai_edit_complete": {
+                                    "$add": [
+                                        { "$ifNull": ["$ai_edit_complete", 0] },
+                                        1
+                                    ]
+                                },
+                                "categories": {
+                                    "$map": {
+                                        "input": "$categories",
+                                        "as": "cat",
+                                        "in": {
+                                            "$cond": [
+                                                { "$eq": ["$$cat.categoryId", category_oid] },
+                                                {
+                                                    "$mergeObjects": [
+                                                        "$$cat",
+                                                        {
+                                                            "click_count": {
+                                                                "$add": [
+                                                                    { "$ifNull": ["$$cat.click_count", 0] },
+                                                                    1
+                                                                ]
+                                                            },
+                                                            "lastClickedAt": now
+                                                        }
+                                                    ]
+                                                },
+                                                "$$cat"
+                                            ]
+                                        }
+                                    }
+                                }
                             }
                         }
-                    },
+                    ],
                     upsert=True
                 )
+
     
         except Exception as e:
             print("CATEGORY_LOG_ERROR:", e)