Update app.py

app.py

@@ -4,8 +4,9 @@ import json
 import PIL.Image
 from io import BytesIO
 import os
+import random
 
-def generate_image(prompt, negative_prompt, scheduler, steps, width, height, restore_faces):
+def generate_image(prompt, negative_prompt, scheduler, steps, width, height, restore_faces, seed, cfg):
     # Define the API endpoint
     apiUrl = os.getenv("API_URL")
     token = os.getenv("API_TOKEN")
@@ -23,11 +24,11 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
         "batch_size": 1,
         "prompt": prompt,
         "negative_prompt": negative_prompt,
-        "seed": 1234,
+        "seed": seed if seed else random.randint(0, 999999999),
         "scheduler": scheduler,
         "n_iter": 1,
         "steps": steps,
-        "cfg": 11.0,
+        "cfg": cfg,
         "offset_noise": 0.0,
         "width": width,
         "height": height,
@@ -75,7 +76,7 @@ def generate_image(prompt, negative_prompt, scheduler, steps, width, height, res
         image = PIL.Image.open(BytesIO(image_response.content))
 
         return image
-   else:
+    else:
         raise Exception("API request failed with status code " + str(response.status_code))
 
 # Define the Gradio interface
@@ -83,7 +84,7 @@ iface = gr.Interface(
     fn=generate_image, 
     inputs=[
         gr.components.Textbox(label="Prompt"),
-        gr.components.Textbox(label="Negative 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.Dropdown(choices=[
             "Euler a",
             "Euler",
@@ -105,10 +106,12 @@ iface = gr.Interface(
             "DDIM",
             "PLMS"
         ], label="Scheduler", default="Euler a"),
-        gr.components.Slider(minimum=10, maximum=60, default=30, label="Steps"),
-        gr.components.Slider(minimum=512, maximum=1500, default=768, label="Width"),
-        gr.components.Slider(minimum=512, maximum=1500, default=768, label="Height"),
-        gr.components.Checkbox(label="Restore Faces")
+        gr.components.Slider(minimum=10, maximum=100, 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.Checkbox(label="Restore Faces"),
+        gr.components.Number(label="Seed", default=None),
+        gr.components.Slider(minimum=4, maximum=12, default==7.0, label="CFG")
     ], 
     outputs=gr.components.Image(),
     title="Freedom Demonstration",