Upload 2 files

app.py

@@ -0,0 +1,41 @@
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import Response
+from retinaface import RetinaFace
+from insightface.app import FaceAnalysis
+from rembg import remove
+from PIL import Image
+import numpy as np
+import io
+
+app = FastAPI()
+
+face_app = FaceAnalysis(name="buffalo_l", providers=["CPUExecutionProvider"])
+face_app.prepare(ctx_id=0)
+
+def resize_to_4x6(img):
+    return img.resize((472, 709), Image.LANCZOS)
+
+@app.post("/process")
+async def process_image(file: UploadFile = File(...)):
+    img_bytes = await file.read()
+    img = Image.open(io.BytesIO(img_bytes)).convert("RGB")
+    np_img = np.array(img)
+
+    faces = face_app.get(np_img)
+    if not faces:
+        return {"error": "No face detected"}
+    face = faces[0]
+    aligned = Image.fromarray(face.normed)
+
+    no_bg = remove(aligned)
+
+    white_bg = Image.new("RGB", no_bg.size, (255, 255, 255))
+    white_bg.paste(no_bg, mask=no_bg.split()[3])
+
+    final_img = resize_to_4x6(white_bg)
+
+    buffer = io.BytesIO()
+    final_img.save(buffer, format="JPEG")
+    buffer.seek(0)
+
+    return Response(buffer.getvalue(), media_type="image/jpeg")

requirements.txt

@@ -0,0 +1,8 @@
+fastapi
+uvicorn
+Pillow
+insightface
+retinaface
+rembg
+numpy
+python-multipart