feat(api): return image directly from /inpaint; docs: mark download optional for ID-based flow

README.md

@@ -18,11 +18,14 @@ Endpoints:
 - `GET /health` → {"status":"healthy"}
 - `POST /upload-image` (form-data: image=file) → {"id":"<image_id>","filename":"name.png"}
 - `POST /upload-mask` (form-data: mask=file) → {"id":"<mask_id>","filename":"mask.png"}
-- `POST /inpaint` (JSON: {image_id, mask_id}) → {"result":"output_xxx.png"}
+- `POST /inpaint` (JSON: {image_id, mask_id}) → returns result directly (no download step)
 - `POST /inpaint-multipart` (form-data: image=file, mask=file) → {"result":"output_xxx.png"}
-- `GET /download/{filename}` → image file
+- `GET /download/{filename}` → image file (optional; not required when using ID-based inpaint)
 - `GET /logs` → recent uploads/results
 
+Note:
+- When using the ID-based `POST /inpaint` endpoint, the API responds with the inpainted image directly, so the `GET /download/{filename}` endpoint is not needed.
+
 Local run:
 - Install deps: `python3 -m pip install -r requirements.txt`
 - Run API: `python3 -m uvicorn api.main:app --host 0.0.0.0 --port 7860`

api/main.py

@@ -129,7 +129,7 @@ def _load_rgba_mask_from_image(img: Image.Image) -> np.ndarray:
 
 
 @app.post("/inpaint")
-def inpaint(req: InpaintRequest, _: None = Depends(bearer_auth)) -> Dict[str, str]:
+def inpaint(req: InpaintRequest, _: None = Depends(bearer_auth)):
     if req.image_id not in file_store or file_store[req.image_id]["type"] != "image":
         raise HTTPException(status_code=404, detail="image_id not found")
     if req.mask_id not in file_store or file_store[req.mask_id]["type"] != "mask":
@@ -145,7 +145,7 @@ def inpaint(req: InpaintRequest, _: None = Depends(bearer_auth)) -> Dict[str, st
     Image.fromarray(result).save(result_path)
 
     logs.append({"result": result_name, "timestamp": datetime.utcnow().isoformat()})
-    return {"result": result_name}
+    return FileResponse(result_path, media_type="image/png", filename=result_name)
 
 
 @app.post("/inpaint-multipart")