Update app.py

app.py

@@ -122,7 +122,7 @@ def generate(
     inference_steps: int = 4,
     randomize_seed: bool = False,
     use_resolution_binning: bool = True,
-    **color_ratios  # Collect color ratios dynamically
+    **kwargs  # This will capture all other keyword arguments dynamically
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
@@ -131,7 +131,10 @@ 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}
+    # 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}
     
     # Apply style and modify prompt based on color selections
     prompt, negative_prompt = apply_style(style, prompt, color_selections)