Upload 2 files

app.py

@@ -0,0 +1,46 @@
+import gradio as gr
+from prodiapy import Prodia
+from PIL import Image
+from io import BytesIO
+import requests
+import base64
+
+client = Prodia()
+
+
+def run_face_swap(source_image, target_image):
+    if source_image is None or target_image is None:
+        return
+
+    print(source_image, target_image)
+
+    return "https://images.prodia.xyz/acfe8b01-b706-46d9-8c2d-287466337609.png"
+
+
+def image_to_base64(image: Image):
+    # Convert the image to bytes
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")  # You can change format to PNG if needed
+
+    # Encode the bytes to base64
+    img_str = base64.b64encode(buffered.getvalue())
+
+    return img_str.decode('utf-8')  # Convert bytes to string
+
+
+with gr.Blocks() as demo:
+    with gr.Column():
+        gr.HTML("<h1><center>Face Swap</center></h1>")
+        gr.DuplicateButton(size="lg")
+
+    with gr.Row():
+        with gr.Column():
+            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.click(fn=run_face_swap, inputs=[source_image, target_image], outputs=result)
+
+demo.launch(show_api=False)

requirements.txt

@@ -0,0 +1,5 @@
+numpy
+gradio~=4.28.3
+pillow~=10.2.0
+prodiapy==5.1.2
+requests~=2.31.0