Add ai_edit_complete and ai_edit_last_date fields to media_clicks collection

api/main.py

@@ -173,17 +173,23 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
         if doc:
             categories = doc.get("categories") or []
             if any(cat.get("categoryId") == category_obj for cat in categories):
+                # Category exists: increment click_count and ai_edit_complete, update dates
                 admin_media_clicks.update_one(
                     {"_id": doc["_id"], "categories.categoryId": category_obj},
                     {
-                        "$inc": {"categories.$.click_count": 1},
+                        "$inc": {
+                            "categories.$.click_count": 1,
+                            "ai_edit_complete": 1,  # $inc handles missing fields (backward compatible)
+                        },
                         "$set": {
                             "categories.$.lastClickedAt": now,
                             "updatedAt": now,
+                            "ai_edit_last_date": now,
                         },
                     },
                 )
             else:
+                # New category to existing document: push category, increment ai_edit_complete
                 admin_media_clicks.update_one(
                     {"_id": doc["_id"]},
                     {
@@ -194,23 +200,38 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
                                 "lastClickedAt": now,
                             }
                         },
-                        "$set": {"updatedAt": now},
+                        "$inc": {"ai_edit_complete": 1},  # $inc handles missing fields
+                        "$set": {
+                            "updatedAt": now,
+                            "ai_edit_last_date": now,
+                        },
                     },
                 )
         else:
-            admin_media_clicks.insert_one(
+            # New user: create document with default ai_edit_complete=0, then increment to 1
+            admin_media_clicks.update_one(
+                {"userId": user_obj},
                 {
-                    "userId": user_obj,
-                    "categories": [
-                        {
-                            "categoryId": category_obj,
-                            "click_count": 1,
-                            "lastClickedAt": now,
-                        }
-                    ],
-                    "createdAt": now,
-                    "updatedAt": now,
-                }
+                    "$setOnInsert": {
+                        "userId": user_obj,
+                        "categories": [
+                            {
+                                "categoryId": category_obj,
+                                "click_count": 1,
+                                "lastClickedAt": now,
+                            }
+                        ],
+                        "createdAt": now,
+                        "updatedAt": now,
+                        "ai_edit_complete": 0,  # Default for new users
+                    },
+                    "$inc": {"ai_edit_complete": 1},  # Increment to 1 on first use
+                    "$set": {
+                        "updatedAt": now,
+                        "ai_edit_last_date": now,
+                    },
+                },
+                upsert=True,
             )
     except Exception as err:
         err_str = str(err)