Update app.py

app.py

@@ -231,7 +231,7 @@ async def generate(
             # DAILY COUNT LOGIC
             # --------------------------------------------------
             now = datetime.utcnow()
-            today_date = now.date()
+            today_date = datetime(now.year, now.month, now.day)
             
             doc = media_clicks_col.find_one(
                 {"userId": user_oid},
@@ -239,47 +239,45 @@ async def generate(
             )
             
             daily_entries = doc.get("ai_edit_daily_count", []) if doc else []
-            daily_updates = []
-            
-            if not daily_entries:
-                daily_updates.append({"date": today_date, "count": 1})
-            else:
-                #  Normalize DB dates (THIS FIXES THE ERROR)
-                existing_dates = {
-                    (entry["date"].date() if isinstance(entry["date"], datetime) else entry["date"]): entry["count"]
-                    for entry in daily_entries
-                }
             
-                last_date_in_db = max(existing_dates.keys())
+            # Build UNIQUE date -> count map
+            daily_map = {}
+            for entry in daily_entries:
+                d = entry["date"]
+                d = datetime(d.year, d.month, d.day) if isinstance(d, datetime) else d
+                daily_map[d] = entry["count"]  # overwrite = no duplicates
             
-                next_day = last_date_in_db + timedelta(days=1)
-                while next_day < today_date:
-                    if next_day not in existing_dates:
-                        daily_updates.append({"date": next_day, "count": 0})
-                    next_day += timedelta(days=1)
+            # Find last known date
+            last_date = max(daily_map.keys()) if daily_map else today_date
             
-                if today_date not in existing_dates:
-                    daily_updates.append({"date": today_date, "count": 1})
+            # Fill ALL missing days with 0
+            next_day = last_date + timedelta(days=1)
+            while next_day < today_date:
+                daily_map.setdefault(next_day, 0)
+                next_day += timedelta(days=1)
             
-            if daily_updates:
-                media_clicks_col.update_one(
-                    {"userId": user_oid},
-                    {"$push": {"ai_edit_daily_count": {"$each": daily_updates}}}
-                )
+            # Mark today as used (binary)
+            daily_map[today_date] = 1
             
-            # Sort + trim
-            doc = media_clicks_col.find_one({"userId": user_oid}, {"ai_edit_daily_count": 1})
-            daily_entries = doc.get("ai_edit_daily_count", []) if doc else []
-            daily_entries.sort(
-                key=lambda x: x["date"].date() if isinstance(x["date"], datetime) else x["date"]
-            )
+            # Rebuild list (OLD → NEW)
+            final_daily_entries = [
+                {"date": d, "count": daily_map[d]}
+                for d in sorted(daily_map.keys())
+            ]
             
-            if len(daily_entries) > 32:
-                daily_entries = daily_entries[-32:]
+            # Keep last 32 days only
+            final_daily_entries = final_daily_entries[-32:]
             
+            # ATOMIC REPLACE (NO PUSH)
             media_clicks_col.update_one(
                 {"userId": user_oid},
-                {"$set": {"ai_edit_daily_count": daily_entries}}
+                {
+                    "$set": {
+                        "ai_edit_daily_count": final_daily_entries,
+                        "ai_edit_last_date": now,
+                        "updatedAt": now
+                    }
+                }
             )
 
 
@@ -318,81 +316,6 @@ async def generate(
     
         except Exception as e:
             print("CATEGORY_LOG_ERROR:", e)
-    
-    # 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()
-    
-    #         # --------------------------------------------------
-    #         # AI EDIT USAGE TRACKING (GLOBAL PER USER)
-    #         # --------------------------------------------------
-    #         media_clicks_col.update_one(
-    #             {"userId": user_oid},
-    #             {
-    #                 "$set": {
-    #                     "ai_edit_last_date": now,
-    #                     "updatedAt": now
-    #                 },
-    #                 "$inc": {
-    #                     "ai_edit_complete": 1
-    #                 },
-    #                 "$setOnInsert": {
-    #                     "createdAt": now
-    #                 }
-    #             },
-    #             upsert=True
-    #         )
-    
-    #         # --------------------------------------------------
-    #         # CATEGORY CLICK LOGIC
-    #         # --------------------------------------------------
-    #         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 category does not exist → push new
-    #         if update_res.matched_count == 0:
-    #             media_clicks_col.update_one(
-    #                 {"userId": user_oid},
-    #                 {
-    #                     "$set": {"updatedAt": now},
-    #                     "$push": {
-    #                         "categories": {
-    #                             "categoryId": category_oid,
-    #                             "click_count": 1,
-    #                             "lastClickedAt": now
-    #                         }
-    #                     }
-    #                 },
-    #                 upsert=True
-    #             )
-    
-    #     except Exception as e:
-    #         print("CATEGORY_LOG_ERROR:", e)
     # -------------------------
     # 4. HF INFERENCE
     # -------------------------