Update app.py

app.py

@@ -122,7 +122,26 @@ def generate(
     inference_steps: int = 4,
     randomize_seed: bool = False,
     use_resolution_binning: bool = True,
-    **color_ratios  # Collect color ratios dynamically
+    purple_selected: bool = False,
+    purple_ratio: float = 0.0,
+    green_selected: bool = False,
+    green_ratio: float = 0.0,
+    maroon_selected: bool = False,
+    maroon_ratio: float = 0.0,
+    orange_selected: bool = False,
+    orange_ratio: float = 0.0,
+    yellow_selected: bool = False,
+    yellow_ratio: float = 0.0,
+    red_selected: bool = False,
+    red_ratio: float = 0.0,
+    blue_selected: bool = False,
+    blue_ratio: float = 0.0,
+    pink_selected: bool = False,
+    pink_ratio: float = 0.0,
+    silver_selected: bool = False,
+    silver_ratio: float = 0.0,
+    beige_selected: bool = False,
+    beige_ratio: float = 0.0,
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
@@ -131,7 +150,18 @@ def generate(
         negative_prompt = None  # type: ignore
     
     # 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}
+    color_selections = {
+        "Purple": {"selected": purple_selected, "ratio": purple_ratio},
+        "Green": {"selected": green_selected, "ratio": green_ratio},
+        "Maroon": {"selected": maroon_selected, "ratio": maroon_ratio},
+        "Orange": {"selected": orange_selected, "ratio": orange_ratio},
+        "Yellow": {"selected": yellow_selected, "ratio": yellow_ratio},
+        "Red": {"selected": red_selected, "ratio": red_ratio},
+        "Blue": {"selected": blue_selected, "ratio": blue_ratio},
+        "Pink": {"selected": pink_selected, "ratio": pink_ratio},
+        "Silver": {"selected": silver_selected, "ratio": silver_ratio},
+        "Beige": {"selected": beige_selected, "ratio": beige_ratio},
+    }
     
     # Apply style and modify prompt based on color selections
     prompt, negative_prompt = apply_style(style, prompt, color_selections)
@@ -260,7 +290,7 @@ with gr.Blocks() as demo:
         api_name=False,
     )
 
-    gr.on(
+        gr.on(
         triggers=[
             prompt.submit,
             negative_prompt.submit,
@@ -277,8 +307,27 @@ with gr.Blocks() as demo:
             height,
             inference_steps,
             randomize_seed,
-            *[color_checkboxes[color] for color in color_attributes],
-            *[color_sliders[color] for color in color_attributes]
+            # Color configurations
+            color_checkboxes["Purple"],
+            color_sliders["Purple"],
+            color_checkboxes["Green"],
+            color_sliders["Green"],
+            color_checkboxes["Maroon"],
+            color_sliders["Maroon"],
+            color_checkboxes["Orange"],
+            color_sliders["Orange"],
+            color_checkboxes["Yellow"],
+            color_sliders["Yellow"],
+            color_checkboxes["Red"],
+            color_sliders["Red"],
+            color_checkboxes["Blue"],
+            color_sliders["Blue"],
+            color_checkboxes["Pink"],
+            color_sliders["Pink"],
+            color_checkboxes["Silver"],
+            color_sliders["Silver"],
+            color_checkboxes["Beige"],
+            color_sliders["Beige"],
         ],
         outputs=[result, seed],
         api_name="run",