Add collage-maker appname support with MongoDB integration

- Add appname parameter to InpaintRequest and all inpaint endpoints
- Add MongoDB helper functions for collage-maker database
- Auto-fetch category_id from collage-maker when appname=collage-maker
- Store media clicks in collage-maker MongoDB when appname=collage-maker
- Store appname in api_logs collection
- Add environment variables: MONGODB_COLLAGE_MAKER, MONGODB_COLLAGE_MAKER_DB_NAME, MONGODB_COLLAGE_MAKER_ADMIN_DB_NAME
- Update all endpoints: /inpaint, /inpaint-url, /inpaint-multipart, /remove-pink

API_Usage_Guide.md

@@ -1,221 +0,0 @@
-# API Usage Guide - Photo Object Removal
-
-## Overview
-This guide provides step-by-step instructions for using the Photo Object Removal API to remove objects from images using AI inpainting.
-
-**Base URL:** `https://logicgoinfotechspaces-object-remover.hf.space`  
-**Authentication:** Bearer token (optional)  
-**Storage:** Uploaded images/masks are saved in MongoDB GridFS (database `object_remover`); IDs returned by upload endpoints are pulled from GridFS before sending to Gemini.
-- Processing is delegated to Google Gemini/Imagen edit API; only lightweight CPU work (mask prep, file IO) happens on this server.
-
-## Quick Start
-
-### 1. Health Check
-Verify the API is running:
-```bash
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  https://logicgoinfotechspaces-object-remover.hf.space/health
-```
-**Response:** `{"status":"healthy"}`
-
-### 2. Upload Image
-Upload the image you want to edit:
-```bash
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  -F image=@your_image.jpg \
-  https://logicgoinfotechspaces-object-remover.hf.space/upload-image
-```
-**Response:** `{"id":"9cf61445-f83b-4c97-9272-c81647f90d68","filename":"your_image.jpg"}`
-
-### 3. Upload Mask
-Upload a mask showing areas to remove (white/colored areas = remove):
-```bash
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  -F mask=@mask.png \
-  https://logicgoinfotechspaces-object-remover.hf.space/upload-mask
-```
-**Response:** `{"id":"d044a390-dde2-408a-b7cf-d508385e56ed","filename":"mask.png"}`
-
-### 4. Process Image
-Remove objects using the uploaded image and mask:
-```bash
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  -H "Content-Type: application/json" \
-  -d '{"image_id":"9cf61445-f83b-4c97-9272-c81647f90d68","mask_id":"d044a390-dde2-408a-b7cf-d508385e56ed","prompt":"Describe what should be removed"}' \
-  https://logicgoinfotechspaces-object-remover.hf.space/inpaint
-```
-**Response:** `{"result":"output_b09568698bbd4aa591b1598c01f2f745.png"}`
-
-### 5. View Result
-Open the result image in your browser:
-```
-https://logicgoinfotechspaces-object-remover.hf.space/result/output_b09568698bbd4aa591b1598c01f2f745.png
-```
-
-## Alternative Methods
-
-### Method 1: Get Direct URL
-Use `/inpaint-url` to get a shareable URL:
-```bash
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  -H "Content-Type: application/json" \
-  -d '{"image_id":"<image_id>","mask_id":"<mask_id>","prompt":"Describe what should be removed"}' \
-  https://logicgoinfotechspaces-object-remover.hf.space/inpaint-url
-```
-**Response:** 
-```json
-{
-  "result":"output_xxx.png",
-  "url":"https://logicgoinfotechspaces-object-remover.hf.space/download/output_xxx.png"
-}
-```
-
-### Method 2: One-Step Processing
-Upload and process in a single request:
-```bash
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  -F image=@image.jpg \
-  -F mask=@mask.jpg \
-  -F prompt="Describe what should be removed" \
-  https://logicgoinfotechspaces-object-remover.hf.space/inpaint-multipart
-```
-
-### Method 3: Download Result
-Download the result file:
-```bash
-curl -L https://logicgoinfotechspaces-object-remover.hf.space/download/output_xxx.png \
-  -o result.png
-```
-
-## Postman Setup
-
-### 1. Upload Image
-- **Method:** POST
-- **URL:** `https://logicgoinfotechspaces-object-remover.hf.space/upload-image`
-- **Headers:** `Authorization: Bearer <API_TOKEN>`
-- **Body:** form-data
-  - Key: `image` (Type: File)
-  - Value: Select your image file
-
-### 2. Upload Mask
-- **Method:** POST
-- **URL:** `https://logicgoinfotechspaces-object-remover.hf.space/upload-mask`
-- **Headers:** `Authorization: Bearer <API_TOKEN>`
-- **Body:** form-data
-  - Key: `mask` (Type: File)
-  - Value: Select your mask file
-
-### 3. Process Image
-- **Method:** POST
-- **URL:** `https://logicgoinfotechspaces-object-remover.hf.space/inpaint`
-- **Headers:** 
-  - `Authorization: Bearer <API_TOKEN>`
-  - `Content-Type: application/json`
-- **Body:** raw JSON
-```json
-{
-  "image_id": "9cf61445-f83b-4c97-9272-c81647f90d68",
-  "mask_id": "d044a390-dde2-408a-b7cf-d508385e56ed",
-  "prompt": "Describe what should be removed"
-}
-```
-
-## Mask Creation Guide
-
-### Mask Formats
-- **RGBA PNG:** Pixels with alpha=0 are treated as areas to remove
-- **RGB/Grayscale:** Pixels with value > 0 are treated as areas to remove
-
-### Creating Masks
-1. **Paint Method:** Use any image editor to paint white/colored areas over objects you want to remove
-2. **Transparency Method:** Create a PNG with transparent areas where you want objects removed
-3. **Grayscale Method:** Create a black image and paint white areas over objects to remove
-
-### Tips
-- Use high contrast (white on black) for best results
-- Make sure mask areas are clearly defined
-- Avoid thin lines or small details in the mask
-
-## Error Handling
-
-### Common Errors
-
-**422 Unprocessable Entity**
-- **Cause:** Missing required fields or invalid file format
-- **Solution:** Check file upload and field names
-
-**404 Not Found**
-- **Cause:** Invalid image_id or mask_id
-- **Solution:** Re-upload files to get fresh IDs
-
-**401 Unauthorized**
-- **Cause:** Missing or invalid API token
-- **Solution:** Add correct Authorization header
-
-**403 Forbidden**
-- **Cause:** Wrong API token
-- **Solution:** Use the correct token you configured
-
-### Troubleshooting
-
-**File Upload Issues in Postman:**
-- Remove warning icons next to filenames
-- Re-select files if warning appears
-- Try different file formats (PNG, JPG)
-- Check file isn't corrupted
-
-**Slow Processing:**
-- Large images take longer to process
-- Typical processing time: 1-3 seconds
-- Wait for completion before checking results
-
-## Complete Workflow Example
-
-```bash
-# 1. Health check
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  https://logicgoinfotechspaces-object-remover.hf.space/health
-
-# 2. Upload image
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  -F image=@photo.jpg \
-  https://logicgoinfotechspaces-object-remover.hf.space/upload-image
-
-# 3. Upload mask
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  -F mask=@mask.png \
-  https://logicgoinfotechspaces-object-remover.hf.space/upload-mask
-
-# 4. Process (replace with actual IDs from steps 2&3)
-curl -H "Authorization: Bearer <API_TOKEN>" \
-  -H "Content-Type: application/json" \
-  -d '{"image_id":"IMAGE_ID","mask_id":"MASK_ID"}' \
-  https://logicgoinfotechspaces-object-remover.hf.space/inpaint
-
-# 5. View result (replace with actual filename from step 4)
-# Open in browser: https://logicgoinfotechspaces-object-remover.hf.space/result/OUTPUT_FILENAME
-```
-
-## API Endpoints Summary
-
-| Method | Endpoint | Description | Auth Required |
-|--------|----------|-------------|---------------|
-| GET | `/health` | Health check | No |
-| GET | `/` | API information | No |
-| POST | `/upload-image` | Upload image file | Optional |
-| POST | `/upload-mask` | Upload mask file | Optional |
-| POST | `/inpaint` | Process with IDs | Optional |
-| POST | `/inpaint-url` | Process with URL response | Optional |
-| POST | `/inpaint-multipart` | One-step processing | Optional |
-| GET | `/download/{filename}` | Download result | No |
-| GET | `/result/{filename}` | View result in browser | No |
-| GET | `/logs` | View processing logs | Optional |
-| GET | `/docs` | API documentation | No |
-
-## Support
-
-For issues or questions:
-1. Check the API documentation at `/docs`
-2. Verify your request format matches examples
-3. Ensure files are properly uploaded
-4. Check authentication token is correct

