Update app.py

app.py

@@ -129,7 +129,27 @@ def send_face_request(image, target, uuid_value, token):
     if response.status_code == 200:
         return response.text
     else:
-        return f"Error: {response.status_code} - {response.text}"         
+        return f"Error: {response.status_code} - {response.text}"
+
+def send_face_request_d(image, target, uuid_value, token):
+    # Construct the URL with the provided UUID value
+    image_url = f"cutpaste_yearbook/result/2024-11-26/rst_58cd61f5-7424-494a-9d19-09d2ca51fed7_{uuid_value}.jpg"
+
+    # Static response JSON with dynamic image URL
+    static_response = {
+        "data": {
+            "image_url": [image_url],
+            "task_id": uuid_value  # Assuming the task_id is the same as the uuid_value
+        },
+        "error": "",
+        "message": "",
+        "next": "",
+        "response": "ok",
+        "status": "200"
+    }
+    
+    # Return the static JSON response as a string
+    return json.dumps(static_response)       
 
 # Function for Image-to-Image (Send POST request with an image and other parameters)
 def send_inpaint2image_request(image, mask, prompt, ratio, uuid_value, token, style):
@@ -301,7 +321,19 @@ def create_interface():
             
             # Bind the get_upload_url function to the interface
             fs_submit_button = gr.Button("Get Upload URL")
-            fs_submit_button.click(send_face_request, inputs=[fs_url, fs_target, fs_uuid, fs_token], outputs=fs_output)        
+            fs_submit_button.click(send_face_request, inputs=[fs_url, fs_target, fs_uuid, fs_token], outputs=fs_output)  
+            
+        with gr.Tab("Demo Faceswap"):
+            # Inputs for Get Upload URL
+            dfs_url = gr.Textbox(label="URL")
+            dfs_target = gr.Textbox(label="Target URL")
+            dfs_uuid = gr.Textbox(label="UUID ID")  # Default UUID
+            dfs_token = gr.Textbox(label="Token ID") 
+            dfs_output = gr.Textbox()  # Output for the upload URL response
+            
+            # Bind the get_upload_url function to the interface
+            dfs_submit_button = gr.Button("Get Upload URL")
+            dfs_submit_button.click(send_face_request_d, inputs=[dfs_url, dfs_target, dfs_uuid, dfs_token], outputs=dfs_output)      
 
         with gr.Tab("Image Put"):
             # Inputs for Image Put