Update app.py

app.py

@@ -100,7 +100,36 @@ def send_enhance_request(image, 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(image, target, uuid_value, token):
+    # Access the URL from Secrets (Private)
+    url = os.getenv('API_URL_FACE')  # Secret variable for the URL
+    if not url:
+        return "Error: API URL not set in environment secrets."
+
+    # Define the headers with the required values
+    headers = {
+        "Uid": uuid_value,  # Using the provided UUID
+        "Versioncode": "16701",
+        "Accept-Version": "v1",
+        "Token": token,  # Replace with the actual token
+        "Content-Type": "application/x-www-form-urlencoded"
+    }
+
+    # Define the values to be sent in the POST request
+    data = {
+        "sourceUrl": image,
+        "targetUrl": target
+    }
+
+    response = requests.post(url, headers=headers, data=data)
+
+    # Return the response text or any relevant information
+    if response.status_code == 200:
+        return response.text
+    else:
+        return f"Error: {response.status_code} - {response.text}"         
 
 # 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):
@@ -262,6 +291,18 @@ def create_interface():
             en_submit_button = gr.Button("Get Upload URL")
             en_submit_button.click(send_enhance_request, inputs=[en_url, en_uuid, en_token], outputs=en_output)    
 
+        with gr.Tab("Get Faceswap"):
+            # Inputs for Get Upload URL
+            fs_url = gr.Textbox(label="URL")
+            fs_target = gr.Textbox(label="Target URL")
+            fs_uuid = gr.Textbox(label="UUID ID")  # Default UUID
+            fs_token = gr.Textbox(label="Token ID") 
+            fs_output = gr.Textbox()  # Output for the upload URL response
+            
+            # 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)        
+
         with gr.Tab("Image Put"):
             # Inputs for Image Put
             image_input = gr.File(label="Upload Image")  # Input for image file