Add ai_edit_daily_count tracking to media_clicks collection

app/database.py

@@ -3,8 +3,8 @@ MongoDB database connection and logging utilities, including admin media click l
 """ 
 import os 
 import logging 
-from datetime import datetime 
-from typing import Optional, Dict, Any, Union 
+from datetime import datetime, timedelta
+from typing import Optional, Dict, Any, Union, List
 from bson import ObjectId 
 from pymongo import MongoClient 
 from pymongo.errors import ConnectionFailure, ServerSelectionTimeoutError 
@@ -157,7 +157,115 @@ def _resolve_category_id(
     if not chosen: 
         chosen = default_category_id or os.getenv("DEFAULT_CATEGORY_FALLBACK", "69368fcd2e46bd68ae1889b2") 
 
-    return _objectid_from_any(chosen) 
+    return _objectid_from_any(chosen)
+
+
+def _get_today_midnight_utc() -> datetime:
+    """Get today's date at midnight UTC (timezone-naive for MongoDB compatibility)."""
+    now = datetime.utcnow()
+    return datetime(now.year, now.month, now.day)
+
+
+def _update_daily_count(collection, user_object_id: ObjectId, today: datetime) -> None:
+    """Update ai_edit_daily_count array: add today with count 1 if missing, fill gaps with count 0."""
+    # Ensure today is at midnight
+    today_start = today.replace(hour=0, minute=0, second=0, microsecond=0)
+    
+    # Check if today's date already exists
+    user_doc = collection.find_one({"userId": user_object_id})
+    
+    if not user_doc:
+        # User document doesn't exist yet - this shouldn't happen as we create it first
+        # But handle it gracefully by initializing the field
+        logger.warning("User document not found when updating daily count, initializing")
+        collection.update_one(
+            {"userId": user_object_id},
+            {
+                "$setOnInsert": {
+                    "ai_edit_daily_count": [{
+                        "date": today_start,
+                        "count": 1
+                    }]
+                }
+            },
+            upsert=False
+        )
+        return
+    
+    # Get existing daily counts
+    existing_counts = user_doc.get("ai_edit_daily_count", [])
+    
+    # Check if today's date already exists
+    today_exists = False
+    for entry in existing_counts:
+        entry_date = entry.get("date")
+        if entry_date:
+            # Normalize to midnight for comparison
+            if isinstance(entry_date, datetime):
+                normalized_date = entry_date.replace(hour=0, minute=0, second=0, microsecond=0)
+                if normalized_date == today_start:
+                    today_exists = True
+                    break
+    
+    # If today exists, do nothing (leave it as is)
+    if today_exists:
+        logger.debug("Today's date already exists in daily count, leaving unchanged: %s", today_start.isoformat())
+        return
+    
+    # Today doesn't exist - need to add it and fill missing dates
+    # Find the latest date in existing counts
+    last_date = None
+    if existing_counts:
+        dates = []
+        for entry in existing_counts:
+            entry_date = entry.get("date")
+            if entry_date:
+                # Normalize to midnight for comparison
+                if isinstance(entry_date, datetime):
+                    dates.append(entry_date.replace(hour=0, minute=0, second=0, microsecond=0))
+        if dates:
+            last_date = max(dates)
+    
+    # Generate missing dates between last_date and today
+    dates_to_add = []
+    if last_date:
+        # Normalize last_date to midnight
+        last_date = last_date.replace(hour=0, minute=0, second=0, microsecond=0)
+        current_date = last_date + timedelta(days=1)
+        # Fill missing dates with count 0
+        while current_date < today_start:
+            dates_to_add.append({
+                "date": current_date,
+                "count": 0
+            })
+            current_date += timedelta(days=1)
+    
+    # Add today's entry with count 1
+    dates_to_add.append({
+        "date": today_start,
+        "count": 1
+    })
+    
+    # Push all missing dates (including today)
+    if dates_to_add:
+        # Ensure ai_edit_daily_count field exists
+        if "ai_edit_daily_count" not in user_doc:
+            collection.update_one(
+                {"userId": user_object_id},
+                {"$set": {"ai_edit_daily_count": []}}
+            )
+        
+        collection.update_one(
+            {"userId": user_object_id},
+            {
+                "$push": {
+                    "ai_edit_daily_count": {
+                        "$each": dates_to_add
+                    }
+                }
+            }
+        )
+        logger.debug("Added %d daily count entries (including today with count 1)", len(dates_to_add)) 
 
 def log_api_call(
     endpoint: str,
@@ -361,6 +469,7 @@ def log_media_click(
         user_object_id = _normalize_object_id(user_id) 
         category_object_id = _resolve_category_id(category_id, endpoint_path, default_category_id) 
         now = datetime.utcnow()
+        today = _get_today_midnight_utc()
         
         logger.info("Media click - user_id input: %s, normalized: %s, category_id input: %s, normalized: %s", 
                    user_id, str(user_object_id), category_id, str(category_object_id)) 
@@ -397,8 +506,8 @@ def log_media_click(
                         }, 
                         "$push": { 
                             "categories": { 
-                                "categoryId": category_object_id, 
-                                "click_count": 1, 
+                                "categoryId": category_object_id,
+                                "click_count": 1,
                                 "lastClickedAt": now, 
                             } 
                         }, 
@@ -412,7 +521,8 @@ def log_media_click(
                     { 
                         "$setOnInsert": {
                             "createdAt": now,
-                            "ai_edit_complete": 0  # Default 0 for new users
+                            "ai_edit_complete": 0,  # Default 0 for new users
+                            "ai_edit_daily_count": []  # Initialize empty array
                         }, 
                         "$inc": {"ai_edit_complete": 1},  # Increment to 1 on first use
                         "$set": {
@@ -421,14 +531,17 @@ def log_media_click(
                         }, 
                         "$push": { 
                             "categories": { 
-                                "categoryId": category_object_id, 
-                                "click_count": 1, 
+                                "categoryId": category_object_id,
+                                "click_count": 1,
                                 "lastClickedAt": now, 
                             } 
                         }, 
                     }, 
                     upsert=True, 
-                ) 
+                )
+        
+        # Update daily count (after document is created/updated)
+        _update_daily_count(collection, user_object_id, today)
 
         logger.info("Media click logged for user %s", str(user_object_id)) 
         return True 

postman_collection.json

@@ -1,436 +0,0 @@
-{
-	"info": {
-		"_postman_id": "image-colorization-api",
-		"name": "Image Colorization API",
-		"description": "API collection for Text-Guided Image Colorization using Hugging Face Inference API",
-		"schema": "https://schema.getpostman.com/json/collection/v2.1.0/collection.json"
-	},
-	"item": [
-		{
-			"name": "Health Check",
-			"request": {
-				"method": "GET",
-				"header": [],
-				"url": {
-					"raw": "{{base_url}}/health",
-					"host": [
-						"{{base_url}}"
-					],
-					"path": [
-						"health"
-					]
-				},
-				"description": "Check if the API is running and models are loaded"
-			},
-			"response": []
-		},
-		{
-			"name": "API Info",
-			"request": {
-				"method": "GET",
-				"header": [],
-				"url": {
-					"raw": "{{base_url}}/api",
-					"host": [
-						"{{base_url}}"
-					],
-					"path": [
-						"api"
-					]
-				},
-				"description": "Get API information and available endpoints"
-			},
-			"response": []
-		},
-		{
-			"name": "Colorize Image - Basic",
-			"request": {
-				"method": "POST",
-				"header": [
-					{
-						"key": "Authorization",
-						"value": "Bearer {{firebase_token}}",
-						"type": "text"
-					},
-					{
-						"key": "X-Firebase-AppCheck",
-						"value": "{{firebase_app_check}}",
-						"type": "text"
-					}
-				],
-				"body": {
-					"mode": "formdata",
-					"formdata": [
-						{
-							"key": "file",
-							"type": "file",
-							"src": [],
-							"description": "Image file to colorize"
-						},
-						{
-							"key": "user_id",
-							"value": "{{user_id}}",
-							"type": "text",
-							"description": "Optional user id (ObjectId or numeric) for media click logging"
-						},
-						{
-							"key": "category_id",
-							"value": "{{category_id}}",
-							"type": "text",
-							"description": "Optional category id; endpoint default used when omitted"
-						},
-						{
-							"key": "categoryId",
-							"value": "{{categoryId}}",
-							"type": "text",
-							"description": "Optional category id (alternative to category_id)"
-						}
-					]
-				},
-				"url": {
-					"raw": "{{base_url}}/colorize",
-					"host": [
-						"{{base_url}}"
-					],
-					"path": [
-						"colorize"
-					]
-				},
-				"description": "Colorize an image with default settings"
-			},
-			"response": []
-		},
-		{
-			"name": "Colorize Image - With Prompt",
-			"request": {
-				"method": "POST",
-				"header": [
-					{
-						"key": "Authorization",
-						"value": "Bearer {{firebase_token}}",
-						"type": "text"
-					},
-					{
-						"key": "X-Firebase-AppCheck",
-						"value": "{{firebase_app_check}}",
-						"type": "text"
-					}
-				],
-				"body": {
-					"mode": "formdata",
-					"formdata": [
-						{
-							"key": "file",
-							"type": "file",
-							"src": [],
-							"description": "Image file to colorize"
-						},
-						{
-							"key": "positive_prompt",
-							"value": "vibrant natural colors, high quality photo",
-							"type": "text",
-							"description": "Additional descriptive text to enhance the caption"
-						},
-						{
-							"key": "user_id",
-							"value": "{{user_id}}",
-							"type": "text",
-							"description": "Optional user id (ObjectId or numeric) for media click logging"
-						},
-						{
-							"key": "category_id",
-							"value": "{{category_id}}",
-							"type": "text",
-							"description": "Optional category id; endpoint default used when omitted"
-						},
-						{
-							"key": "categoryId",
-							"value": "{{categoryId}}",
-							"type": "text",
-							"description": "Optional category id (alternative to category_id)"
-						}
-					]
-				},
-				"url": {
-					"raw": "{{base_url}}/colorize",
-					"host": [
-						"{{base_url}}"
-					],
-					"path": [
-						"colorize"
-					]
-				},
-				"description": "Colorize an image with custom positive prompt"
-			},
-			"response": []
-		},
-		{
-			"name": "Colorize Image - Full Options",
-			"request": {
-				"method": "POST",
-				"header": [
-					{
-						"key": "Authorization",
-						"value": "Bearer {{firebase_token}}",
-						"type": "text"
-					},
-					{
-						"key": "X-Firebase-AppCheck",
-						"value": "{{firebase_app_check}}",
-						"type": "text"
-					}
-				],
-				"body": {
-					"mode": "formdata",
-					"formdata": [
-						{
-							"key": "file",
-							"type": "file",
-							"src": [],
-							"description": "Image file to colorize"
-						},
-						{
-							"key": "positive_prompt",
-							"value": "vibrant natural colors, high quality photo, detailed lighting",
-							"type": "text",
-							"description": "Additional descriptive text to enhance the caption"
-						},
-						{
-							"key": "negative_prompt",
-							"value": "low quality, monochrome, blurry, grainy",
-							"type": "text",
-							"description": "Words or phrases to avoid during generation"
-						},
-						{
-							"key": "seed",
-							"value": "123",
-							"type": "text",
-							"description": "Random seed for reproducible generation"
-						},
-						{
-							"key": "num_inference_steps",
-							"value": "8",
-							"type": "text",
-							"description": "Number of inference steps"
-						},
-						{
-							"key": "user_id",
-							"value": "{{user_id}}",
-							"type": "text",
-							"description": "Optional user id (ObjectId or numeric) for media click logging"
-						},
-						{
-							"key": "category_id",
-							"value": "{{category_id}}",
-							"type": "text",
-							"description": "Optional category id; endpoint default used when omitted"
-						},
-						{
-							"key": "categoryId",
-							"value": "{{categoryId}}",
-							"type": "text",
-							"description": "Optional category id (alternative to category_id)"
-						}
-					]
-				},
-				"url": {
-					"raw": "{{base_url}}/colorize",
-					"host": [
-						"{{base_url}}"
-					],
-					"path": [
-						"colorize"
-					]
-				},
-				"description": "Colorize an image with all available options"
-			},
-			"response": []
-		},
-		{
-			"name": "Download Result",
-			"request": {
-				"method": "GET",
-				"header": [
-					{
-						"key": "Authorization",
-						"value": "Bearer {{firebase_token}}",
-						"type": "text"
-					}
-				],
-				"url": {
-					"raw": "{{base_url}}/download/{{result_id}}",
-					"host": [
-						"{{base_url}}"
-					],
-					"path": [
-						"download",
-						"{{result_id}}"
-					]
-				},
-				"description": "Download colorized image by result ID (requires authentication)"
-			},
-			"response": []
-		},
-		{
-			"name": "Get Result (Public)",
-			"request": {
-				"method": "GET",
-				"header": [],
-				"url": {
-					"raw": "{{base_url}}/results/{{filename}}",
-					"host": [
-						"{{base_url}}"
-					],
-					"path": [
-						"results",
-						"{{filename}}"
-					]
-				},
-				"description": "Public endpoint to access colorized images by filename"
-			},
-			"response": []
-		},
-		{
-			"name": "Upload Image",
-			"request": {
-				"method": "POST",
-				"header": [
-					{
-						"key": "Authorization",
-						"value": "Bearer {{firebase_token}}",
-						"type": "text"
-					},
-					{
-						"key": "X-Firebase-AppCheck",
-						"value": "{{firebase_app_check}}",
-						"type": "text"
-					}
-				],
-				"body": {
-					"mode": "formdata",
-					"formdata": [
-						{
-							"key": "file",
-							"type": "file",
-							"src": [],
-							"description": "Image file to upload"
-						},
-						{
-							"key": "user_id",
-							"value": "{{user_id}}",
-							"type": "text",
-							"description": "Optional user id (ObjectId or numeric) for media click logging"
-						},
-						{
-							"key": "category_id",
-							"value": "{{category_id}}",
-							"type": "text",
-							"description": "Optional category id; endpoint default used when omitted"
-						},
-						{
-							"key": "categoryId",
-							"value": "{{categoryId}}",
-							"type": "text",
-							"description": "Optional category id (alternative to category_id)"
-						}
-					]
-				},
-				"url": {
-					"raw": "{{base_url}}/upload",
-					"host": [
-						"{{base_url}}"
-					],
-					"path": [
-						"upload"
-					]
-				},
-				"description": "Upload an image and get the uploaded image URL"
-			},
-			"response": []
-		},
-		{
-			"name": "Colorize Image - Without Auth",
-			"request": {
-				"method": "POST",
-				"header": [],
-				"body": {
-					"mode": "formdata",
-					"formdata": [
-						{
-							"key": "file",
-							"type": "file",
-							"src": [],
-							"description": "Image file to colorize"
-						},
-						{
-							"key": "positive_prompt",
-							"value": "colorize this image with vibrant colors",
-							"type": "text"
-						}
-					]
-				},
-				"url": {
-					"raw": "{{base_url}}/colorize",
-					"host": [
-						"{{base_url}}"
-					],
-					"path": [
-						"colorize"
-					]
-				},
-				"description": "Colorize image without authentication (only works if DISABLE_AUTH=true)"
-			},
-			"response": []
-		}
-	],
-	"variable": [
-		{
-			"key": "base_url",
-			"value": "https://YOUR_SPACE_URL",
-			"type": "string",
-			"description": "Base URL of your Hugging Face Space or local server"
-		},
-		{
-			"key": "firebase_token",
-			"value": "YOUR_FIREBASE_ID_TOKEN",
-			"type": "string",
-			"description": "Firebase authentication token"
-		},
-		{
-			"key": "result_id",
-			"value": "",
-			"type": "string",
-			"description": "Result ID from colorize response"
-		},
-		{
-			"key": "filename",
-			"value": "",
-			"type": "string",
-			"description": "Filename from colorize response (e.g., uuid.png)"
-		},
-		{
-			"key": "user_id",
-			"value": "",
-			"type": "string",
-			"description": "User ID for media click logging (ObjectId string, integer, or leave empty to auto-generate)"
-		},
-		{
-			"key": "category_id",
-			"value": "",
-			"type": "string",
-			"description": "Category ID for media click logging (ObjectId string, or leave empty for endpoint default)"
-		},
-		{
-			"key": "categoryId",
-			"value": "",
-			"type": "string",
-			"description": "Category ID (alternative to category_id)"
-		},
-		{
-			"key": "firebase_app_check",
-			"value": "YOUR_FIREBASE_APP_CHECK_TOKEN",
-			"type": "string",
-			"description": "Firebase App Check token"
-		}
-	]
-}