feat(api): add passthrough flag to return original image unchanged for /inpaint and /inpaint-multipart

api/main.py

@@ -54,6 +54,7 @@ class InpaintRequest(BaseModel):
     image_id: str
     mask_id: str
     invert_mask: bool = True  # True => selected/painted area is removed
+    passthrough: bool = False  # If True, return the original image unchanged
 
 
 @app.get("/")
@@ -147,7 +148,10 @@ def inpaint(req: InpaintRequest, _: None = Depends(bearer_auth)) -> Dict[str, st
     mask_img = Image.open(file_store[req.mask_id]["path"])  # may be RGB/gray/RGBA
     mask_rgba = _load_rgba_mask_from_image(mask_img)
 
-    result = process_inpaint(np.array(img_rgba), mask_rgba, invert_mask=req.invert_mask)
+    if req.passthrough:
+        result = np.array(img_rgba.convert("RGB"))
+    else:
+        result = process_inpaint(np.array(img_rgba), mask_rgba, invert_mask=req.invert_mask)
     result_name = f"output_{uuid.uuid4().hex}.png"
     result_path = os.path.join(OUTPUT_DIR, result_name)
     Image.fromarray(result).save(result_path)
@@ -168,7 +172,10 @@ def inpaint_url(req: InpaintRequest, request: Request, _: None = Depends(bearer_
     mask_img = Image.open(file_store[req.mask_id]["path"])  # may be RGB/gray/RGBA
     mask_rgba = _load_rgba_mask_from_image(mask_img)
 
-    result = process_inpaint(np.array(img_rgba), mask_rgba, invert_mask=req.invert_mask)
+    if req.passthrough:
+        result = np.array(img_rgba.convert("RGB"))
+    else:
+        result = process_inpaint(np.array(img_rgba), mask_rgba, invert_mask=req.invert_mask)
     result_name = f"output_{uuid.uuid4().hex}.png"
     result_path = os.path.join(OUTPUT_DIR, result_name)
     Image.fromarray(result).save(result_path)
@@ -185,12 +192,36 @@ def inpaint_multipart(
     request: Request = None,
     invert_mask: bool = True,
     mask_is_painted: bool = False,  # if True, mask file is the painted-on image (e.g., black strokes on original)
+    passthrough: bool = False,
     _: None = Depends(bearer_auth),
 ) -> Dict[str, str]:
     # Load in-memory
     img = Image.open(image.file).convert("RGBA")
     m = Image.open(mask.file).convert("RGBA")
 
+    if passthrough:
+        # Just echo the input image, ignore mask
+        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)
+
+        url: Optional[str] = None
+        try:
+            if request is not None:
+                url = str(request.url_for("download_file", filename=result_name))
+        except Exception:
+            url = None
+
+        entry: Dict[str, str] = {"result": result_name, "timestamp": datetime.utcnow().isoformat()}
+        if url:
+            entry["url"] = url
+        logs.append(entry)
+        resp: Dict[str, str] = {"result": result_name}
+        if url:
+            resp["url"] = url
+        return resp
+
     if mask_is_painted:
         # Derive mask by differencing painted image vs original
         img_rgb = cv2.cvtColor(np.array(img), cv2.COLOR_RGBA2RGB)