Update app.py

app.py

@@ -319,12 +319,13 @@ async def face_swap_api(
             subcategory_oid = doc["_id"]
     
             # ------------------------------------------------------------------
+            # # MEDIA_CLICKS (ONLY IF user_id PRESENT
             # MEDIA_CLICKS (ONLY IF user_id PRESENT)
             if user_id:
                 try:
                     user_oid = ObjectId(user_id.strip())
                     now = datetime.utcnow()
-
+            
                     # 1) Try updating existing subCategory
                     update_result = await media_clicks_col.update_one(
                         {
@@ -332,37 +333,97 @@ async def face_swap_api(
                             "subCategories.subCategoryId": subcategory_oid
                         },
                         {
-                            "$set": {
-                                "updatedAt": now,
-                                "subCategories.$.lastClickedAt": now
-                            },
                             "$inc": {
-                                "subCategories.$.click_count": 1
+                                "subCategories.$.click_count": 1,
+                                "subCategories.$.ai_edit_complete": 1
+                            },
+                            "$set": {
+                                "subCategories.$.lastClickedAt": now,
+                                "subCategories.$.ai_edit_last_date": now,
+                                "updatedAt": now
                             }
                         }
                     )
-
-                    # 2) If no subCategory exists → push new one (or create doc)
+            
+                    # 2) If subCategory does NOT exist → push new entry
                     if update_result.matched_count == 0:
                         await media_clicks_col.update_one(
                             { "userId": user_oid },
                             {
-                                "$setOnInsert": { "createdAt": now },
-                                "$set": { "updatedAt": now },
+                                "$setOnInsert": {
+                                    "userId": user_oid,
+                                    "createdAt": now
+                                },
+                                "$set": {
+                                    "updatedAt": now
+                                },
                                 "$push": {
                                     "subCategories": {
                                         "subCategoryId": subcategory_oid,
                                         "click_count": 1,
-                                        "lastClickedAt": now
+                                        "lastClickedAt": now,
+                                        "ai_edit_complete": 1,
+                                        "ai_edit_last_date": now
                                     }
                                 }
                             },
                             upsert=True
                         )
-
+            
+                    logger.info(
+                        "MEDIA_CLICK_SUBCATEGORY | user=%s | subCategory=%s | ai_edit_complete+1",
+                        user_id,
+                        str(subcategory_oid)
+                    )
+            
                 except Exception as media_err:
                     logger.error(f"MEDIA_CLICK ERROR: {media_err}")
 
+
+        
+            # if user_id:
+            #     try:
+            #         user_oid = ObjectId(user_id.strip())
+            #         now = datetime.utcnow()
+
+            #         # 1) Try updating existing subCategory
+            #         update_result = await media_clicks_col.update_one(
+            #             {
+            #                 "userId": user_oid,
+            #                 "subCategories.subCategoryId": subcategory_oid
+            #             },
+            #             {
+            #                 "$set": {
+            #                     "updatedAt": now,
+            #                     "subCategories.$.lastClickedAt": now
+            #                 },
+            #                 "$inc": {
+            #                     "subCategories.$.click_count": 1
+            #                 }
+            #             }
+            #         )
+
+            #         # 2) If no subCategory exists → push new one (or create doc)
+            #         if update_result.matched_count == 0:
+            #             await media_clicks_col.update_one(
+            #                 { "userId": user_oid },
+            #                 {
+            #                     "$setOnInsert": { "createdAt": now },
+            #                     "$set": { "updatedAt": now },
+            #                     "$push": {
+            #                         "subCategories": {
+            #                             "subCategoryId": subcategory_oid,
+            #                             "click_count": 1,
+            #                             "lastClickedAt": now
+            #                         }
+            #                     }
+            #                 },
+            #                 upsert=True
+            #             )
+
+            #     except Exception as media_err:
+            #         logger.error(f"MEDIA_CLICK ERROR: {media_err}")
+
 #         # ------------------------------------------------------------------
 #         # CASE 2 : target_category_id → DigitalOcean path (unchanged logic)
 #         # ------------------------------------------------------------------