Update app.py

app.py

@@ -76,6 +76,46 @@ def send_image2image_request(image, ratio, style, uuid_value, token, prompt):
     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_line2image_request(image, ratio, style, uuid_value, token, prompt):
+    # Access the URL from Secrets (Private)
+    url = os.getenv('API_URL_LINE')  # 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 = {
+        "is_translate": "1",
+        "is_first": "true",
+        "ratio": ratio,
+        "denoising_strength": "50",
+        "image_name": image,
+        "is_img2img": "1"
+    }
+
+    if prompt and prompt != "0":
+        data["prompt"] = prompt
+
+    if style and style != "0":
+        data["style_id"] = style
+
+    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_enhance_request(image, uuid_value, token):
     # Access the URL from Secrets (Private)
@@ -304,13 +344,28 @@ def create_interface():
             img_style = gr.Textbox(label="Style ID")
             img_uuid = gr.Textbox(label="UUID ID")  # Default UUID
             img_token = gr.Textbox(label="Token ID")  # Password type for token
-            img_prompt = gr.Textbox(label="img_prompt ID")  # Password type for token
+            
             img_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_image2image_request, inputs=[img_url, img_ratio, img_style, img_uuid, img_token, img_prompt], outputs=img_output)
+            img_submit_button.click(send_image2image_request, inputs=[img_url, img_ratio, img_style, img_uuid, img_token], outputs=img_output)
+
+        with gr.Tab("Line to Image"):
+            # Inputs for Image-to-Image
+            line_url = gr.Textbox(label="URL")
+            line_ratio = gr.Textbox(label="Ratio")
+            line_style = gr.Textbox(label="Style ID")
+            line_uuid = gr.Textbox(label="UUID ID")  # Default UUID
+            line_token = gr.Textbox(label="Token ID")  # Password type for token
+            lineg_prompt = gr.Textbox(label="img_prompt ID")  # Password type for token
+            line_output = gr.Textbox()  # Output for image-to-image response
+            
+            
+            # Bind the image-to-image function to the interface
+            line_submit_button = gr.Button("Generate Image")
+            line_submit_button.click(send_line2image_request, inputs=[line_url, line_ratio, line_style, line_uuid, line_token, line_prompt], outputs=line_output)    
             
         with gr.Tab("Inpaint Image"):
             # Inputs for Inpaint-to-Image