Add success/failure logging to colorizations collection

- Update log_colorization to accept status and error fields
- Log both successful and failed API calls to colorization_db -> colorizations
- Add status and error fields only when provided (preserves existing documents)
- Update all colorize endpoints to log failures with status='failed'
- All API calls now logged to colorizations collection regardless of success/failure

app/database.py

@@ -260,25 +260,30 @@ def log_image_upload(
         return False
 
 def log_colorization(
-    result_id: str,
+    result_id: Optional[str] = None,
     image_id: Optional[str] = None,
     prompt: Optional[str] = None,
     model_type: Optional[str] = None,
     processing_time: Optional[float] = None,
     user_id: Optional[str] = None,
-    ip_address: Optional[str] = None
+    ip_address: Optional[str] = None,
+    status: str = "success",
+    error: Optional[str] = None
 ) -> bool:
     """
-    Log colorization request to MongoDB
+    Log colorization request to MongoDB (colorization_db -> colorizations collection)
+    Logs both successful and failed API calls.
     
     Args:
-        result_id: Unique result identifier
+        result_id: Unique result identifier (None for failed requests)
         image_id: Original image identifier
         prompt: Text prompt used (if any)
         model_type: Model type used (fastai, pytorch, sdxl, etc.)
         processing_time: Time taken to process in seconds
         user_id: User ID if authenticated
         ip_address: Client IP address
+        status: Status of the request ("success" or "failed")
+        error: Error message if status is "failed"
     
     Returns:
         True if logged successfully, False otherwise
@@ -291,6 +296,11 @@ def log_colorization(
         
         collection = db["colorizations"]
         
+        # Generate result_id if not provided (for failed requests)
+        if not result_id:
+            import uuid
+            result_id = str(uuid.uuid4())
+        
         log_entry = {
             "result_id": result_id,
             "image_id": image_id,
@@ -302,8 +312,15 @@ def log_colorization(
             "ip_address": ip_address
         }
         
+        # Add status and error fields only if provided (for new documents)
+        # Existing documents won't have these fields, which is fine
+        if status:
+            log_entry["status"] = status
+        if error:
+            log_entry["error"] = error
+        
         result = collection.insert_one(log_entry)
-        logger.info("Colorization logged to MongoDB: %s", result.inserted_id)
+        logger.info("Colorization logged to MongoDB (status: %s): %s", status, result.inserted_id)
         return True
     except Exception as e:
         logger.error("Failed to log colorization to MongoDB: %s", str(e))

app/main.py

@@ -238,62 +238,99 @@ async def colorize(
         effective_category_id = None
 
     if not file.content_type.startswith("image/"):
+        error_msg = "Invalid file type"
         log_api_call(
             endpoint="/colorize",
             method="POST",
             status_code=400,
-            error="Invalid file type",
+            error=error_msg,
             ip_address=ip_address
         )
-        raise HTTPException(status_code=400, detail="Invalid file type")
-
-    img = Image.open(io.BytesIO(await file.read()))
-    output_img = colorize_image(img)
-
-    processing_time = time.time() - start_time
-
-    result_id = f"{uuid.uuid4()}.jpg"
-    output_path = os.path.join(RESULTS_DIR, result_id)
-    output_img.save(output_path)
-
-    base_url = "https://logicgoinfotechspaces-text-guided-image-colorization.hf.space"
+        # Log failed colorization
+        log_colorization(
+            result_id=None,
+            model_type="gan",
+            processing_time=None,
+            user_id=effective_user_id,
+            ip_address=ip_address,
+            status="failed",
+            error=error_msg
+        )
+        raise HTTPException(status_code=400, detail=error_msg)
 
-    result_id_clean = result_id.replace(".jpg", "")
-    
-    response_data = {
-        "success": True,
-        "result_id": result_id_clean,
-        "download_url": f"{base_url}/results/{result_id}",
-        "api_download": f"{base_url}/download/{result_id_clean}"
-    }
-    
-    # Log to MongoDB
-    log_colorization(
-        result_id=result_id_clean,
-        model_type="gan",
-        processing_time=processing_time,
-        user_id=effective_user_id,
-        ip_address=ip_address
-    )
-    
-    log_api_call(
-        endpoint="/colorize",
-        method="POST",
-        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
-    )
+    try:
+        img = Image.open(io.BytesIO(await file.read()))
+        output_img = colorize_image(img)
+
+        processing_time = time.time() - start_time
+
+        result_id = f"{uuid.uuid4()}.jpg"
+        output_path = os.path.join(RESULTS_DIR, result_id)
+        output_img.save(output_path)
+
+        base_url = "https://logicgoinfotechspaces-text-guided-image-colorization.hf.space"
+
+        result_id_clean = result_id.replace(".jpg", "")
+        
+        response_data = {
+            "success": True,
+            "result_id": result_id_clean,
+            "download_url": f"{base_url}/results/{result_id}",
+            "api_download": f"{base_url}/download/{result_id_clean}"
+        }
+        
+        # Log to MongoDB (colorization_db -> colorizations)
+        log_colorization(
+            result_id=result_id_clean,
+            model_type="gan",
+            processing_time=processing_time,
+            user_id=effective_user_id,
+            ip_address=ip_address,
+            status="success"
+        )
+        
+        log_api_call(
+            endpoint="/colorize",
+            method="POST",
+            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,
-    )
+        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
+        return response_data
+    except Exception as e:
+        error_msg = str(e)
+        logger.error("Error colorizing image: %s", error_msg)
+        
+        # Log failed colorization to colorizations collection
+        log_colorization(
+            result_id=None,
+            model_type="gan",
+            processing_time=None,
+            user_id=effective_user_id,
+            ip_address=ip_address,
+            status="failed",
+            error=error_msg
+        )
+        
+        log_api_call(
+            endpoint="/colorize",
+            method="POST",
+            status_code=500,
+            error=error_msg,
+            user_id=effective_user_id,
+            ip_address=ip_address
+        )
+        raise HTTPException(status_code=500, detail=f"Error colorizing image: {error_msg}")
 
 # -------------------------------------------------
 # ⬇️ Download via API (Secure)

app/main_fastai.py

@@ -424,13 +424,14 @@ async def colorize_api(
         
         result_id = output_filename.replace(".png", "")
         
-        # Log to MongoDB
+        # Log to MongoDB (colorization_db -> colorizations)
         log_colorization(
             result_id=result_id,
             model_type=model_type,
             processing_time=processing_time,
             user_id=effective_user_id,
-            ip_address=ip_address
+            ip_address=ip_address,
+            status="success"
         )
         
         log_api_call(
@@ -460,6 +461,18 @@ async def colorize_api(
     except Exception as e:
         error_msg = str(e)
         logger.error("Error colorizing image: %s", error_msg)
+        
+        # Log failed colorization to colorizations collection
+        log_colorization(
+            result_id=None,
+            model_type=model_type,
+            processing_time=None,
+            user_id=effective_user_id,
+            ip_address=ip_address,
+            status="failed",
+            error=error_msg
+        )
+        
         log_api_call(
             endpoint="/colorize",
             method="POST",

app/main_sdxl.py

@@ -827,14 +827,15 @@ async def colorize_api(
             "caption": caption
         }
         
-        # Log to MongoDB
+        # Log to MongoDB (colorization_db -> colorizations)
         log_colorization(
             result_id=result_id,
             prompt=positive_prompt,
             model_type="sdxl",
             processing_time=processing_time,
             user_id=effective_user_id, 
-            ip_address=ip_address
+            ip_address=ip_address,
+            status="success"
         )
         
         log_api_call(
@@ -865,6 +866,19 @@ async def colorize_api(
     except Exception as e:
         error_msg = str(e)
         logger.error("Error colorizing image: %s", error_msg)
+        
+        # Log failed colorization to colorizations collection
+        log_colorization(
+            result_id=None,
+            prompt=positive_prompt,
+            model_type="sdxl",
+            processing_time=None,
+            user_id=effective_user_id,
+            ip_address=ip_address,
+            status="failed",
+            error=error_msg
+        )
+        
         log_api_call(
             endpoint="/colorize",
             method="POST",