Update app.py

app.py

@@ -105,6 +105,14 @@ def prepare_image(file_bytes: bytes) -> Image.Image:
     Open image and resize if larger than 1024x1024
     """
     img = Image.open(io.BytesIO(file_bytes)).convert("RGB")
+
+    # ✅ MIN SIZE CHECK
+    if img.width < 200 or img.height < 200:
+        raise HTTPException(
+            status_code=400,
+            detail="Image size is below 200x200 pixels. Please upload a larger image."
+        )
+    img = Image.open(io.BytesIO(file_bytes)).convert("RGB")
     img = resize_image_if_needed(img, max_size=(1024, 1024))
     return img
     
@@ -222,6 +230,8 @@ async def generate(
             # --------------------------------------------------
             # DAILY COUNT LOGIC
             # --------------------------------------------------
+            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 []
             
@@ -231,8 +241,8 @@ async def generate(
                 # First-ever usage → only today
                 daily_updates.append({"date": today_date, "count": 1})
             else:
-                existing_dates = {entry["date"].date(): entry["count"] for entry in daily_entries}
-                last_date_in_db = max(entry["date"].date() for entry in daily_entries)
+                existing_dates = {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)