Update app.py

app.py

@@ -8,7 +8,7 @@ import numpy as np
 import threading
 import subprocess
 import logging
-from datetime import datetime, timezone
+from datetime import datetime, timezone,timedelta
 import insightface
 from insightface.app import FaceAnalysis
 from huggingface_hub import hf_hub_download
@@ -165,13 +165,17 @@ async def log_faceswap_hit(user_email: str, status: str, start_time: datetime, e
 async def log_media_click(user_id: str, category_oid_str: str):
     """
     Logs a click event to the media_clicks collection against the Category.
-    Safely updates ai_edit_last_date and ai_edit_complete at ROOT level.
+    Safely updates ai_edit_last_date, ai_edit_complete, and ai_edit_daily_count.
     """
     try:
         user_oid = ObjectId(user_id.strip())
         category_oid = ObjectId(category_oid_str.strip())
         now = datetime.utcnow()
 
+        # Normalize dates (UTC midnight)
+        today_date = datetime(now.year, now.month, now.day)
+        yesterday_date = today_date - timedelta(days=1)
+
         # Optional safety check
         if not await categories_col.find_one({"_id": category_oid}):
             logger.warning(
@@ -189,12 +193,44 @@ async def log_media_click(user_id: str, category_oid_str: str):
                 "$setOnInsert": {
                     "userId": user_oid,
                     "createdAt": now,
-                    "ai_edit_complete": 0
+                    "ai_edit_complete": 0,
+                    "ai_edit_daily_count": []
                 }
             },
             upsert=True
         )
 
+        # -------------------------------------------------
+        # STEP 1.5: Handle ai_edit_daily_count
+        # -------------------------------------------------
+        doc = await media_clicks_col.find_one(
+            {"userId": user_oid},
+            {"ai_edit_daily_count": 1}
+        )
+
+        daily_entries = doc.get("ai_edit_daily_count", []) if doc else []
+        is_first_time = len(daily_entries) == 0
+
+        existing_dates = {entry["date"].date(): entry["count"] for entry in daily_entries}
+        daily_updates = []
+
+        if is_first_time:
+            # First ever usage → only today
+            daily_updates.append({"date": today_date, "count": 1})
+        else:
+            # Backfill yesterday if missing
+            if yesterday_date.date() not in existing_dates:
+                daily_updates.append({"date": yesterday_date, "count": 0})
+            # Insert today if missing
+            if today_date.date() not in existing_dates:
+                daily_updates.append({"date": today_date, "count": 1})
+
+        if daily_updates:
+            await media_clicks_col.update_one(
+                {"userId": user_oid},
+                {"$push": {"ai_edit_daily_count": {"$each": daily_updates}}}
+            )
+
         # -------------------------------------------------
         # STEP 2: Try updating existing category
         # -------------------------------------------------
@@ -241,52 +277,81 @@ async def log_media_click(user_id: str, category_oid_str: str):
             )
 
         logger.info(
-            "[MEDIA_CLICK] user=%s category=%s ai_edit_complete incremented",
+            "[MEDIA_CLICK] user=%s category=%s ai_edit_complete incremented & daily_tracked",
             user_id,
             category_oid_str
         )
 
     except Exception as media_err:
         logger.error("MEDIA_CLICK LOGGING ERROR: %s", media_err)
-
 # async def log_media_click(user_id: str, category_oid_str: str):
 #     """
 #     Logs a click event to the media_clicks collection against the Category.
+#     Safely updates ai_edit_last_date and ai_edit_complete at ROOT level.
 #     """
 #     try:
 #         user_oid = ObjectId(user_id.strip())
 #         category_oid = ObjectId(category_oid_str.strip())
 #         now = datetime.utcnow()
 
-#         # Check if category exists (optional, but prevents logging invalid IDs)
+#         # Optional safety check
 #         if not await categories_col.find_one({"_id": category_oid}):
-#             logger.warning(f"Category ID {category_oid_str} not found. Skipping click logging.")
+#             logger.warning(
+#                 "Category ID %s not found. Skipping media click logging.",
+#                 category_oid_str
+#             )
 #             return
 
-#         # 1) Try updating existing category click entry
+#         # -------------------------------------------------
+#         # STEP 1: Ensure user document + root fields exist
+#         # -------------------------------------------------
+#         await media_clicks_col.update_one(
+#             {"userId": user_oid},
+#             {
+#                 "$setOnInsert": {
+#                     "userId": user_oid,
+#                     "createdAt": now,
+#                     "ai_edit_complete": 0
+#                 }
+#             },
+#             upsert=True
+#         )
+
+#         # -------------------------------------------------
+#         # STEP 2: Try updating existing category
+#         # -------------------------------------------------
 #         update_result = await media_clicks_col.update_one(
 #             {
 #                 "userId": user_oid,
 #                 "categories.categoryId": category_oid
 #             },
 #             {
-#                 "$set": {
-#                     "updatedAt": now,
-#                     "categories.$.lastClickedAt": now
-#                 },
 #                 "$inc": {
-#                     "categories.$.click_count": 1
+#                     "categories.$.click_count": 1,
+#                     "ai_edit_complete": 1
+#                 },
+#                 "$set": {
+#                     "categories.$.lastClickedAt": now,
+#                     "ai_edit_last_date": now,
+#                     "updatedAt": now
 #                 }
 #             }
 #         )
 
-#         # 2) If no category entry exists → push new one (or create doc)
+#         # -------------------------------------------------
+#         # STEP 3: Category not present → push new
+#         # -------------------------------------------------
 #         if update_result.matched_count == 0:
 #             await media_clicks_col.update_one(
-#                 { "userId": user_oid },
+#                 {"userId": user_oid},
 #                 {
-#                     "$setOnInsert": { "createdAt": now },
-#                     "$set": { "updatedAt": now },
+#                     "$inc": {
+#                         "ai_edit_complete": 1
+#                     },
+#                     "$set": {
+#                         "ai_edit_last_date": now,
+#                         "updatedAt": now
+#                     },
 #                     "$push": {
 #                         "categories": {
 #                             "categoryId": category_oid,
@@ -294,14 +359,18 @@ async def log_media_click(user_id: str, category_oid_str: str):
 #                             "lastClickedAt": now
 #                         }
 #                     }
-#                 },
-#                 upsert=True
+#                 }
 #             )
-        
-#         logger.info(f"Media click logged for User {user_id} on Category {category_oid_str}")
+
+#         logger.info(
+#             "[MEDIA_CLICK] user=%s category=%s ai_edit_complete incremented",
+#             user_id,
+#             category_oid_str
+#         )
 
 #     except Exception as media_err:
-#         logger.error(f"MEDIA_CLICK LOGGING ERROR: {media_err}")
+#         logger.error("MEDIA_CLICK LOGGING ERROR: %s", media_err)
+
 
 # --------------------- Face Swap Pipeline ---------------------
 swap_lock = threading.Lock()