added

app.py

@@ -127,19 +127,31 @@ api = FastAPI()
 
 @api.post("/api/faceswap")
 async def faceswap(src_img: UploadFile = File(...), tgt_img: UploadFile = File(...)):
-    src = cv2.imdecode(np.frombuffer(await src_img.read(), np.uint8), cv2.IMREAD_COLOR)
-    tgt = cv2.imdecode(np.frombuffer(await tgt_img.read(), np.uint8), cv2.IMREAD_COLOR)
+    """
+    REST API endpoint for face swapping.
+    Accepts source and target images as file uploads.
+    Returns enhanced face-swapped image as PNG.
+    """
+    try:
+        # Read images from uploads
+        src = cv2.imdecode(np.frombuffer(await src_img.read(), np.uint8), cv2.IMREAD_COLOR)
+        tgt = cv2.imdecode(np.frombuffer(await tgt_img.read(), np.uint8), cv2.IMREAD_COLOR)
 
-    result_img, err = face_swap_and_enhance(
-        cv2.cvtColor(src, cv2.COLOR_BGR2RGB),
-        cv2.cvtColor(tgt, cv2.COLOR_BGR2RGB)
-    )
-    if result_img is None:
-        return {"error": err}
+        # Run face swap + CodeFormer enhancement
+        result_img, err = face_swap_and_enhance(
+            cv2.cvtColor(src, cv2.COLOR_BGR2RGB),
+            cv2.cvtColor(tgt, cv2.COLOR_BGR2RGB)
+        )
 
-    _, buffer = cv2.imencode(".png", cv2.cvtColor(result_img, cv2.COLOR_RGB2BGR))
-    return StreamingResponse(io.BytesIO(buffer.tobytes()), media_type="image/png")
+        if result_img is None:
+            return {"error": err}
 
+        # Encode as PNG and return
+        _, buffer = cv2.imencode(".png", cv2.cvtColor(result_img, cv2.COLOR_RGB2BGR))
+        return StreamingResponse(io.BytesIO(buffer.tobytes()), media_type="image/png")
+
+    except Exception as e:
+        return {"error": f"Exception: {str(e)}"}
 # -------------------------------------------------
 # Gradio UI
 # -------------------------------------------------