Update app.py

app.py

@@ -122,24 +122,23 @@ def generate(
     inference_steps: int = 4,
     randomize_seed: bool = False,
     use_resolution_binning: bool = True,
-    **kwargs  # This will capture all other keyword arguments dynamically
+    **kwargs
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
     
     if not use_negative_prompt:
-        negative_prompt = None  # type: ignore
+        negative_prompt = None
     
-    # Process color selections and their ratios
-    # Extract color selections and ratios from kwargs
-    color_selections = {color: {"selected": kwargs.get(f"{color.lower()}_selected", False), 
-                                "ratio": kwargs.get(f"{color.lower()}_ratio", 0)} 
-                        for color in color_attributes}
+    color_selections = {
+        color: {
+            "selected": kwargs.get(f"{color.lower()}_selected", False),
+            "ratio": kwargs.get(f"{color.lower()}_ratio", 0)
+        } for color in color_attributes
+    }
     
-    # Apply style and modify prompt based on color selections
     prompt, negative_prompt = apply_style(style, prompt, color_selections)
     
-    # Generate images
     try:
         images = pipe(
             prompt=prompt,
@@ -161,134 +160,47 @@ def generate(
     print(image_paths)
     return image_paths, seed
 
-# 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",
-]
+# Example setup for Gradio interface
 
-# Set up the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown(DESCRIPTION)
-    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)
-    
-    # Color selection and ratio configuration in the UI
-    with gr.Accordion("Color Influences", open=False):
-        with gr.Group():
-            color_checkboxes = {}
-            color_sliders = {}
-            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.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(
-                choices=STYLE_NAMES,
-                value=DEFAULT_STYLE_NAME,
-                label="Image Style",
-                show_label=True,
-                container=True,
-                interactive=True,
-            )
-            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,
-    )
-
-    # Dynamic updates based on user interactions
-    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,
-        ],
+    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(
         fn=generate,
-        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",
+        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
     )
 
-# Launch the Gradio app
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
-    # Uncomment the next line to launch the server with specific options
-    # demo.queue(max_size=20).launch(server_name="0.0.0.0", server_port=11900, debug=True)
+    demo.launch()