Update app.py

app.py

@@ -6,10 +6,9 @@ from io import BytesIO
 import os
 import random
 
-def generate_image(prompt, negative_prompt, scheduler, steps, width, height, restore_faces, seed, cfg):
+def generate_image(prompt, negative_prompt, scheduler, steps, width, height, restore_faces, seed, cfg, token):
     # Define the API endpoint
-    apiUrl = os.getenv("API_URL")
-    token = os.getenv("API_TOKEN")
+    apiUrl = "https://apiv2.makeai.run/v2/txt2img"
     # Define the request headers
     headers = {
         "Content-Type": "application/json",
@@ -54,13 +53,7 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
         "controlnet_weight": 1,
         "controlnet_guidance_start": 0,
         "controlnet_guidance_end": 1,
-        "controlnet_ref_img_url": "https://upload.wikimedia.org/wikipedia/commons/d/d1/Image_not_available.png",
-        "controlnet_mask": [],
-        "controlnet_resize_mode": "Scale to Fit (Inner Fit)",
-        "controlnet_lowvram": False,
-        "controlnet_processor_res": 512,
-        "controlnet_threshold_a": 100,
-        "controlnet_threshold_b": 200
+        "controlnet_ref_img_url": "",      
     }
 
     # Send the request
@@ -68,18 +61,19 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
 
     # Check the response status
     if response.status_code == 200:
-        # Get the image URL from the response
-        response_json = response.json()
-        if 'results' in response_json and isinstance(response_json['results'], list) and len(response_json['results']) > 0:
-            image_url = response_json['results'][0]
+        
+       # Get the image URL from the response
+       response_json = response.json()
+       if 'results' in response_json and isinstance(response_json['results'], list) and len(response_json['results']) > 0:
+           image_url = response_json['results'][0]
 
-            # Get the image from the URL
-            image_response = requests.get(image_url)
-            image = PIL.Image.open(BytesIO(image_response.content))
+           # Get the image from the URL
+           image_response = requests.get(image_url)
+           image = PIL.Image.open(BytesIO(image_response.content))
 
-            return image
-        else:
-            raise Exception("Unexpected API response format")
+           return image
+       else:
+           raise Exception("Unexpected API response format")
     else:
         raise Exception("API request failed with status code " + str(response.status_code))
 
@@ -89,7 +83,6 @@ iface = gr.Interface(
     inputs=[
         gr.components.Textbox(label="Prompt"),
         gr.components.Textbox(default="ugly, tiling, poorlydrawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, disfigured, deformed, body out of frame, blurry, bad anatomy, blurred, watermark, grainy, signature, cut off, draft", label="Negative Prompt"),
-        gr.components.Slider(minimum=10, maximum=100, step=1.0,default=30, label="Steps"),
         gr.components.Dropdown(choices=[
             "Euler a",
             "Euler",
@@ -111,11 +104,13 @@ iface = gr.Interface(
             "DDIM",
             "PLMS"
         ], label="Scheduler", default="Euler a"),
+        gr.components.Slider(minimum=10, maximum=100, step=1.0,default=30, label="Steps"),
         gr.components.Slider(minimum=512, maximum=1600, default=768, label="Width"),
         gr.components.Slider(minimum=512, maximum=1600, default=768, label="Height"),
         gr.components.Slider(minimum=4, maximum=12, step=0.5, default=7.0, label="CFG"),
         gr.components.Number(label="Seed", default=None),
         gr.components.Checkbox(label="Restore Faces"),
+        gr.components.Textbox(label="Token", type="password")
     ], 
     outputs=gr.components.Image(),
     title="Freedom Demonstration",