Update app.py

app.py

@@ -4,14 +4,14 @@ import json
 import PIL.Image
 from io import BytesIO
 import os
-def generate_image(prompt):
+def generate_image(prompt, negative_prompt, scheduler, steps, width, height, restore_faces):
     # Define the API endpoint
     apiUrl = os.getenv("API_URL")
     token = os.getenv("API_TOKEN")
     # Define the request headers
     headers = {
         "Content-Type": "application/json",
-        "token": token
+        "token": "token"
     }
 
     # Define the request body
@@ -21,20 +21,20 @@ def generate_image(prompt):
         "tiling": False,
         "batch_size": 1,
         "prompt": prompt,
-        "negative_prompt": "worst quality, lowres",
+        "negative_prompt": negative_prompt,
         "seed": 1234,
-        "scheduler": "Euler a",
+        "scheduler": scheduler,
         "n_iter": 1,
-        "steps": 25,
+        "steps": steps,
         "cfg": 11.0,
         "offset_noise": 0.0,
-        "width": 512,
-        "height": 512,
+        "width": width,
+        "height": height,
         "clip_skip": 1,
         "loras": [{"name": "", "strength": 1.0}],
         "embeddings": [{"name": "", "strength": 1.0}],
         "vae": "vae-ft-mse-840000-ema-pruned.ckpt",
-        "restore_faces": False,
+        "restore_faces": restore_faces,
         "fr_model": "CodeFormer",
         "codeformer_weight": 0.5,
         "enable_hr": False,
@@ -80,7 +80,35 @@ def generate_image(prompt):
 # Define the Gradio interface
 iface = gr.Interface(
     fn=generate_image, 
-    inputs="text", 
+    inputs=[
+        gr.inputs.Textbox(label="Prompt"),
+        gr.inputs.Textbox(label="Negative Prompt"),
+        gr.inputs.Dropdown(choices=[
+            "Euler a",
+            "Euler",
+            "LMS",
+            "Heun",
+            "DPM2",
+            "DPM2 a",
+            "DPM++ 2S a",
+            "DPM++ 2M",
+            "DPM++ SDE",
+            "DPM fast",
+            "DPM adaptive",
+            "LMS Karras",
+            "DPM2 Karras",
+            "DPM2 a Karras",
+            "DPM++ 2S a Karras",
+            "DPM++ 2M Karras",
+            "DPM++ SDE Karras",
+            "DDIM",
+            "PLMS"
+        ], label="Scheduler", default="Euler a"),
+        gr.inputs.Slider(minimum=10, maximum=60, default=30, label="Steps"),
+        gr.inputs.Slider(minimum=512, maximum=1500, default=768, label="Width"),
+        gr.inputs.Slider(minimum=512, maximum=1500, default=768, label="Height"),
+        gr.inputs.Checkbox(label="Restore Faces")
+    ], 
     outputs="image",
     title="Freedom Demonstration",
     description = """