Update app.py

app.py

@@ -123,6 +123,10 @@ class Prodia:
         response = self._get(f"{self.base}/sdxl/samplers")
         return response.json()
 
+    def face_swap(self):
+        response = self._get(f"{self.base}/faceswap")
+        return response.json()
+
     def _post(self, url, params):
         headers = {
             **self.headers,
@@ -335,8 +339,25 @@ def xl_img2img(input_image, denoising, prompt, negative_prompt, model, steps, sa
 
 
 
+def face_swaper(source, target, progress=gr.Progress()):
+    progress(0, desc="Starting")
+    time.sleep(1.5)
+    progress(0.10, desc="Uploading source image")
+    time.sleep(1)
+    progress(0.15, desc="Uploading target image")
+    time.sleep(1)
+    progress(0.25, desc="Swapping")
 
-
+    result = prodia_client.face_swap({
+        "sourceUrl": source,
+        "targetUrl": target,
+    })
+    
+    progress(0.75, desc="Opening image")
+    job = prodia_client.wait(result)
+    progress(0.99, desc="Sending image")
+    time.sleep(0.5)
+    return [job["imageUrl"]], job["imageUrl"]
 
 css = """
 #generate {
@@ -624,6 +645,15 @@ with gr.Blocks(css=css) as demo:
 
             hf_text_button.click(hf_inference, inputs=[hf_prompt, hf_negative_prompt, hf_model, hf_steps, sampler, hf_cfg_scale, hf_width, hf_height,
                                                hf_seed], outputs=hf_image_output, concurrency_limit=64)
+        with gr.Tab("Face Swap"):
+            with gr.Row():
+                source_image = gr.File(label="Source image", type="filepath", file_types="image", file_count="single", interactive=True)
+                target_image = gr.File(label="Target image", type="filepath", file_types="image", file_count="single", interactive=True)
+            swap_button = gr.Button(value="Swap")
+            with gr.Row():
+                swaped_result = gr.Gallery(show_label=False, rows2, allow_preview=True, preview=True)
+                swaped_past = gr.Textbox(visiable=False, interactive=False)
+            swap_button.click(face_swaper, inputs=[source_image, target_image], outputs=[swaped_result, swaped_past])
         with gr.Tab("Prompt Generator"):
             gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Stable-Diffusion', tokenizer='gpt2')
             with open("ideas.txt", "r") as f: