Restore multiple style selection in Space

README.md

@@ -23,8 +23,9 @@ Space repository. The Space builds from the root `requirements.txt` and starts
 
 The Space offers text-to-image generation, one image per request, style
 selection, negative prompts, seed, dimensions, steps, guidance, and a model
-picker. It loads SDXL Base 1.0 from Hugging Face at startup. If the Juggernaut XL
-Ragnarok checkpoint is absent at startup, it downloads the matching Civitai
+picker and multi-style selection. It loads SDXL Base 1.0 from Hugging Face at
+startup. If the Juggernaut XL Ragnarok checkpoint is absent at startup, it
+downloads the matching Civitai
 checkpoint version (`1759168`) and verifies its SHA-256 before loading it.
 Subsequent requests reuse the cached file while the Space disk remains intact.
 The optional `CIVITAI_TOKEN` Space secret is sent as a bearer token if Civitai

app.py

@@ -77,7 +77,7 @@ with gr.Blocks(title="RuinedFooocus · ZeroGPU") as demo:
         with gr.Column():
             prompt = gr.Textbox(label="Prompt", lines=4, placeholder="Describe the image you want to create")
             negative_prompt = gr.Textbox(label="Negative prompt", lines=2)
-            style = gr.Dropdown(choices=list(STYLES), value="None", label="RuinedFooocus style")
+            style = gr.Dropdown(choices=list(STYLES), value=[], multiselect=True, label="Style Selection")
             model = gr.Dropdown(choices=MODEL_CHOICES, value=RAGNAROK if RAGNAROK in PIPELINES else SDXL_BASE, label="Model")
             model_status = gr.Markdown()
             with gr.Row():

hf_space.py

@@ -55,11 +55,17 @@ def prepare_prompts(prompt, negative_prompt, style, seed, styles):
     prompt = str(prompt).strip()
     if not prompt:
         raise ValueError("Enter a prompt before generating.")
-    if style not in styles:
-        raise ValueError(f"Unknown style: {style}")
+    selected_styles = [style] if isinstance(style, str) else list(style or [])
+    for selected_style in selected_styles:
+        if selected_style not in styles:
+            raise ValueError(f"Unknown style: {selected_style}")
     rng = random.Random(seed)
     prompt = expand_wildcards(prompt, rng)
-    template, style_negative = styles[style]
-    styled_prompt = template.replace("{prompt}", prompt)
-    negatives = [part.strip() for part in (negative_prompt, style_negative) if part and part.strip()]
+    styled_prompt = prompt
+    negatives = [negative_prompt.strip()] if negative_prompt and negative_prompt.strip() else []
+    for selected_style in selected_styles:
+        template, style_negative = styles[selected_style]
+        styled_prompt = template.replace("{prompt}", styled_prompt)
+        if style_negative and style_negative.strip():
+            negatives.append(style_negative.strip())
     return styled_prompt, ", ".join(negatives)

tests/test_hf_space.py

@@ -0,0 +1,24 @@
+import unittest
+
+from hf_space import prepare_prompts
+
+
+class PromptStylesTests(unittest.TestCase):
+    def test_combines_selected_styles_and_negative_prompts(self):
+        styles = {
+            "one": ("oil painting of {prompt}", "blurry"),
+            "two": ("dramatic {prompt}", "low contrast"),
+        }
+        positive, negative = prepare_prompts("a fox", "grainy", ["one", "two"], 42, styles)
+        self.assertEqual(positive, "dramatic oil painting of a fox")
+        self.assertEqual(negative, "grainy, blurry, low contrast")
+
+    def test_preserves_single_style_calls(self):
+        self.assertEqual(
+            prepare_prompts("a fox", "", "one", 42, {"one": ("{prompt} in snow", "")}),
+            ("a fox in snow", ""),
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()