Update app.py

app.py

@@ -320,72 +320,12 @@ 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}")
-
-
-        
+            # # 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(
             #             {
@@ -393,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)
 #         # ------------------------------------------------------------------