Update app.py

app.py

@@ -175,27 +175,52 @@ async def generate(
     # -------------------------
     # 3. OPTIONAL CATEGORY CLICK LOGIC
     # -------------------------
+    # -------------------------
+    # 3. OPTIONAL CATEGORY CLICK LOGIC
+    # -------------------------
     
     if user_id and category_id:
         try:
             admin_client = MongoClient(os.getenv("ADMIN_MONGODB_URI"))
             admin_db = admin_client["adminPanel"]
-
+    
             categories_col = admin_db.categories
             media_clicks_col = admin_db.media_clicks
-
+    
             # Validate user_oid & category_oid
             user_oid = ObjectId(user_id)
             category_oid = ObjectId(category_id)
-
+    
             # Check category exists
             category_doc = categories_col.find_one({"_id": category_oid})
             if not category_doc:
                 raise HTTPException(400, f"Invalid category_id: {category_id}")
-
+    
             now = datetime.utcnow()
-
-            # Try updating existing category click
+    
+            # --------------------------------------------------
+            # AI EDIT USAGE TRACKING (GLOBAL PER USER)
+            # --------------------------------------------------
+            media_clicks_col.update_one(
+                {"userId": user_oid},
+                {
+                    "$set": {
+                        "ai_edit_last_date": now,
+                        "updatedAt": now
+                    },
+                    "$inc": {
+                        "ai_edit_complete": 1
+                    },
+                    "$setOnInsert": {
+                        "createdAt": now
+                    }
+                },
+                upsert=True
+            )
+    
+            # --------------------------------------------------
+            # CATEGORY CLICK LOGIC
+            # --------------------------------------------------
             update_res = media_clicks_col.update_one(
                 {"userId": user_oid, "categories.categoryId": category_oid},
                 {
@@ -203,16 +228,17 @@ async def generate(
                         "updatedAt": now,
                         "categories.$.lastClickedAt": now
                     },
-                    "$inc": {"categories.$.click_count": 1}
+                    "$inc": {
+                        "categories.$.click_count": 1
+                    }
                 }
             )
-
-            # If user or category does not exist → create/push new
+    
+            # If category does not exist → push new
             if update_res.matched_count == 0:
                 media_clicks_col.update_one(
                     {"userId": user_oid},
                     {
-                        "$setOnInsert": {"createdAt": now},
                         "$set": {"updatedAt": now},
                         "$push": {
                             "categories": {
@@ -224,9 +250,63 @@ async def generate(
                     },
                     upsert=True
                 )
-
+    
         except Exception as e:
             print("CATEGORY_LOG_ERROR:", e)
+            # DO NOT break image generation
+
+    
+    # if user_id and category_id:
+    #     try:
+    #         admin_client = MongoClient(os.getenv("ADMIN_MONGODB_URI"))
+    #         admin_db = admin_client["adminPanel"]
+
+    #         categories_col = admin_db.categories
+    #         media_clicks_col = admin_db.media_clicks
+
+    #         # Validate user_oid & category_oid
+    #         user_oid = ObjectId(user_id)
+    #         category_oid = ObjectId(category_id)
+
+    #         # Check category exists
+    #         category_doc = categories_col.find_one({"_id": category_oid})
+    #         if not category_doc:
+    #             raise HTTPException(400, f"Invalid category_id: {category_id}")
+
+    #         now = datetime.utcnow()
+
+    #         # Try updating existing category click
+    #         update_res = media_clicks_col.update_one(
+    #             {"userId": user_oid, "categories.categoryId": category_oid},
+    #             {
+    #                 "$set": {
+    #                     "updatedAt": now,
+    #                     "categories.$.lastClickedAt": now
+    #                 },
+    #                 "$inc": {"categories.$.click_count": 1}
+    #             }
+    #         )
+
+    #         # If user or category does not exist → create/push new
+    #         if update_res.matched_count == 0:
+    #             media_clicks_col.update_one(
+    #                 {"userId": user_oid},
+    #                 {
+    #                     "$setOnInsert": {"createdAt": now},
+    #                     "$set": {"updatedAt": now},
+    #                     "$push": {
+    #                         "categories": {
+    #                             "categoryId": category_oid,
+    #                             "click_count": 1,
+    #                             "lastClickedAt": now
+    #                         }
+    #                     }
+    #                 },
+    #                 upsert=True
+    #             )
+
+    #     except Exception as e:
+    #         print("CATEGORY_LOG_ERROR:", e)
             # but do NOT stop the image generation
 
     # -------------------------