Update app.py

app.py

@@ -175,44 +175,61 @@ 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)
-
-            # Check category exists
-            category_doc = categories_col.find_one({"_id": category_oid})
-            if not category_doc:
+    
+            # Ensure category exists
+            if not categories_col.find_one({"_id": category_oid}):
                 raise HTTPException(400, f"Invalid category_id: {category_id}")
-
+    
             now = datetime.utcnow()
-
-            # Try updating existing category click
+    
+            # Try update existing category + AI edit fields
             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
+                        "categories.$.lastClickedAt": now,
+                        "ai_edit_last_date": now
                     },
-                    "$inc": {"categories.$.click_count": 1}
+                    "$inc": {
+                        "categories.$.click_count": 1,
+                        "ai_edit_complete": 1
+                    }
                 }
             )
-
-            # If user or category does not exist → create/push new
+    
+            # If category or user does not exist
             if update_res.matched_count == 0:
                 media_clicks_col.update_one(
                     {"userId": user_oid},
                     {
-                        "$setOnInsert": {"createdAt": now},
-                        "$set": {"updatedAt": now},
+                        "$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,
@@ -223,9 +240,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
 
     # -------------------------