fix: update with dropdown

app.py

@@ -55,6 +55,8 @@ VARIANTS = {
     },
 }
 
+VARIANT_CHOICES = list(VARIANTS)
+
 VARIANT_ALIASES = {
     "6-steps": "6-steps",
     "6-step": "6-steps",
@@ -183,15 +185,32 @@ def _run_pipe(prompt: str, spec: dict, seed: int, image=None):
     return pipe(**kwargs).images[0]
 
 
+def _variant_dropdown() -> gr.Dropdown:
+    return gr.Dropdown(
+        choices=VARIANT_CHOICES,
+        value=DEFAULT_VARIANT,
+        label="Variant",
+    )
+
+
 @spaces.GPU(duration=120)
-def generate(prompt: str, variant: str = DEFAULT_VARIANT, seed: int = 42) -> dict:
+def generate(
+    prompt: str,
+    variant: str = _variant_dropdown(),
+    seed: int = 42,
+) -> dict:
     """Text-to-image with P-Q-Image-2.1 (no CFG)."""
     spec = _ensure_variant(variant)
     return _to_file(_run_pipe(prompt, spec, seed))
 
 
 @spaces.GPU(duration=120)
-def edit(image, prompt: str, variant: str = DEFAULT_VARIANT, seed: int = 42) -> dict:
+def edit(
+    image,
+    prompt: str,
+    variant: str = _variant_dropdown(),
+    seed: int = 42,
+) -> dict:
     """Edit an image with P-Q-Image-2.1 (no CFG)."""
     spec = _ensure_variant(variant)
     return _to_file(_run_pipe(prompt, spec, seed, image=_coerce_image(image)))

workflow.json

@@ -15,13 +15,13 @@
     },
     {
       "id": "ref_variant",
-      "label": "Variant (6-steps | 6-steps-upsampled | 8-steps | 5-steps)",
+      "label": "Variant",
       "role": "reference",
       "asset_type": "text",
       "x": 40,
       "y": 320,
-      "inputs": [{"id": "in", "label": "Text", "type": "text"}],
-      "outputs": [{"id": "out", "label": "Text", "type": "text"}],
+      "inputs": [{"id": "in", "label": "Variant", "type": "text", "choices": ["6-steps", "6-steps-upsampled", "8-steps", "5-steps"]}],
+      "outputs": [{"id": "out", "label": "Variant", "type": "text", "choices": ["6-steps", "6-steps-upsampled", "8-steps", "5-steps"]}],
       "data": {"out": "6-steps"}
     },
     {
@@ -68,7 +68,7 @@
       "y": 100,
       "inputs": [
         {"id": "prompt", "label": "Prompt", "type": "text", "required": true},
-        {"id": "variant", "label": "Variant", "type": "text"},
+        {"id": "variant", "label": "Variant", "type": "text", "choices": ["6-steps", "6-steps-upsampled", "8-steps", "5-steps"], "default_value": "6-steps"},
         {"id": "seed", "label": "Seed", "type": "number"}
       ],
       "outputs": [{"id": "out_0", "label": "Image", "type": "image", "output_index": 0}]
@@ -84,7 +84,7 @@
       "inputs": [
         {"id": "image", "label": "Image", "type": "image", "required": true},
         {"id": "prompt", "label": "Edit Prompt", "type": "text", "required": true},
-        {"id": "variant", "label": "Variant", "type": "text"},
+        {"id": "variant", "label": "Variant", "type": "text", "choices": ["6-steps", "6-steps-upsampled", "8-steps", "5-steps"], "default_value": "6-steps"},
         {"id": "seed", "label": "Seed", "type": "number"}
       ],
       "outputs": [{"id": "out_0", "label": "Edited Image", "type": "image", "output_index": 0}]