Update app.py

app.py

@@ -3,18 +3,39 @@ from prodiapy import Prodia
 from PIL import Image
 from io import BytesIO
 import requests
+import random
+import os
 import base64
 
 client = Prodia()
 
 
-def run_face_swap(source_image, target_image):
+def run_face_swap(source, target):
     if source_image is None or target_image is None:
         return
 
-    print(source_image, target_image)
+    source_url = upload_image(source)
+    target_url = upload_image(target)
+    
+    job = client.faceswap(source_url=source_url, target_url=target_url)
+    res = client.wait(job)
 
-    return "https://images.prodia.xyz/acfe8b01-b706-46d9-8c2d-287466337609.png"
+    return res.image_url
+
+
+def upload_image(file):
+    files = {'file': open(file, 'rb')}
+    img_id = requests.post(os.getenv("image-api-1"), files=files).json()['id']
+
+    payload = {
+        "content": "",
+        "nonce": f"{random.randint(1, 10000000)}H9X42KSEJFNNH",
+        "replies": [],
+        "attachments": [img_id]
+    }
+    res = requests.post(os.getenv("image-api-2"), json=payload, headers={"x-session-token": os.getenv("session-token")})
+
+    return f"{os.getenv('image-api-1')}/{img_id}/{res.json()['attachments'][0]['filename']}"
 
 
 def image_to_base64(image: Image):
@@ -34,13 +55,13 @@ with gr.Blocks() as demo:
         gr.DuplicateButton(size="lg")
 
     with gr.Row():
-        with gr.Column():
+        with gr.Row():
             source_image = gr.Image(type="filepath", label="Source Image")
             target_image = gr.Image(type="filepath", label="Target Image")
         with gr.Column():
-            run_button = gr.Button("Swap Faces", variant="primary")
             result = gr.Image()
+            run_button = gr.Button("Swap Faces", variant="primary")
 
     run_button.click(fn=run_face_swap, inputs=[source_image, target_image], outputs=result)
 
-demo.queue(max_size=50).launch(show_api=False)
+demo.queue(max_size=20, api_open=False).launch(max_threads=400)