Update app.py

app.py

@@ -122,24 +122,21 @@ def generate(
     inference_steps: int = 4,
     randomize_seed: bool = False,
     use_resolution_binning: bool = True,
-    **kwargs  # This will capture all the dynamic inputs like color selections and ratios
+    **color_ratios  # Collect color ratios dynamically
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
     
     if not use_negative_prompt:
-        negative_prompt = None
+        negative_prompt = None  # type: ignore
     
-    # 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
-    }
+    # Process color selections and their ratios
+    color_selections = {color: {"selected": color_ratios.get(f"{color.lower()}_selected", False), "ratio": color_ratios.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,
@@ -158,6 +155,7 @@ def generate(
         return [], seed
     
     image_paths = [save_image(img) for img in images]
+    print(image_paths)
     return image_paths, seed
 
 # Example prompts
@@ -170,42 +168,81 @@ examples = [
     "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
-# Assuming the generate function and other necessary imports and initializations are already defined
-
-# Initialize dynamic UI components for color influences
-color_checkboxes = {color: gr.Checkbox(label=f"{color} Selected", value=False) for color in color_attributes}
-color_sliders = {color: gr.Slider(label=f"{color} Influence Ratio", minimum=0, maximum=1, step=0.01, value=0) for color in color_attributes}
-
 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)
     
-    with gr.Row():
-        prompt = gr.Text(label="Prompt", placeholder="Enter your prompt")
-        run_button = gr.Button("Run")
-    
-    result = gr.Gallery(label="Result", columns=NUM_IMAGES_PER_PROMPT)
-    
+    # Color selection and ratio configuration in the UI
     with gr.Accordion("Color Influences", open=False):
-        for color in color_attributes:
-            with gr.Row():
-                color_checkboxes[color]
-                color_sliders[color]
-
+        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.Column():
-            negative_prompt = gr.Text(label="Negative Prompt", placeholder="Enter a negative prompt")
-            use_negative_prompt = gr.Checkbox(label="Use Negative Prompt", value=False)
-            style_selection = gr.Radio(choices=STYLE_NAMES, label="Style", value=DEFAULT_STYLE_NAME)
-            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", value=True)
+        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,
@@ -215,33 +252,40 @@ with gr.Blocks() as demo:
         cache_examples=CACHE_EXAMPLES,
     )
 
+    # Dynamic updates based on user interactions
     use_negative_prompt.change(
-        fn=lambda visible: gr.update(visible=visible),
+        fn=lambda x: gr.update(visible=x),
         inputs=use_negative_prompt,
         outputs=negative_prompt,
+        api_name=False,
     )
 
-    # List of inputs for the generate function
-    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]
-    ]
-
-    # Connect the UI with the generate function using the run_button click
-    run_button.click(
+    gr.on(
+        triggers=[
+            prompt.submit,
+            negative_prompt.submit,
+            run_button.click,
+        ],
         fn=generate,
-        inputs=generate_inputs,
-        outputs=[result, seed]
+        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",
     )
 
+# Launch the Gradio app
 if __name__ == "__main__":
-    demo.launch()
+    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)