Update app.py

app.py

@@ -328,7 +328,7 @@ def generate_image(
     png_filepath = save_as_png(images[0])
     return png_filepath, gr.update(visible=True)
 
-# MODERN PROFESSIONAL UI
+# MODERN PROFESSIONAL UI CSS
 css = """
 :root {
     --primary: #2563eb;
@@ -766,15 +766,59 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
         value="(lowres, low quality, worst quality:1.2), (text:1.2), watermark, (frame:1.2), deformed, ugly, deformed eyes, blur, out of focus, blurry, deformed cat, deformed, photo, anthropomorphic cat, monochrome, pet collar, gun, weapon, blue, 3d, drones, drone, buildings in background, green",
         visible=False
     )
-    num_steps = gr.Slider(value=30, visible=False)
-    guidance_scale = gr.Slider(value=5.0, visible=False)
-    seed = gr.Slider(value=42, visible=False)
-    scheduler = gr.Dropdown(value="EulerDiscreteScheduler", visible=False)
+    num_steps = gr.Slider(
+        minimum=5,
+        maximum=100,
+        step=1,
+        value=30,
+        label="Number of steps",
+        visible=False
+    )
+    guidance_scale = gr.Slider(
+        minimum=0.1,
+        maximum=20.0,
+        step=0.1,
+        value=5.0,
+        label="Guidance scale",
+        visible=False
+    )
+    seed = gr.Slider(
+        minimum=0,
+        maximum=MAX_SEED,
+        step=1,
+        value=42,
+        label="Seed",
+        visible=False
+    )
+    scheduler = gr.Dropdown(
+        value="EulerDiscreteScheduler",
+        choices=["EulerDiscreteScheduler", "EulerAncestralDiscreteScheduler", "DPMSolverMultistepScheduler"],
+        visible=False
+    )
     randomize_seed = gr.Checkbox(value=True, visible=False)
     enhance_face_region = gr.Checkbox(value=True, visible=False)
-    controlnet_selection = gr.CheckboxGroup(value=["depth"], visible=False)
-    canny_strength = gr.Slider(value=0.4, visible=False)
-    depth_strength = gr.Slider(value=0.4, visible=False)
+    controlnet_selection = gr.CheckboxGroup(
+        choices=["canny", "depth"],
+        value=["depth"],
+        label="Controlnet",
+        visible=False
+    )
+    canny_strength = gr.Slider(
+        minimum=0,
+        maximum=1.5,
+        step=0.01,
+        value=0.4,
+        label="Canny strength",
+        visible=False
+    )
+    depth_strength = gr.Slider(
+        minimum=0,
+        maximum=1.5,
+        step=0.01,
+        value=0.4,
+        label="Depth strength",
+        visible=False
+    )
 
     def show_success():
         return """
@@ -816,5 +860,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
         queue=False,
     )
 
-demo.queue(api_open=False)
-demo.launch(share=True)
+if __name__ == "__main__":
+    demo.queue(api_open=False)
+    demo.launch(share=True)