Update api/main.py

api/main.py

@@ -3,7 +3,7 @@ import os
 import uuid
 import shutil
 import re
-from datetime import datetime
+from datetime import datetime, timedelta, date
 from typing import Dict, List, Optional
 
 import numpy as np
@@ -111,6 +111,89 @@ def _admin_logging_status() -> Dict[str, object]:
         "collection": admin_media_clicks.name,
     }
 
+
+def _build_ai_edit_daily_count(
+    existing: Optional[List[Dict[str, object]]],
+    today: date,
+) -> List[Dict[str, object]]:
+    """
+    Build / extend the ai_edit_daily_count array with the following rules:
+
+    - Case A (no existing data): return [{date: today, count: 1}]
+    - Case B (today already recorded): return list unchanged
+    - Case C (gap in days): fill missing days with count=0 and append today with count=1
+
+    The stored "date" value is a midnight UTC (naive UTC) datetime for the given day.
+    """
+
+    def _to_date_only(value: object) -> date:
+        if isinstance(value, datetime):
+            return value.date()
+        if isinstance(value, date):
+            return value
+        # Fallback: try parsing ISO string "YYYY-MM-DD" or full datetime
+        try:
+            text = str(value)
+            if len(text) == 10:
+                return datetime.strptime(text, "%Y-%m-%d").date()
+            return datetime.fromisoformat(text).date()
+        except Exception:
+            # If parsing fails, just treat as today to avoid crashing
+            return today
+
+    # Case A: first ever use (no array yet)
+    if not existing:
+        return [
+            {
+                "date": datetime(today.year, today.month, today.day),
+                "count": 1,
+            }
+        ]
+
+    # Work on a shallow copy so we don't mutate original in-place
+    result: List[Dict[str, object]] = list(existing)
+
+    last_entry = result[-1] if result else None
+    if not last_entry or "date" not in last_entry:
+        # If structure is unexpected, re-initialize safely
+        return [
+            {
+                "date": datetime(today.year, today.month, today.day),
+                "count": 1,
+            }
+        ]
+
+    last_date = _to_date_only(last_entry["date"])
+
+    # If somehow the last stored date is in the future, do nothing to avoid corrupting history
+    if last_date > today:
+        return result
+
+    # Case B: today's date already present as the last entry → unchanged
+    if last_date == today:
+        return result
+
+    # Case C: there is a gap, fill missing days with count=0 and append today with count=1
+    cursor = last_date + timedelta(days=1)
+    while cursor < today:
+        result.append(
+            {
+                "date": datetime(cursor.year, cursor.month, cursor.day),
+                "count": 0,
+            }
+        )
+        cursor += timedelta(days=1)
+
+    # Finally add today's presence indicator
+    result.append(
+        {
+            "date": datetime(today.year, today.month, today.day),
+            "count": 1,
+        }
+    )
+
+    return result
+
 def bearer_auth(authorization: Optional[str] = Header(default=None)) -> None:
     if not ENV_TOKEN:
         return
@@ -168,9 +251,12 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
         user_obj = _coerce_object_id(user_id)
         category_obj = _coerce_category_id(category_id)
         now = datetime.utcnow()
+        today = now.date()
 
         doc = admin_media_clicks.find_one({"userId": user_obj})
         if doc:
+            existing_daily = doc.get("ai_edit_daily_count")
+            updated_daily = _build_ai_edit_daily_count(existing_daily, today)
             categories = doc.get("categories") or []
             if any(cat.get("categoryId") == category_obj for cat in categories):
                 # Category exists: increment click_count and ai_edit_complete, update dates
@@ -185,6 +271,7 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
                             "categories.$.lastClickedAt": now,
                             "updatedAt": now,
                             "ai_edit_last_date": now,
+                            "ai_edit_daily_count": updated_daily,
                         },
                     },
                 )
@@ -204,11 +291,13 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
                         "$set": {
                             "updatedAt": now,
                             "ai_edit_last_date": now,
+                            "ai_edit_daily_count": updated_daily,
                         },
                     },
                 )
         else:
             # New user: create document with default ai_edit_complete=0, then increment to 1
+            daily_for_new = _build_ai_edit_daily_count(None, today)
             admin_media_clicks.update_one(
                 {"userId": user_obj},
                 {
@@ -224,6 +313,7 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
                         "createdAt": now,
                         "updatedAt": now,
                         "ai_edit_complete": 0,  # Default for new users
+                        "ai_edit_daily_count": daily_for_new,
                     },
                     "$inc": {"ai_edit_complete": 1},  # Increment to 1 on first use
                     "$set": {