api/main.py

@@ -98,6 +98,93 @@ ADMIN_MONGO_URI = os.environ.get("MONGODB_ADMIN")
 DEFAULT_CATEGORY_ID = "69368f722e46bd68ae188984"
 admin_media_clicks = None
 
+# Collage-maker MongoDB configuration
+COLLAGE_MAKER_MONGO_URI = os.environ.get("MONGODB_COLLAGE_MAKER")
+COLLAGE_MAKER_DB_NAME = os.environ.get("MONGODB_COLLAGE_MAKER_DB_NAME", "collage-maker")
+COLLAGE_MAKER_ADMIN_DB_NAME = os.environ.get("MONGODB_COLLAGE_MAKER_ADMIN_DB_NAME", "adminPanel")
+collage_maker_client = None
+collage_maker_db = None
+collage_maker_admin_db = None
+collage_maker_media_clicks = None
+collage_maker_categories = None
+
+
+def get_collage_maker_client() -> Optional[MongoClient]:
+    """Get collage-maker MongoDB client."""
+    global collage_maker_client
+    if collage_maker_client is None and COLLAGE_MAKER_MONGO_URI:
+        try:
+            collage_maker_client = MongoClient(COLLAGE_MAKER_MONGO_URI)
+            log.info("Collage-maker MongoDB client initialized")
+        except Exception as err:
+            log.error("Failed to initialize collage-maker MongoDB client: %s", err)
+            collage_maker_client = None
+    return collage_maker_client
+
+
+def get_collage_maker_database() -> Optional[Any]:
+    """Get collage-maker database instance."""
+    global collage_maker_db
+    client = get_collage_maker_client()
+    if client is None:
+        return None
+    if collage_maker_db is None:
+        try:
+            collage_maker_db = client[COLLAGE_MAKER_DB_NAME]
+            log.info("Collage-maker database initialized: %s", COLLAGE_MAKER_DB_NAME)
+        except Exception as err:
+            log.error("Failed to get collage-maker database: %s", err)
+            collage_maker_db = None
+    return collage_maker_db
+
+
+def _init_collage_maker_mongo() -> None:
+    """Initialize collage-maker MongoDB connections."""
+    global collage_maker_admin_db, collage_maker_media_clicks, collage_maker_categories
+    client = get_collage_maker_client()
+    if client is None:
+        log.info("Collage-maker Mongo URI not provided; collage-maker features disabled")
+        return
+    try:
+        collage_maker_admin_db = client[COLLAGE_MAKER_ADMIN_DB_NAME]
+        collage_maker_media_clicks = collage_maker_admin_db["media_clicks"]
+        collage_maker_categories = collage_maker_admin_db["categories"]
+        log.info(
+            "Collage-maker admin initialized: db=%s, media_clicks=%s, categories=%s",
+            COLLAGE_MAKER_ADMIN_DB_NAME,
+            collage_maker_media_clicks.name,
+            collage_maker_categories.name,
+        )
+    except Exception as err:
+        log.error("Failed to init collage-maker admin Mongo: %s", err)
+        collage_maker_admin_db = None
+        collage_maker_media_clicks = None
+        collage_maker_categories = None
+
+
+_init_collage_maker_mongo()
+
+
+def get_category_id_from_collage_maker() -> Optional[str]:
+    """Query category ID from collage-maker categories collection."""
+    if collage_maker_categories is None:
+        log.warning("Collage-maker categories collection not initialized")
+        return None
+    try:
+        # Query the categories collection - you may need to adjust the query based on your schema
+        # This assumes there's a default category or we get the first one
+        category_doc = collage_maker_categories.find_one()
+        if category_doc:
+            category_id = str(category_doc.get("_id", ""))
+            log.info("Found category ID from collage-maker: %s", category_id)
+            return category_id
+        else:
+            log.warning("No categories found in collage-maker collection")
+            return None
+    except Exception as err:
+        log.error("Failed to query collage-maker categories: %s", err)
+        return None
+
 
 def _init_admin_mongo() -> None:
     global admin_media_clicks
