Add ai_edit_complete and ai_edit_last_date fields to media_clicks collection

app.py

@@ -226,21 +226,8 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
                 lastClickedAt: Date
             }
         ],
-        createdAt: Date,
-        updatedAt: Date
-    }
-    """
-    """Save or update media click in admin database media_clicks collection
-    Structure:
-    {
-        userId: ObjectId,
-        categories: [
-            {
-                categoryId: ObjectId,
-                click_count: int,
-                lastClickedAt: Date
-            }
-        ],
+        ai_edit_complete: int,  # Count of times user used any model (default: 0 for new users)
+        ai_edit_last_date: str,  # Last date user used any model (format: DD/MM/YYYY, e.g., "12/12/2025")
         createdAt: Date,
         updatedAt: Date
     }
@@ -303,6 +290,8 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
             return False
         
         now = datetime.utcnow()
+        # Format date as DD/MM/YYYY (e.g., 12/12/2025)
+        date_formatted = now.strftime("%d/%m/%Y")
         
         # Check if document with userId exists
         existing_doc = collection.find_one({"userId": user_object_id})
@@ -327,15 +316,20 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
             
             if category_exists:
                 # Category exists, increment click_count and update lastClickedAt
+                # For existing users: increment ai_edit_complete by 1 and update ai_edit_last_date to current date
                 result = collection.update_one(
                     {
                         "userId": user_object_id,
                         "categories.categoryId": category_object_id
                     },
                     {
-                        "$inc": {"categories.$.click_count": 1},
+                        "$inc": {
+                            "categories.$.click_count": 1,
+                            "ai_edit_complete": 1  # Increment, default to 0 if field doesn't exist
+                        },
                         "$set": {
                             "categories.$.lastClickedAt": now,
+                            "ai_edit_last_date": date_formatted,
                             "updatedAt": now
                         }
                     }
@@ -346,6 +340,7 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
                     print(f"[MongoDB] WARNING: Update matched but did not modify - userId: {user_object_id}, categoryId: {category_object_id}, matched: {result.matched_count}, modified: {result.modified_count}")
             else:
                 # Category doesn't exist, add new category to array
+                # For existing users: increment ai_edit_complete by 1 and update ai_edit_last_date to current date
                 print(f"[MongoDB] Category not found in existing document, adding new category - userId: {user_object_id}, categoryId: {category_object_id}")
                 result = collection.update_one(
                     {"userId": user_object_id},
@@ -357,7 +352,13 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
                                 "lastClickedAt": now
                             }
                         },
-                        "$set": {"updatedAt": now}
+                        "$inc": {
+                            "ai_edit_complete": 1  # Increment, default to 0 if field doesn't exist
+                        },
+                        "$set": {
+                            "ai_edit_last_date": date_formatted,
+                            "updatedAt": now
+                        }
                     }
                 )
                 if result.modified_count > 0:
@@ -367,13 +368,15 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
         else:
             # Document doesn't exist, try to create new document
             # Use find_one_and_update with upsert which handles conflicts better
+            # For new users: ai_edit_complete = 0 (default), then increment to 1 when adding category
             try:
                 result = collection.find_one_and_update(
                     {"userId": user_object_id},
                     {
                         "$setOnInsert": {
                             "userId": user_object_id,
-                            "createdAt": now
+                            "createdAt": now,
+                            "ai_edit_complete": 0  # Default value for new users
                         },
                         "$push": {
                             "categories": {
@@ -382,7 +385,11 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
                                 "lastClickedAt": now
                             }
                         },
+                        "$inc": {
+                            "ai_edit_complete": 1  # Increment from 0 to 1 when adding first category
+                        },
                         "$set": {
+                            "ai_edit_last_date": date_formatted,
                             "updatedAt": now
                         }
                     },
@@ -409,7 +416,13 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
                                         "lastClickedAt": now
                                     }
                                 },
-                                "$set": {"updatedAt": now}
+                                "$inc": {
+                                    "ai_edit_complete": 1  # Increment, default to 0 if field doesn't exist
+                                },
+                                "$set": {
+                                    "ai_edit_last_date": date_formatted,
+                                    "updatedAt": now
+                                }
                             }
                         )
                         print(f"[MongoDB] Added category to existing document - userId: {user_object_id}, categoryId: {category_object_id}")
@@ -426,7 +439,8 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
                                 {
                                     "$setOnInsert": {
                                         "userId": user_object_id,
-                                        "createdAt": now
+                                        "createdAt": now,
+                                        "ai_edit_complete": 0  # Default value for new users
                                     },
                                     "$push": {
                                         "categories": {
@@ -435,7 +449,11 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
                                             "lastClickedAt": now
                                         }
                                     },
+                                    "$inc": {
+                                        "ai_edit_complete": 1  # Increment from 0 to 1 when adding first category
+                                    },
                                     "$set": {
+                                        "ai_edit_last_date": date_formatted,
                                         "updatedAt": now
                                     }
                                 },
@@ -463,6 +481,8 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
                 existing = collection.find_one({"userId": user_object_id})
                 if existing:
                     # Document exists, try to add category
+                    now_retry = datetime.utcnow()
+                    date_formatted_retry = now_retry.strftime("%d/%m/%Y")
                     result = collection.update_one(
                         {"userId": user_object_id},
                         {
@@ -470,10 +490,16 @@ def save_media_click(user_id: Optional[Union[int, str]], category_id: str):
                                 "categories": {
                                     "categoryId": category_object_id,
                                     "click_count": 1,
-                                    "lastClickedAt": datetime.utcnow()
+                                    "lastClickedAt": now_retry
                                 }
                             },
-                            "$set": {"updatedAt": datetime.utcnow()}
+                            "$inc": {
+                                "ai_edit_complete": 1  # Increment, default to 0 if field doesn't exist
+                            },
+                            "$set": {
+                                "ai_edit_last_date": date_formatted_retry,
+                                "updatedAt": now_retry
+                            }
                         }
                     )
                     print(f"[MongoDB] Recovered from duplicate key error - updated document")