Add MongoDB media click logging with per-user category counts

- Add MONGODB_ADMIN connection for admin database
- Implement log_media_click() with user/category normalization
- Support user_id from ObjectId, integer (deterministic), or auto-generate
- Support category_id and categoryId form fields
- Add endpoint defaults for colorization endpoints
- Update /upload and /colorize endpoints in all main files
- Update Postman collection with new fields
- Store data in media_clicks collection with userId, categories array, click_count, and timestamps

app/config.py

@@ -58,6 +58,10 @@ class Settings(BaseSettings):
     # MONGODB_URI should be set in Hugging Face Space secrets
     MONGODB_URI: str = os.getenv("MONGODB_URI", "")
     MONGODB_DB_NAME: str = os.getenv("MONGODB_DB_NAME", "colorization_db")
+    MONGODB_ADMIN: str = os.getenv("MONGODB_ADMIN", "") 
+    MONGODB_ADMIN_DB_NAME: str = os.getenv("MONGODB_ADMIN_DB_NAME", None) or os.getenv("MONGODB_DB_NAME", "colorization_db") 
+    DEFAULT_CATEGORY_COLORIZATION: str = os.getenv("DEFAULT_CATEGORY_COLORIZATION", "") 
+    DEFAULT_CATEGORY_FALLBACK: str = os.getenv("DEFAULT_CATEGORY_FALLBACK", "69368f722e46bd68ae188984") 
     
     class Config:
         env_file = ".env"

app/database.py

@@ -1,18 +1,23 @@
-"""
-MongoDB database connection and logging utilities
-"""
-import os
-import logging
-from datetime import datetime
-from typing import Optional, Dict, Any
-from pymongo import MongoClient
-from pymongo.errors import ConnectionFailure, ServerSelectionTimeoutError
+""" 
+MongoDB database connection and logging utilities, including admin media click logging. 
+""" 
+import os 
+import logging 
+from datetime import datetime 
+from typing import Optional, Dict, Any, Union 
+from bson import ObjectId 
+from pymongo import MongoClient 
+from pymongo.errors import ConnectionFailure, ServerSelectionTimeoutError 
 
 logger = logging.getLogger(__name__)
 
-# MongoDB connection
-_client: Optional[MongoClient] = None
-_db = None
+# MongoDB connection 
+_client: Optional[MongoClient] = None 
+_db = None 
+
+# Admin MongoDB connection (media_clicks) 
+_admin_client: Optional[MongoClient] = None 
+_admin_db = None 
 
 def get_mongodb_client() -> Optional[MongoClient]:
     """Get or create MongoDB client"""
@@ -48,6 +53,103 @@ def get_database():
             logger.warning("MongoDB client not available")
     return _db
 