@@ -320,6 +407,7 @@ class InpaintRequest(BaseModel):
     prompt: Optional[str] = None  # Optional: describe what to remove
     user_id: Optional[str] = None
     category_id: Optional[str] = None
+    appname: Optional[str] = None  # Optional: app name (e.g., "collage-maker")
 
 
 class SimpleRemoveRequest(BaseModel):
@@ -349,10 +437,23 @@ def _coerce_category_id(category_id: Optional[str]) -> ObjectId:
     return _coerce_object_id(raw_str)
 
 
-def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
-    """Log to admin media_clicks collection only if user_id is provided."""
-    if admin_media_clicks is None:
-        return
+def log_media_click(user_id: Optional[str], category_id: Optional[str], appname: Optional[str] = None) -> None:
+    """Log to admin media_clicks collection only if user_id is provided.
+    
+    If appname='collage-maker', logs to collage-maker MongoDB instead of regular admin MongoDB.
+    """
+    # Determine which media_clicks collection to use
+    target_media_clicks = None
+    if appname == "collage-maker":
+        target_media_clicks = collage_maker_media_clicks
+        if target_media_clicks is None:
+            log.warning("Collage-maker media_clicks not initialized, skipping log")
+            return
+    else:
+        target_media_clicks = admin_media_clicks
+        if target_media_clicks is None:
+            return
+    
     # Only log if user_id is provided (not None/empty)
     if not user_id or not user_id.strip():
         return
