Update app.py

app.py

@@ -160,46 +160,130 @@ def generate(
     print(image_paths)
     return image_paths, seed
 
-# Example setup for Gradio interface
+# Example prompts
+examples = [
+    "A Monkey with a happy face in the Sahara desert.",
+    "Eiffel Tower was Made up of ICE.",
+    "Color photo of a corgi made of transparent glass, standing on the riverside in Yosemite National Park.",
+    "A close-up photo of a woman. She wore a blue coat with a gray dress underneath and has blue eyes.",
+    "A litter of golden retriever puppies playing in the snow. Their heads pop out of the snow, covered in.",
+    "an astronaut sitting in a diner, eating fries, cinematic, analog film",
+]
+
+# Initialize dictionaries for dynamic UI components
+color_checkboxes = {}
+color_sliders = {}
 
+# Set up the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown(DESCRIPTION)
-    with gr.Row():
-        with gr.Column():
-            prompt = gr.Text(label="Prompt", placeholder="Enter your prompt")
-            negative_prompt = gr.Text(label="Negative Prompt", placeholder="Enter a negative prompt")
-            style_selection = gr.Radio(choices=STYLE_NAMES, label="Style")
-            use_negative_prompt = gr.Checkbox(label="Use Negative Prompt")
-            seed = gr.Number(label="Seed", value=0)
-            width = gr.Number(label="Width", value=1024)
-            height = gr.Number(label="Height", value=1024)
-            inference_steps = gr.Slider(minimum=4, maximum=20, label="Inference Steps", value=4)
-            randomize_seed = gr.Checkbox(label="Randomize Seed")
-            with gr.Accordion("Color Influences"):
-                for color in color_attributes:
-                    with gr.Row():
-                        color_checkboxes[color] = gr.Checkbox(label=f"{color} Selected", value=False)
-                        color_sliders[color] = gr.Slider(label=f"{color} Ratio", minimum=0, maximum=1, step=0.01, value=0.1)
-
-    run_button = gr.Button("Generate")
-    result = gr.Gallery()
-
-    run_button.click(
+    with gr.Row(equal_height=False):
+        with gr.Group():
+            with gr.Row():
+                prompt = gr.Text(
+                    label="Prompt",
+                    show_label=False,
+                    max_lines=1,
+                    placeholder="Enter your prompt",
+                    container=False,
+                )
+                run_button = gr.Button("Run", scale=0)
+            result = gr.Gallery(label="Result", columns=NUM_IMAGES_PER_PROMPT, show_label=False)
+                    
+    with gr.Accordion("Color Influences", open=False):
+        with gr.Group():
+            for color in color_attributes:
+                with gr.Row():
+                    color_checkboxes[color] = gr.Checkbox(label=f"{color} Selected", value=False)
+                    color_sliders[color] = gr.Slider(label=f"{color} Influence Ratio", minimum=0, maximum=1, step=0.01, value=0)
+    
+    with gr.Accordion("Advanced options", open=False):
+        with gr.Group():
+            with gr.Row():
+                use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=False, visible=True)
+                negative_prompt = gr.Text(
+                    label="Negative prompt",
+                    max_lines=1,
+                    placeholder="Enter a negative prompt",
+                    visible=True,
+                )
+            style_selection = gr.Radio(
+                show_label=True,
+                container=True,
+                interactive=True,
+                choices=STYLE_NAMES,
+                value=DEFAULT_STYLE_NAME,
+                label="Image Style",
+            )
+            seed = gr.Slider(
+                label="Seed",
+                minimum=0,
+                maximum=MAX_SEED,
+                step=1,
+                value=0,
+            )
+            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+            with gr.Row(visible=True):
+                width = gr.Slider(
+                    label="Width",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=1024,
+                )
+                height = gr.Slider(
+                    label="Height",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=1024,
+                )
+        with gr.Row():
+            inference_steps = gr.Slider(
+                label="Steps",
+                minimum=4,
+                maximum=20,
+                step=1,
+                value=4,
+            )
+
+    gr.Examples(
+        examples=examples,
+        inputs=prompt,
+        outputs=[result, seed],
+        fn=generate,
+        cache_examples=CACHE_EXAMPLES,
+    )
+
+    use_negative_prompt.change(
+        fn=lambda x: gr.update(visible=x),
+        inputs=use_negative_prompt,
+        outputs=negative_prompt,
+        api_name=False,
+    )
+
+    gr.on(
+        triggers=[
+            prompt.submit,
+            negative_prompt.submit,
+            run_button.click,
+        ],
         fn=generate,
-        inputs={
-            "prompt": prompt,
-            "negative_prompt": negative_prompt,
-            "style": style_selection,
-            "use_negative_prompt": use_negative_prompt,
-            "seed": seed,
-            "width": width,
-            "height": height,
-            "inference_steps": inference_steps,
-            "randomize_seed": randomize_seed,
-            **{f"{color.lower()}_selected": color_checkboxes[color] for color in color_attributes},
-            **{f"{color.lower()}_ratio": color_sliders[color] for color in color_attributes}
-        },
-        outputs=result
+        inputs=[
+            prompt,
+            negative_prompt,
+            style_selection,
+            use_negative_prompt,
+            seed,
+            width,
+            height,
+            inference_steps,
+            randomize_seed,
+            *[color_checkboxes[color] for color in color_attributes],
+            *[color_sliders[color] for color in color_attributes],
+        ],
+        outputs=[result, seed],
+        api_name="run",
     )
 
 if __name__ == "__main__":