Update app.py

app.py

@@ -321,71 +321,11 @@ async def face_swap_api(
             # ------------------------------------------------------------------
             # # 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(
-            #             {
-            #                 "userId": user_oid,
-            #                 "subCategories.subCategoryId": subcategory_oid
-            #             },
-            #             {
-            #                 "$inc": {
-            #                     "subCategories.$.click_count": 1,
-            #                     "subCategories.$.ai_edit_complete": 1
-            #                 },
-            #                 "$set": {
-            #                     "subCategories.$.lastClickedAt": now,
-            #                     "subCategories.$.ai_edit_last_date": now,
-            #                     "updatedAt": now
-            #                 }
-            #             }
-            #         )
-            
-            #         # 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": {
-            #                         "userId": user_oid,
-            #                         "createdAt": now
-            #                     },
-            #                     "$set": {
-            #                         "updatedAt": now
-            #                     },
-            #                     "$push": {
-            #                         "subCategories": {
-            #                             "subCategoryId": subcategory_oid,
-            #                             "click_count": 1,
-            #                             "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(
                         {
@@ -393,23 +333,38 @@ async def face_swap_api(
                             "subCategories.subCategoryId": subcategory_oid
                         },
                         {
+                            "$inc": {
+                                "subCategories.$.click_count": 1,
+                                "ai_edit_complete": 1
+                            },
                             "$set": {
-                                "updatedAt": now,
-                                "subCategories.$.lastClickedAt": now
+                                "subCategories.$.lastClickedAt": now,
+                                "ai_edit_last_date": now,
+                                "updatedAt": now
                             },
-                            "$inc": {
-                                "subCategories.$.click_count": 1
+                            "$setOnInsert": {
+                                "createdAt": now
                             }
                         }
                     )
-
+            
                     # 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 },
+                                "$setOnInsert": {
+                                    "userId": user_oid,
+                                    "createdAt": now,
+                                    "ai_edit_complete": 0   # initialize safely
+                                },
+                                "$inc": {
+                                    "ai_edit_complete": 1
+                                },
+                                "$set": {
+                                    "ai_edit_last_date": now,
+                                    "updatedAt": now
+                                },
                                 "$push": {
                                     "subCategories": {
                                         "subCategoryId": subcategory_oid,
@@ -420,10 +375,62 @@ async def face_swap_api(
                             },
                             upsert=True
                         )
-
+            
+                    logger.info(
+                        "MEDIA_CLICK | user=%s | subCategory=%s | ai_edit_complete++",
+                        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)
 #         # ------------------------------------------------------------------