Update app.py

app.py

@@ -175,7 +175,9 @@ color_checkboxes = {}
 color_sliders = {}
 
 # Set up the Gradio interface
-# Initialize color checkboxes and sliders before using them
+# 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}
 
@@ -205,7 +207,6 @@ with gr.Blocks() as demo:
             inference_steps = gr.Slider(minimum=4, maximum=20, label="Inference Steps", value=4)
             randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
 
-    # Set examples for predefined prompts
     gr.Examples(
         examples=examples,
         inputs=prompt,
@@ -220,22 +221,22 @@ with gr.Blocks() as demo:
         outputs=negative_prompt,
     )
 
-    # Define the inputs for the generate function using a dictionary for clear mapping
-    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}
-    }
-
-    # Connect the UI with the generate function
+    # 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(
         fn=generate,
         inputs=generate_inputs,