@@ -362,14 +463,14 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
         now = datetime.utcnow()
         today = now.date()
 
-        doc = admin_media_clicks.find_one({"userId": user_obj})
+        doc = target_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
-                admin_media_clicks.update_one(
+                target_media_clicks.update_one(
                     {"_id": doc["_id"], "categories.categoryId": category_obj},
                     {
                         "$inc": {
@@ -386,7 +487,7 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
                 )
             else:
                 # New category to existing document: push category, increment ai_edit_complete
-                admin_media_clicks.update_one(
+                target_media_clicks.update_one(
                     {"_id": doc["_id"]},
                     {
                         "$push": {
@@ -407,7 +508,7 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
         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(
+            target_media_clicks.update_one(
                 {"userId": user_obj},
                 {
                     "$setOnInsert": {
@@ -432,15 +533,17 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str]) -> None:
             )
     except Exception as err:
         err_str = str(err)
+        target_name = "collage-maker" if appname == "collage-maker" else "admin"
         if "Unauthorized" in err_str or "not authorized" in err_str.lower():
             log.warning(
-                "Admin media click logging failed (permissions): user lacks read/write on db=%s collection=%s. "
+                "%s media click logging failed (permissions): user lacks read/write on db=%s collection=%s. "
                 "Check MongoDB user permissions.",
-                admin_media_clicks.database.name,
-                admin_media_clicks.name,
+                target_name,
+                target_media_clicks.database.name,
+                target_media_clicks.name,
             )
         else:
-            log.warning("Admin media click logging failed: %s", err)
+            log.warning("%s media click logging failed: %s", target_name, err)
 
 
 @app.get("/")
@@ -612,6 +715,13 @@ def inpaint(req: InpaintRequest, request: Request, _: None = Depends(bearer_auth
     compressed_url = None
 
     try:
+        # Handle appname="collage-maker": get category_id from collage-maker if not provided
+        category_id = req.category_id
+        if req.appname == "collage-maker" and not category_id:
+            category_id = get_category_id_from_collage_maker()
+            if category_id:
+                log.info("Using category_id from collage-maker: %s", category_id)
+
         img_rgba = _load_rgba_image_from_gridfs(req.image_id, "image")
         mask_img = _load_rgba_image_from_gridfs(req.mask_id, "mask")
         mask_rgba = _load_rgba_mask_from_image(mask_img)
@@ -643,7 +753,7 @@ def inpaint(req: InpaintRequest, request: Request, _: None = Depends(bearer_auth
             log.warning("Failed to create compressed image: %s", compress_err)
             compressed_url = None
 
-        log_media_click(req.user_id, req.category_id)
+        log_media_click(req.user_id, category_id, req.appname)
         response = {"result": output_name}
         if compressed_url:
             response["Compressed_Image_URL"] = compressed_url
@@ -668,6 +778,10 @@ def inpaint(req: InpaintRequest, request: Request, _: None = Depends(bearer_auth
             "response_time_ms": response_time_ms
         }
 
+        # Store appname in api_logs if provided
+        if req.appname:
+            log_doc["appname"] = req.appname
+
         if error_msg:
             log_doc["error"] = error_msg
 
@@ -729,6 +843,13 @@ def inpaint_url(req: InpaintRequest, request: Request, _: None = Depends(bearer_
     result_name = None
 
     try:
+        # Handle appname="collage-maker": get category_id from collage-maker if not provided
+        category_id = req.category_id
+        if req.appname == "collage-maker" and not category_id:
+            category_id = get_category_id_from_collage_maker()
+            if category_id:
+                log.info("Using category_id from collage-maker: %s", category_id)
+
         img_rgba = _load_rgba_image_from_gridfs(req.image_id, "image")
         mask_img = _load_rgba_image_from_gridfs(req.mask_id, "mask")  # may be RGB/gray/RGBA
         mask_rgba = _load_rgba_mask_from_image(mask_img)
@@ -748,7 +869,7 @@ def inpaint_url(req: InpaintRequest, request: Request, _: None = Depends(bearer_
 
         url = str(request.url_for("download_file", filename=result_name))
         logs.append({"result": result_name, "url": url, "timestamp": datetime.utcnow().isoformat()})
-        log_media_click(req.user_id, req.category_id)
+        log_media_click(req.user_id, category_id, req.appname)
         return {"result": result_name, "url": url}
     except Exception as e:
         status = "fail"
@@ -767,6 +888,9 @@ def inpaint_url(req: InpaintRequest, request: Request, _: None = Depends(bearer_
             "ts": int(time.time()),
             "response_time_ms": response_time_ms,
         }
+        # Store appname in api_logs if provided
+        if req.appname:
+            log_doc["appname"] = req.appname
         if error_msg:
             log_doc["error"] = error_msg
         if mongo_logs is not None:
@@ -802,6 +926,7 @@ def inpaint_multipart(
     prompt: Optional[str] = Form(None),
     user_id: Optional[str] = Form(None),
     category_id: Optional[str] = Form(None),
+    appname: Optional[str] = Form(None),
     _: None = Depends(bearer_auth),
 ) -> Dict[str, str]:
     start_time = time.time()
@@ -810,6 +935,13 @@ def inpaint_multipart(
     result_name = None
     
     try:
+        # Handle appname="collage-maker": get category_id from collage-maker if not provided
+        final_category_id = category_id
+        if appname == "collage-maker" and not final_category_id:
+            final_category_id = get_category_id_from_collage_maker()
+            if final_category_id:
+                log.info("Using category_id from collage-maker: %s", final_category_id)
+        
         # Load in-memory
         img = Image.open(image.file).convert("RGBA")
         m = Image.open(mask.file).convert("RGBA")
@@ -835,7 +967,7 @@ def inpaint_multipart(
             resp: Dict[str, str] = {"result": result_name}
             if url:
                 resp["url"] = url
-            log_media_click(user_id, category_id)
+            log_media_click(user_id, final_category_id, appname)
             return resp
 
         if mask_is_painted:
@@ -936,7 +1068,7 @@ def inpaint_multipart(
         resp: Dict[str, str] = {"result": result_name}
         if url:
             resp["url"] = url
-        log_media_click(user_id, category_id)
+        log_media_click(user_id, final_category_id, appname)
         return resp
     except Exception as e:
         status = "fail"
@@ -954,6 +1086,9 @@ def inpaint_multipart(
             "ts": int(time.time()),
             "response_time_ms": response_time_ms,
         }
+        # Store appname in api_logs if provided
+        if appname:
+            log_doc["appname"] = appname
         if error_msg:
             log_doc["error"] = error_msg
         if mongo_logs is not None:
@@ -984,6 +1119,7 @@ def remove_pink_segments(
     request: Request = None,
     user_id: Optional[str] = Form(None),
     category_id: Optional[str] = Form(None),
+    appname: Optional[str] = Form(None),
     _: None = Depends(bearer_auth),
 ) -> Dict[str, str]:
     """
@@ -998,6 +1134,13 @@ def remove_pink_segments(
     result_name = None
     
     try:
+        # Handle appname="collage-maker": get category_id from collage-maker if not provided
+        final_category_id = category_id
+        if appname == "collage-maker" and not final_category_id:
+            final_category_id = get_category_id_from_collage_maker()
+            if final_category_id:
+                log.info("Using category_id from collage-maker: %s", final_category_id)
+        
         log.info(f"Simple remove-pink: processing image {image.filename}")
         
         # Load the image (with pink paint on it)
@@ -1094,7 +1237,7 @@ def remove_pink_segments(
         resp: Dict[str, str] = {"result": result_name, "pink_segments_detected": str(nonzero)}
         if url:
             resp["url"] = url
-        log_media_click(user_id, category_id)
+        log_media_click(user_id, final_category_id, appname)
         return resp
     except Exception as e:
         status = "fail"
@@ -1112,6 +1255,9 @@ def remove_pink_segments(
             "ts": int(time.time()),
             "response_time_ms": response_time_ms,
         }
+        # Store appname in api_logs if provided
+        if appname:
+            log_doc["appname"] = appname
         if error_msg:
             log_doc["error"] = error_msg
         if mongo_logs is not None: