feat(api): add /remove-pink endpoint - simple one-parameter endpoint that auto-detects pink segments and removes them

api/main.py

@@ -57,6 +57,10 @@ class InpaintRequest(BaseModel):
     passthrough: bool = False  # If True, return the original image unchanged
 
 
+class SimpleRemoveRequest(BaseModel):
+    image_id: str  # Image with pink/magenta segments to remove
+
+
 @app.get("/")
 def root() -> Dict[str, object]:
     return {
@@ -359,6 +363,90 @@ def inpaint_multipart(
     return resp
 
 
+@app.post("/remove-pink")
+def remove_pink_segments(
+    image: UploadFile = File(...),
+    request: Request = None,
+    _: None = Depends(bearer_auth),
+) -> Dict[str, str]:
+    """
+    Simple endpoint: just upload an image with pink/magenta segments you want to remove.
+    The API automatically detects pink areas and removes them, keeping everything else.
+    """
+    log.info(f"Simple remove-pink: processing image {image.filename}")
+    
+    # Load the image (with pink paint on it)
+    img = Image.open(image.file).convert("RGBA")
+    img_rgb = cv2.cvtColor(np.array(img), cv2.COLOR_RGBA2RGB)
+    
+    # Auto-detect pink/magenta segments
+    magenta_detected = (
+        (img_rgb[:, :, 0] > 240) &  # Red: high
+        (img_rgb[:, :, 1] < 30) &   # Green: low
+        (img_rgb[:, :, 2] > 240)    # Blue: high
+    ).astype(np.uint8) * 255
+    
+    # Clean up the mask
+    kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
+    binmask = cv2.morphologyEx(magenta_detected, cv2.MORPH_CLOSE, kernel, iterations=2)
+    binmask = cv2.morphologyEx(binmask, cv2.MORPH_OPEN, kernel, iterations=1)
+    
+    nonzero = int((binmask > 0).sum())
+    log.info(f"Detected {nonzero} pink pixels to remove")
+    
+    if nonzero < 50:
+        log.warning("Very few pink pixels detected! Trying strict magenta detection...")
+        magenta_strict = np.all(img_rgb == [255, 0, 255], axis=2).astype(np.uint8) * 255
+        binmask = cv2.morphologyEx(magenta_strict, cv2.MORPH_CLOSE, kernel, iterations=3)
+        nonzero = int((binmask > 0).sum())
+        log.info(f"Strict detection: {nonzero} pixels")
+    
+    if nonzero < 50:
+        log.error("No pink segments detected! Returning original image.")
+        result = np.array(img.convert("RGB"))
+        result_name = f"output_{uuid.uuid4().hex}.png"
+        result_path = os.path.join(OUTPUT_DIR, result_name)
+        Image.fromarray(result).save(result_path)
+        return {
+            "result": result_name,
+            "error": "No pink/magenta segments detected. Please paint areas to remove with magenta/pink color (RGB 255,0,255)."
+        }
+    
+    # Create mask: white = pink areas (remove), black = everything else (keep)
+    mask_rgba = np.zeros((binmask.shape[0], binmask.shape[1], 4), dtype=np.uint8)
+    mask_rgba[:, :, 0] = binmask  # R
+    mask_rgba[:, :, 1] = binmask  # G
+    mask_rgba[:, :, 2] = binmask  # B
+    mask_rgba[:, :, 3] = 255      # Alpha
+    
+    log.info(f"Created mask: {nonzero} white pixels (remove), {binmask.shape[0]*binmask.shape[1]-nonzero} black pixels (keep)")
+    
+    # Process with invert_mask=False because white pixels = remove (standard)
+    result = process_inpaint(np.array(img), mask_rgba, invert_mask=False)
+    result_name = f"output_{uuid.uuid4().hex}.png"
+    result_path = os.path.join(OUTPUT_DIR, result_name)
+    Image.fromarray(result).save(result_path)
+    
+    url: Optional[str] = None
+    try:
+        if request is not None:
+            url = str(request.url_for("download_file", filename=result_name))
+    except Exception:
+        url = None
+    
+    logs.append({
+        "result": result_name,
+        "filename": image.filename,
+        "pink_pixels": nonzero,
+        "timestamp": datetime.utcnow().isoformat()
+    })
+    
+    resp: Dict[str, str] = {"result": result_name, "pink_segments_detected": str(nonzero)}
+    if url:
+        resp["url"] = url
+    return resp
+
+
 @app.get("/download/{filename}")
 def download_file(filename: str):
     path = os.path.join(OUTPUT_DIR, filename)