Update app.py

app.py

@@ -74,7 +74,7 @@ def send_image2image_request(image, ratio, style, uuid_value, token):
         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):
+def send_inpaint2image_request(image, mask, prompt, ratio, uuid_value, token, style):
     # Access the URL from Secrets (Private)
     url = os.getenv('API_URL_IMAGE')  # Secret variable for the URL
     if not url:
@@ -200,17 +200,18 @@ def create_interface():
             
         with gr.Tab("Inpaint Image"):
             # Inputs for Inpaint-to-Image
-            img_url = gr.Textbox(label="URL")
-            img_mask = gr.Textbox(label="Mask URL")
-            img_prompt = gr.Textbox(label="Prompt")
-            img_ratio = gr.Textbox(label="Ratio")
-            img_uuid = gr.Textbox(label="UUID ID")  # Default UUID
-            img_token = gr.Textbox(label="Token ID")  # Password type for token
-            img_output = gr.Textbox()  # Output for image-to-image response
+            in_url = gr.Textbox(label="URL")
+            in_mask = gr.Textbox(label="Mask URL")
+            in_prompt = gr.Textbox(label="Prompt")
+            in_ratio = gr.Textbox(label="Ratio")
+            in_uuid = gr.Textbox(label="UUID ID")  # Default UUID
+            in_token = gr.Textbox(label="Token ID") 
+            in_style = gr.Textbox(label="Style ID") # Password type for token
+            in_output = gr.Textbox()  # Output for image-to-image response
             
             # Bind the image-to-image function to the interface
-            img_submit_button = gr.Button("Generate Image")
-            img_submit_button.click(send_inpaint2image_request, inputs=[img_url, img_mask, img_prompt, img_ratio, img_uuid, img_token], outputs=img_output)    
+            in_submit_button = gr.Button("Generate Image")
+            in_submit_button.click(send_inpaint2image_request, inputs=[in_url, in_mask, in_prompt, in_ratio, in_uuid, in_token, in_style], outputs=in_output)    
         
         with gr.Tab("Get Upload URL"):
             # Inputs for Get Upload URL