feat(api): add /inpaint-url endpoint returning public download URL; docs updated

README.md

@@ -19,12 +19,13 @@ Endpoints:
 - `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}) → returns result directly (no download step)
+- `POST /inpaint-url` (JSON: {image_id, mask_id}) → {"result":"output_xxx.png","url":"https://.../download/output_xxx.png"}
 - `POST /inpaint-multipart` (form-data: image=file, mask=file) → {"result":"output_xxx.png"}
-- `GET /download/{filename}` → image file (optional; not required when using ID-based inpaint)
+- `GET /download/{filename}` → image file (public; optional for 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.
+- `POST /inpaint` returns image bytes directly. Use `POST /inpaint-url` if you need a shareable URL in the response.
 
 Local run:
 - Install deps: `python3 -m pip install -r requirements.txt`

api/main.py

@@ -5,7 +5,7 @@ from datetime import datetime
 from typing import Dict, List, Optional
 
 import numpy as np
-from fastapi import FastAPI, UploadFile, File, HTTPException, Depends, Header
+from fastapi import FastAPI, UploadFile, File, HTTPException, Depends, Header, Request
 from fastapi.responses import FileResponse, JSONResponse
 from pydantic import BaseModel
 from PIL import Image
@@ -148,10 +148,33 @@ def inpaint(req: InpaintRequest, _: None = Depends(bearer_auth)):
     return FileResponse(result_path, media_type="image/png", filename=result_name)
 
 
+@app.post("/inpaint-url")
+def inpaint_url(req: InpaintRequest, request: Request, _: None = Depends(bearer_auth)) -> Dict[str, str]:
+    """Same as /inpaint but returns a JSON with a public download URL instead of image bytes."""
+    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":
+        raise HTTPException(status_code=404, detail="mask_id not found")
+
+    img_rgba = _load_rgba_image(file_store[req.image_id]["path"])
+    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)
+    result_name = f"output_{uuid.uuid4().hex}.png"
+    result_path = os.path.join(OUTPUT_DIR, result_name)
+    Image.fromarray(result).save(result_path)
+
+    url = str(request.url_for("download_file", filename=result_name))
+    logs.append({"result": result_name, "url": url, "timestamp": datetime.utcnow().isoformat()})
+    return {"result": result_name, "url": url}
+
+
 @app.post("/inpaint-multipart")
 def inpaint_multipart(
     image: UploadFile = File(...),
     mask: UploadFile = File(...),
+    request: Request = None,
     _: None = Depends(bearer_auth),
 ) -> Dict[str, str]:
     # Load in-memory
@@ -164,8 +187,21 @@ def inpaint_multipart(
     result_path = os.path.join(OUTPUT_DIR, result_name)
     Image.fromarray(result).save(result_path)
 
-    logs.append({"result": result_name, "timestamp": datetime.utcnow().isoformat()})
-    return {"result": result_name}
+    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
 
 
 @app.get("/download/{filename}")