Update app.py

app.py

@@ -175,152 +175,58 @@ async def generate(
     # -------------------------
     # 3. OPTIONAL CATEGORY CLICK LOGIC
     # -------------------------
-    # -------------------------
-# 3. OPTIONAL CATEGORY CLICK + AI EDIT TRACKING
-# -------------------------
+    
     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)
-    
-            # Ensure category exists
-            if not categories_col.find_one({"_id": category_oid}):
+
+            # 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 update existing category + AI edit fields
+
+            # Try updating existing category click
             update_res = media_clicks_col.update_one(
-                {
-                    "userId": user_oid,
-                    "categories.categoryId": category_oid
-                },
+                {"userId": user_oid, "categories.categoryId": category_oid},
                 {
                     "$set": {
                         "updatedAt": now,
-                        "categories.$.lastClickedAt": now,
-                        "ai_edit_last_date": now
+                        "categories.$.lastClickedAt": now
                     },
-                    "$inc": {
-                        "categories.$.click_count": 1,
-                        "ai_edit_complete": 1
-                    }
+                    "$inc": {"categories.$.click_count": 1}
                 }
             )
-    
-            # If category or user does not exist
+
+            # If user or category does not exist → create/push new
             if update_res.matched_count == 0:
                 media_clicks_col.update_one(
+                    {"userId": user_oid},
                     {
-                        "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"
-                                            ]
-                                        }
-                                    }
-                                }
+                        "$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)
-            # 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
 
     # -------------------------