+
+def get_admin_client() -> Optional[MongoClient]: 
+    """Get or create admin MongoDB client (for media_clicks collection).""" 
+    global _admin_client 
+    if _admin_client is None: 
+        mongodb_uri = os.getenv("MONGODB_ADMIN") 
+        if not mongodb_uri: 
+            logger.warning("MONGODB_ADMIN environment variable not set. Admin MongoDB features will be disabled.") 
+            return None 
+        try: 
+            _admin_client = MongoClient( 
+                mongodb_uri, 
+                serverSelectionTimeoutMS=5000, 
+                connectTimeoutMS=5000, 
+            ) 
+            _admin_client.admin.command("ping") 
+            logger.info("Admin MongoDB connection established successfully") 
+        except (ConnectionFailure, ServerSelectionTimeoutError) as exc: 
+            logger.error("Failed to connect to Admin MongoDB: %s", str(exc)) 
+            _admin_client = None 
+    return _admin_client 
+
+
+def get_admin_database(): 
+    """Get admin database instance.""" 
+    global _admin_db 
+    if _admin_db is None: 
+        client = get_admin_client() 
+        if client: 
+            db_name = os.getenv("MONGODB_ADMIN_DB_NAME", os.getenv("MONGODB_DB_NAME", "colorization_db")) 
+            _admin_db = client[db_name] 
+        else: 
+            logger.warning("Admin MongoDB client not available") 
+    return _admin_db 
+
+
+def _normalize_object_id(raw_value: Optional[Union[str, int, ObjectId]]) -> ObjectId: 
+    """Normalize user id inputs into a deterministic ObjectId.""" 
+    if isinstance(raw_value, ObjectId): 
+        return raw_value 
+
+    if raw_value is None: 
+        return ObjectId() 
+
+    try: 
+        if isinstance(raw_value, int) or (isinstance(raw_value, str) and raw_value.strip().lstrip("-").isdigit()): 
+            int_value = int(str(raw_value).strip()) 
+            hex_str = format(abs(int_value), "x").zfill(24)[-24:] 
+            if ObjectId.is_valid(hex_str): 
+                return ObjectId(hex_str) 
+    except Exception as exc:  # pragma: no cover - defensive 
+        logger.debug("Numeric user id normalization failed: %s", str(exc)) 
+
+    if isinstance(raw_value, str): 
+        candidate = raw_value.strip() 
+        if ObjectId.is_valid(candidate): 
+            return ObjectId(candidate) 
+
+    return ObjectId() 
+
+
+def _objectid_from_any(value: str) -> ObjectId: 
+    """Convert arbitrary string into an ObjectId deterministically when possible.""" 
+    if ObjectId.is_valid(value): 
+        return ObjectId(value) 
+    try: 
+        hex_str = value.encode("utf-8").hex().zfill(24)[-24:] 
+        if ObjectId.is_valid(hex_str): 
+            return ObjectId(hex_str) 
+    except Exception as exc:  # pragma: no cover - defensive 
+        logger.debug("Category id normalization failed: %s", str(exc)) 
+    return ObjectId() 
+
+
+def _resolve_category_id( 
+    category_id: Optional[str], 
+    endpoint_path: Optional[str], 
+    default_category_id: Optional[str], 
+) -> ObjectId: 
+    """Pick category id from explicit value, endpoint default, or fallback.""" 
+    endpoint_map = { 
+        "colorization": os.getenv("DEFAULT_CATEGORY_COLORIZATION"), 
+        "upload": os.getenv("DEFAULT_CATEGORY_COLORIZATION"), 
+        "colorize": os.getenv("DEFAULT_CATEGORY_COLORIZATION"), 
+    } 
+    normalized_endpoint = None 
+    if endpoint_path: 
+        normalized_endpoint = endpoint_path.strip("/").split("/")[0].lower() or None 
+
+    chosen = category_id 
+    if not chosen and normalized_endpoint and endpoint_map.get(normalized_endpoint): 
+        chosen = endpoint_map[normalized_endpoint] 
+    if not chosen: 
+        chosen = default_category_id or os.getenv("DEFAULT_CATEGORY_FALLBACK", "69368f722e46bd68ae188984") 
+
+    return _objectid_from_any(chosen) 
+
 def log_api_call(
     endpoint: str,
     method: str,
@@ -198,12 +300,75 @@ def log_colorization(
         logger.error("Failed to log colorization to MongoDB: %s", str(e))
         return False
 
+
+def log_media_click( 
+    user_id: Optional[Union[str, int, ObjectId]], 
+    category_id: Optional[str], 
+    *, 
+    endpoint_path: Optional[str] = None, 
+    default_category_id: Optional[str] = None, 
+) -> bool: 
+    """Log media clicks into the admin MongoDB (media_clicks collection).""" 
+    try: 
+        db = get_admin_database() 
+        if db is None: 
+            logger.warning("Admin MongoDB not available, skipping media click log") 
+            return False 
+
+        collection = db["media_clicks"] 
+
+        # Drop legacy index to avoid duplicate key errors (best effort) 
+        try: 
+            collection.drop_index("user_id_1_header_1_media_id_1") 
+        except Exception as exc: 
+            logger.debug("Legacy index drop skipped: %s", str(exc)) 
+
+        user_object_id = _normalize_object_id(user_id) 
+        category_object_id = _resolve_category_id(category_id, endpoint_path, default_category_id) 
+        now = datetime.utcnow() 
+
+        update_existing = collection.update_one( 
+            {"userId": user_object_id, "categories.categoryId": category_object_id}, 
+            { 
+                "$inc": {"categories.$.click_count": 1}, 
+                "$set": {"categories.$.lastClickedAt": now, "updatedAt": now}, 
+            }, 
+        ) 
+
+        if update_existing.matched_count == 0: 
+            collection.update_one( 
+                {"userId": user_object_id}, 
+                { 
+                    "$setOnInsert": {"createdAt": now}, 
+                    "$set": {"updatedAt": now}, 
+                    "$push": { 
+                        "categories": { 
+                            "categoryId": category_object_id, 
+                            "click_count": 1, 
+                            "lastClickedAt": now, 
+                        } 
+                    }, 
+                }, 
+                upsert=True, 
+            ) 
+
+        logger.info("Media click logged for user %s", str(user_object_id)) 
+        return True 
+    except Exception as exc: 
+        logger.error("Failed to log media click to admin MongoDB: %s", str(exc)) 
+        return False 
+
 def close_connection():
     """Close MongoDB connection"""
-    global _client, _db
+    global _client, _db, _admin_client, _admin_db 
     if _client:
         _client.close()
         _client = None
         _db = None
         logger.info("MongoDB connection closed")
+    if _admin_client: 
+        _admin_client.close() 
+        _admin_client = None 
+        _admin_db = None 
+        logger.info("Admin MongoDB connection closed") 
 

app/main.py

@@ -1,4 +1,5 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException, Header, Request
+from fastapi import FastAPI, File, UploadFile, HTTPException, Header, Request, Form
+from typing import Optional
 from fastapi.responses import FileResponse
 from huggingface_hub import hf_hub_download
 import uuid
@@ -8,7 +9,18 @@ import json
 from PIL import Image
 import torch
 from torchvision import transforms
-from app.database import get_database, log_api_call, log_image_upload, log_colorization, close_connection
+from app.database import (
+    get_database,
+    log_api_call,
+    log_image_upload,
+    log_colorization,
+    log_media_click,
+    close_connection,
+)
+try:
+    from firebase_admin import auth as firebase_auth
+except ImportError:
+    firebase_auth = None
 
 # -------------------------------------------------
 # 🚀 FastAPI App
@@ -43,6 +55,8 @@ RESULTS_DIR = "/tmp/results"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 os.makedirs(RESULTS_DIR, exist_ok=True)
 
+MEDIA_CLICK_DEFAULT_CATEGORY = os.getenv("DEFAULT_CATEGORY_FALLBACK", "69368f722e46bd68ae188984")
+
 # -------------------------------------------------
 # 🧠 Load GAN Colorization Model
 # -------------------------------------------------
@@ -114,6 +128,10 @@ def verify_app_check_token(token: str):
         raise HTTPException(status_code=401, detail="Invalid Firebase App Check token")
     return True
 
+def _resolve_user_id(request: Request, supplied_user_id: Optional[str]) -> Optional[str]:
+    """Return supplied user_id if provided, otherwise None (will auto-generate in log_media_click)."""
+    return supplied_user_id
+
 # -------------------------------------------------
 # 📤 Upload Image
 # -------------------------------------------------
@@ -121,11 +139,16 @@ def verify_app_check_token(token: str):
 async def upload_image(
     request: Request,
     file: UploadFile = File(...),
-    x_firebase_appcheck: str = Header(None)
+    x_firebase_appcheck: str = Header(None),
+    user_id: Optional[str] = Form(None),
+    category_id: Optional[str] = Form(None),
+    categoryId: Optional[str] = Form(None),
 ):
     verify_app_check_token(x_firebase_appcheck)
     
     ip_address = request.client.host if request.client else None
+    effective_user_id = _resolve_user_id(request, user_id)
+    effective_category_id = category_id or categoryId
 
     if not file.content_type.startswith("image/"):
         log_api_call(
@@ -160,6 +183,7 @@ async def upload_image(
         filename=file.filename or image_id,
         file_size=file_size,
         content_type=file.content_type or "image/jpeg",
+        user_id=effective_user_id,
         ip_address=ip_address
     )
     
@@ -169,9 +193,17 @@ async def upload_image(
         status_code=200,
         request_data={"filename": file.filename, "content_type": file.content_type},
         response_data=response_data,
+        user_id=effective_user_id,
         ip_address=ip_address
     )
 
+    log_media_click(
+        user_id=effective_user_id,
+        category_id=effective_category_id,
+        endpoint_path=str(request.url.path),
+        default_category_id=MEDIA_CLICK_DEFAULT_CATEGORY,
+    )
+
     return response_data
 
 # -------------------------------------------------
@@ -181,7 +213,10 @@ async def upload_image(
 async def colorize(
     request: Request,
     file: UploadFile = File(...),
-    x_firebase_appcheck: str = Header(None)
+    x_firebase_appcheck: str = Header(None),
+    user_id: Optional[str] = Form(None),
+    category_id: Optional[str] = Form(None),
+    categoryId: Optional[str] = Form(None),
 ):
     import time
     start_time = time.time()
@@ -189,6 +224,8 @@ async def colorize(
     verify_app_check_token(x_firebase_appcheck)
     
     ip_address = request.client.host if request.client else None
+    effective_user_id = _resolve_user_id(request, user_id)
+    effective_category_id = category_id or categoryId
 
     if not file.content_type.startswith("image/"):
         log_api_call(
@@ -225,6 +262,7 @@ async def colorize(
         result_id=result_id_clean,
         model_type="gan",
         processing_time=processing_time,
+        user_id=effective_user_id,
         ip_address=ip_address
     )
     
@@ -234,9 +272,17 @@ async def colorize(
         status_code=200,
         request_data={"filename": file.filename, "content_type": file.content_type},
         response_data=response_data,
+        user_id=effective_user_id,
         ip_address=ip_address
     )
 
+    log_media_click(
+        user_id=effective_user_id,
+        category_id=effective_category_id,
+        endpoint_path=str(request.url.path),
+        default_category_id=MEDIA_CLICK_DEFAULT_CATEGORY,
+    )
+
     return response_data
 
 # -------------------------------------------------

app/main_fastai.py

@@ -18,7 +18,7 @@ import logging
 from pathlib import Path
 from typing import Optional
 
-from fastapi import FastAPI, UploadFile, File, HTTPException, Depends, Request
+from fastapi import FastAPI, UploadFile, File, HTTPException, Depends, Request, Form
 from fastapi.responses import FileResponse, JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
@@ -37,7 +37,14 @@ from huggingface_hub import from_pretrained_fastai
 
 from app.config import settings
 from app.pytorch_colorizer import PyTorchColorizer
-from app.database import get_database, log_api_call, log_image_upload, log_colorization, close_connection
+from app.database import (
+    get_database,
+    log_api_call,
+    log_image_upload,
+    log_colorization,
+    log_media_click,
+    close_connection,
+)
 
 # Configure logging
 logging.basicConfig(
@@ -200,6 +207,11 @@ async def verify_request(request: Request):
     # Neither token required nor provided → allow (App Check disabled)
     return True
 
+
+def _resolve_user_id(request: Request, supplied_user_id: Optional[str]) -> Optional[str]:
+    """Return supplied user_id if provided, otherwise None (will auto-generate in log_media_click)."""
+    return supplied_user_id
+
 @app.get("/api")
 async def api_info(request: Request):
     """API info endpoint"""
@@ -357,6 +369,9 @@ def colorize_pil(image: Image.Image) -> Image.Image:
 async def colorize_api(
     request: Request,
     file: UploadFile = File(...),
+    user_id: Optional[str] = Form(None),
+    category_id: Optional[str] = Form(None),
+    categoryId: Optional[str] = Form(None),
     verified: bool = Depends(verify_request)
 ):
     """
@@ -366,10 +381,9 @@ async def colorize_api(
     import time
     start_time = time.time()
     
-    user_id = None
-    if hasattr(request, 'state') and hasattr(request.state, 'user'):
-        user_id = request.state.user.get("uid")
-    
+    effective_user_id = _resolve_user_id(request, user_id)
+    effective_category_id = category_id or categoryId
+ 
     ip_address = request.client.host if request.client else None
     
     # Allow fallback colorization even if model isn't loaded
@@ -382,7 +396,7 @@ async def colorize_api(
             method="POST",
             status_code=400,
             error="File must be an image",
-            user_id=user_id,
+            user_id=effective_user_id,
             ip_address=ip_address
         )
         raise HTTPException(status_code=400, detail="File must be an image")
@@ -409,7 +423,7 @@ async def colorize_api(
             result_id=result_id,
             model_type=model_type,
             processing_time=processing_time,
-            user_id=user_id,
+            user_id=effective_user_id,
             ip_address=ip_address
         )
         
@@ -419,9 +433,17 @@ async def colorize_api(
             status_code=200,
             request_data={"filename": file.filename, "content_type": file.content_type},
             response_data={"result_id": result_id, "filename": output_filename},
-            user_id=user_id,
+            user_id=effective_user_id,
             ip_address=ip_address
         )
+
+        # Best-effort media click tracking (admin DB)
+        log_media_click(
+            user_id=effective_user_id,
+            category_id=effective_category_id,
+            endpoint_path=str(request.url.path),
+            default_category_id=settings.DEFAULT_CATEGORY_FALLBACK,
+        )
         
         # Return the image file
         return FileResponse(
@@ -437,7 +459,7 @@ async def colorize_api(
             method="POST",
             status_code=500,
             error=error_msg,
-            user_id=user_id,
+            user_id=effective_user_id,
             ip_address=ip_address
         )
         raise HTTPException(status_code=500, detail=f"Error colorizing image: {error_msg}")

app/main_sdxl.py

@@ -9,7 +9,7 @@ import logging
 from pathlib import Path
 from typing import Optional, Tuple
 
-from fastapi import FastAPI, UploadFile, File, HTTPException, Depends, Request, Body
+from fastapi import FastAPI, UploadFile, File, HTTPException, Depends, Request, Body, Form 
 from fastapi.responses import FileResponse, JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
@@ -25,7 +25,14 @@ from pydantic import BaseModel, EmailStr
 from huggingface_hub import InferenceClient
 
 from app.config import settings
-from app.database import get_database, log_api_call, log_image_upload, log_colorization, close_connection
+from app.database import ( 
+    get_database, 
+    log_api_call, 
+    log_image_upload, 
+    log_colorization, 
+    log_media_click, 
+    close_connection, 
+) 
 
 # Configure logging
 logging.basicConfig(
@@ -300,6 +307,11 @@ async def verify_request(request: Request):
     return True
 
 
+def _resolve_user_id(request: Request, supplied_user_id: Optional[str]) -> Optional[str]: 
+    """Return supplied user_id if provided, otherwise None (will auto-generate in log_media_click).""" 
+    return supplied_user_id 
+
+
 # ========== Auth Endpoints ==========
 
 @app.post("/auth/register", response_model=TokenResponse)
@@ -626,16 +638,18 @@ def colorize_image_sdxl(
 async def upload_image(
     request: Request,
     file: UploadFile = File(...),
+    user_id: Optional[str] = Form(None), 
+    category_id: Optional[str] = Form(None),
+    categoryId: Optional[str] = Form(None),
     verified: bool = Depends(verify_request)
 ):
     """
     Upload an image and get the uploaded image URL.
     Requires Firebase App Check authentication.
     """
-    user_id = None
-    if hasattr(request, 'state') and hasattr(request.state, 'user'):
-        user_id = request.state.user.get("uid")
-    
+    effective_user_id = _resolve_user_id(request, user_id)
+    effective_category_id = category_id or categoryId 
+ 
     ip_address = request.client.host if request.client else None
     
     if not file.content_type or not file.content_type.startswith("image/"):
@@ -644,7 +658,7 @@ async def upload_image(
             method="POST",
             status_code=400,
             error="File must be an image",
-            user_id=user_id,
+            user_id=effective_user_id, 
             ip_address=ip_address
         )
         raise HTTPException(status_code=400, detail="File must be an image")
@@ -682,7 +696,7 @@ async def upload_image(
             filename=file.filename or image_id,
             file_size=file_size,
             content_type=file.content_type or "image/jpeg",
-            user_id=user_id,
+            user_id=effective_user_id, 
             ip_address=ip_address
         )
         
@@ -692,9 +706,17 @@ async def upload_image(
             status_code=200,
             request_data={"filename": file.filename, "content_type": file.content_type},
             response_data=response_data,
-            user_id=user_id,
+            user_id=effective_user_id, 
             ip_address=ip_address
         )
+
+        # Best-effort media click tracking (admin DB) 
+        log_media_click( 
+            user_id=effective_user_id, 
+            category_id=effective_category_id, 
+            endpoint_path=str(request.url.path), 
+            default_category_id=settings.DEFAULT_CATEGORY_FALLBACK, 
+        ) 
         
         return JSONResponse(response_data)
     except Exception as e:
@@ -705,7 +727,7 @@ async def upload_image(
             method="POST",
             status_code=500,
             error=error_msg,
-            user_id=user_id,
+            user_id=effective_user_id, 
             ip_address=ip_address
         )
         raise HTTPException(status_code=500, detail=f"Error uploading image: {error_msg}")
@@ -719,6 +741,9 @@ async def colorize_api(
     negative_prompt: Optional[str] = None,
     seed: int = 123,
     num_inference_steps: int = 8,
+    user_id: Optional[str] = Form(None), 
+    category_id: Optional[str] = Form(None),
+    categoryId: Optional[str] = Form(None),
     verified: bool = Depends(verify_request)
 ):
     """
@@ -728,10 +753,9 @@ async def colorize_api(
     import time
     start_time = time.time()
     
-    user_id = None
-    if hasattr(request, 'state') and hasattr(request.state, 'user'):
-        user_id = request.state.user.get("uid")
-    
+    effective_user_id = _resolve_user_id(request, user_id)
+    effective_category_id = category_id or categoryId 
+ 
     ip_address = request.client.host if request.client else None
     
     if inference_client is None:
@@ -740,7 +764,7 @@ async def colorize_api(
             method="POST",
             status_code=503,
             error="Inference API client not initialized",
-            user_id=user_id,
+            user_id=effective_user_id, 
             ip_address=ip_address
         )
         raise HTTPException(status_code=503, detail="Inference API client not initialized")
@@ -751,7 +775,7 @@ async def colorize_api(
             method="POST",
             status_code=400,
             error="File must be an image",
-            user_id=user_id,
+            user_id=effective_user_id, 
             ip_address=ip_address
         )
         raise HTTPException(status_code=400, detail="File must be an image")
@@ -799,7 +823,7 @@ async def colorize_api(
             prompt=positive_prompt,
             model_type="sdxl",
             processing_time=processing_time,
-            user_id=user_id,
+            user_id=effective_user_id, 
             ip_address=ip_address
         )
         
@@ -815,9 +839,17 @@ async def colorize_api(
                 "num_inference_steps": num_inference_steps
             },
             response_data=response_data,
-            user_id=user_id,
+            user_id=effective_user_id, 
             ip_address=ip_address
         )
+
+        # Best-effort media click tracking (admin DB) 
+        log_media_click( 
+            user_id=effective_user_id, 
+            category_id=effective_category_id, 
+            endpoint_path=str(request.url.path), 
+            default_category_id=settings.DEFAULT_CATEGORY_FALLBACK, 
+        ) 
         
         return JSONResponse(response_data)
     except Exception as e:
@@ -828,7 +860,7 @@ async def colorize_api(
             method="POST",
             status_code=500,
             error=error_msg,
-            user_id=user_id,
+            user_id=effective_user_id, 
             ip_address=ip_address
         )
         raise HTTPException(status_code=500, detail=f"Error colorizing image: {error_msg}")

postman_collection.json

@@ -51,6 +51,11 @@
 						"key": "Authorization",
 						"value": "Bearer {{firebase_token}}",
 						"type": "text"
+					},
+					{
+						"key": "X-Firebase-AppCheck",
+						"value": "{{firebase_app_check}}",
+						"type": "text"
 					}
 				],
 				"body": {
@@ -61,6 +66,18 @@
 							"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"
 						}
 					]
 				},
@@ -86,6 +103,11 @@
 						"key": "Authorization",
 						"value": "Bearer {{firebase_token}}",
 						"type": "text"
+					},
+					{
+						"key": "X-Firebase-AppCheck",
+						"value": "{{firebase_app_check}}",
+						"type": "text"
 					}
 				],
 				"body": {
@@ -102,6 +124,18 @@
 							"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"
 						}
 					]
 				},
@@ -127,6 +161,11 @@
 						"key": "Authorization",
 						"value": "Bearer {{firebase_token}}",
 						"type": "text"
+					},
+					{
+						"key": "X-Firebase-AppCheck",
+						"value": "{{firebase_app_check}}",
+						"type": "text"
 					}
 				],
 				"body": {
@@ -161,6 +200,18 @@
 							"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"
 						}
 					]
 				},