add: swap image endpoint

mannequin_to_model.py

@@ -76,6 +76,25 @@ async def mannequinToModel(
     return response
 
 
+@secure_router.get("/swap_image")
+async def swapImage(image_one: UploadFile = File(...), ref_face: UploadFile = File(...)):
+    image_one, ref_face = await image_one.read(), await ref_face.read()
+    image_one_, ref_face_ = Image.open(BytesIO(image_one)), Image.open(BytesIO(ref_face))
+
+    mannequin_image = cv2.cvtColor(np.array(image_one_), cv2.COLOR_RGB2BGR)
+    person_image = cv2.cvtColor(np.array(ref_face_), cv2.COLOR_RGB2BGR)
+    result = pipeline.face_swap(mannequin_image, person_image)
+    result = Image.fromarray(cv2.cvtColor(result, cv2.COLOR_BGR2RGB))
+    inMemFile = BytesIO()
+    result.save(inMemFile, format="WEBP", quality=85)
+    outputBytes = inMemFile.getvalue()
+    response = {
+        "code": 200,
+        "output": f"data:image/WEBP;base64,{base64.b64encode(outputBytes).decode('utf-8')}",
+    }
+    return response
+
+
 @secure_router.get("/mannequin_catalogue")
 async def returnJsonData(gender: str):
     folderImageURL = supabase.storage.get_bucket("JSON").create_signed_url(