Update app.py

app.py

@@ -232,46 +232,56 @@ async def generate(
             # --------------------------------------------------
             now = datetime.utcnow()
             today_date = now.date()
-            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 []
             
+            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_updates = []
             
             if not daily_entries:
-                # First-ever usage → only today
                 daily_updates.append({"date": today_date, "count": 1})
             else:
-                existing_dates = {entry["date"]: entry["count"] for entry in daily_entries}
+                #  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())
             
-                # Fill all missing days between last_date_in_db and today-1 with 0
                 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)
             
-                # Add today if not already present
                 if today_date not in existing_dates:
                     daily_updates.append({"date": today_date, "count": 1})
             
-            # Push updates if any
             if daily_updates:
                 media_clicks_col.update_one(
                     {"userId": user_oid},
                     {"$push": {"ai_edit_daily_count": {"$each": daily_updates}}}
                 )
             
-            # Sort oldest → newest and trim to last 32 entries
+            # 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"])
+            daily_entries.sort(
+                key=lambda x: x["date"].date() if isinstance(x["date"], datetime) else x["date"]
+            )
+            
             if len(daily_entries) > 32:
                 daily_entries = daily_entries[-32:]
-                media_clicks_col.update_one(
-                    {"userId": user_oid},
-                    {"$set": {"ai_edit_daily_count": daily_entries}}
-                )
+            
+            media_clicks_col.update_one(
+                {"userId": user_oid},
+                {"$set": {"ai_edit_daily_count": daily_entries}}
+            )
+
 
             # --------------------------------------------------
             # CATEGORY CLICK LOGIC