Update app.py

app.py

@@ -8,32 +8,45 @@ from PIL import Image
 import json
 import numpy as np
 import rsa
-import insightface
-from insightface.app import FaceAnalysis
 
 def FaceSwap(source_img, dest_img):
-    
-    swapper = insightface.model_zoo.get_model('./inswapper_128.onnx', download=False, download_zip=False)
 
-    app = FaceAnalysis(name="buffalo_l")
-    app.prepare(ctx_id=0, det_size=(640, 640))
+    secret = os.environ['API_KEY']
+    privateKey = os.environ["PRIVATE_KEY"]
+    publicKey = os.environ["PUBLIC_KEY"]
+    baseurl = os.environ["BASE_URL"]
+
+    n,e,d,p,q = privateKey.split(",")
+    n1,e1 = publicKey.split(",")
+    privateKey = rsa.PrivateKey(int(n),int(e),int(d),int(p),int(q))
+    publicKey = rsa.PublicKey(int(n1), int(e1))
+
+    encMessage = rsa.encrypt(secret.encode('utf-8'), publicKey)
+    encMessage = base64.b64encode(encMessage).decode("utf-8")
+    
+    src_size = source_img.size
+    dest_size = dest_img.size
     
-    s_faces = app.get(source_img)
-    if len(s_faces) == 0:
-        raise gr.Error("No Face Detected in Source Image")
+    src_base64 = base64.b64encode(bytearray(source_img.tobytes())).decode('utf-8')
+    dest_base64 = base64.b64encode(bytearray(dest_img.tobytes())).decode('utf-8')
 
-    d_faces = app.get(dest_img)
-    if len(d_faces) == 0:
-        raise gr.Error("No Face Detected in Destination Image")
+    #Sending request to a FaceSwap API I deployed it on AWS ec2. 
+    #Contact [email protected] to get access to the API.
+    api_url = f"{baseurl}/api/faceswap"
     
-    source_face = s_faces[0]
-    dest_face = d_faces[0]
+    payload = {"source_img": str(src_base64), "dest_img": str(dest_base64), "source_size": src_size, "dest_size": dest_size}
+    headers = {'Content-Type': 'multipart/form-data', "Authorization": encMessage}
+    res = requests.post(api_url, json=payload, headers=headers)
 
-    res = dest_img.copy()
-    res = swapper.get(res, dest_face, source_face, paste_back=True)
-    return res
+    if (res.status_code >= 200 and res.status_code <= 299):
+        res = np.array(res.json())
+        return res
+    elif (res.status_code >= 400 and res.status_code <= 499):
+        raise gr.Error(f"Code {res.status_code}: Erron on client")
+    else:
+        raise gr.Error(f"Code {res.status_code}: Erron on server. Try after some time")
     
 if __name__ == "__main__":
     with gr.Blocks(title="Face Swap | Meet With Your new Personality") as demo:
-        gr.Interface(fn=FaceSwap, inputs=["image","image"], outputs=["image"], allow_flagging="never")
+        gr.Interface(fn=FaceSwap, inputs=[gr.Image(type="pil"), gr.Image(type="pil")], outputs=["image"], allow_flagging="never")
     demo.launch(show_api=False, favicon_path="./favicon.ico")