change

app.py

@@ -89,13 +89,17 @@ def _headers() -> dict:
     return {"Authorization": f"Bearer {API_TOKEN}"}
 
 
-def create_request(prompt, wh_ratio):
+def create_request(prompt, wh_ratio, enable_prompt_refine=True, seed=-1):
     """
     Submit an image generation request to the API.
 
     Args:
         prompt (str): Text prompt describing the image to generate
         wh_ratio (str): Aspect ratio for the output image (e.g. "16:9")
+        enable_prompt_refine (bool): Whether to let the backend rewrite/expand
+            the prompt before generation. Sent to the API as 0 / 1.
+        seed (int): Generation seed. -1 means the backend will pick one
+            randomly; any other integer fixes the seed for reproducible runs.
 
     Returns:
         str: Task ID
@@ -104,7 +108,8 @@ def create_request(prompt, wh_ratio):
         APIError: If the API request fails
     """
     logger.info(
-        f"Starting create_request with prompt='{prompt[:50]}...', wh_ratio={wh_ratio}"
+        f"Starting create_request with prompt='{prompt[:50]}...', "
+        f"wh_ratio={wh_ratio}, enable_prompt_refine={enable_prompt_refine}, seed={seed}"
     )
 
     if not prompt or not prompt.strip():
@@ -115,11 +120,19 @@ def create_request(prompt, wh_ratio):
         logger.error(f"Invalid wh_ratio: {wh_ratio}. Valid options: {WH_RATIO_OPTIONS}")
         raise ValueError(f"Invalid aspect ratio. Must be one of: {', '.join(WH_RATIO_OPTIONS)}")
 
+    try:
+        seed_int = int(seed)
+    except (TypeError, ValueError):
+        logger.warning(f"Invalid seed value '{seed}', falling back to -1 (random)")
+        seed_int = -1
+
     model_params = {
         "prompt": prompt,
         "wh_ratio": wh_ratio,
         "model_id": MODEL_ID,
         "n": 1,
+        "enable_prompt_refine": 1 if enable_prompt_refine else 0,
+        "seed": seed_int,
     }
 
     url = _build_request_url()
@@ -527,6 +540,84 @@ textarea:focus, input:focus,
     background: #ffffff !important;
 }
 
+/* Advanced options row — keeps the refine switch + seed input visually paired */
+.advanced-row {
+    gap: 14px !important;
+    margin-top: 2px;
+}
+
+/* Refine toggle — render the gradio checkbox as an Apple-style pill switch */
+.refine-toggle {
+    background: #f5f5f7 !important;
+    border-radius: 12px !important;
+    border: 1px solid rgba(0,0,0,0.05) !important;
+    padding: 10px 14px !important;
+}
+.refine-toggle label {
+    display: flex !important;
+    align-items: center !important;
+    gap: 10px !important;
+    cursor: pointer;
+}
+.refine-toggle input[type="checkbox"] {
+    appearance: none;
+    -webkit-appearance: none;
+    width: 38px;
+    height: 22px;
+    border-radius: 999px !important;
+    background: #d2d2d7 !important;
+    position: relative;
+    cursor: pointer;
+    transition: background 0.18s ease;
+    border: none !important;
+    flex-shrink: 0;
+    margin: 0 !important;
+}
+.refine-toggle input[type="checkbox"]::after {
+    content: "";
+    position: absolute;
+    top: 2px;
+    left: 2px;
+    width: 18px;
+    height: 18px;
+    border-radius: 50%;
+    background: #ffffff;
+    box-shadow: 0 1px 3px rgba(0,0,0,0.15), 0 1px 1px rgba(0,0,0,0.06);
+    transition: transform 0.18s ease;
+}
+.refine-toggle input[type="checkbox"]:checked {
+    background: #30d158 !important;
+}
+.refine-toggle input[type="checkbox"]:checked::after {
+    transform: translateX(16px);
+}
+.refine-toggle input[type="checkbox"]:focus-visible {
+    box-shadow: 0 0 0 4px rgba(0,113,227,0.15) !important;
+}
+.refine-toggle .wrap > label > span,
+.refine-toggle label > span {
+    color: #1d1d1f !important;
+    font-size: 14px !important;
+    font-weight: 500 !important;
+    letter-spacing: 0.01em;
+}
+
+/* Seed number input — match the prompt/dropdown rounding */
+.seed-input input[type="number"] {
+    border-radius: 12px !important;
+    padding: 10px 14px !important;
+    font-variant-numeric: tabular-nums;
+}
+/* Hide the native spinner buttons on number inputs for a cleaner look */
+.seed-input input[type="number"]::-webkit-outer-spin-button,
+.seed-input input[type="number"]::-webkit-inner-spin-button {
+    -webkit-appearance: none;
+    margin: 0;
+}
+.seed-input input[type="number"] {
+    -moz-appearance: textfield;
+}
+
 /* Footer tagline */
 .tagline {
     text-align: center;
@@ -780,6 +871,24 @@ def create_ui():
                     info="Width : Height",
                 )
 
+                with gr.Row(elem_classes=["advanced-row"], equal_height=True):
+                    enable_prompt_refine = gr.Checkbox(
+                        value=True,
+                        label="Prompt Refine",
+                        info="Let the model rewrite & enrich your prompt",
+                        elem_classes=["refine-toggle"],
+                        scale=1,
+                    )
+                    seed = gr.Number(
+                        value=-1,
+                        label="Seed",
+                        info="Use -1 for a random seed",
+                        precision=0,
+                        minimum=-1,
+                        elem_classes=["seed-input"],
+                        scale=1,
+                    )
+
                 with gr.Row():
                     clear_btn = gr.Button("Clear", variant="secondary", scale=1)
                     generate_btn = gr.Button("Generate", variant="primary", scale=3)
@@ -799,9 +908,11 @@ def create_ui():
                     elem_classes=["image-output"],
                 )
 
-        def generate_with_status(prompt, wh_ratio_value):
+        def generate_with_status(prompt, wh_ratio_value, enable_prompt_refine_value, seed_value):
             logger.info(
-                f"Starting image generation with prompt='{(prompt or '')[:50]}...', wh_ratio={wh_ratio_value}"
+                f"Starting image generation with prompt='{(prompt or '')[:50]}...', "
+                f"wh_ratio={wh_ratio_value}, enable_prompt_refine={enable_prompt_refine_value}, "
+                f"seed={seed_value}"
             )
 
             yield None, _status_html("Sending request to API…", "running")
@@ -817,13 +928,23 @@ def create_ui():
                     yield None, _status_html(f"Invalid aspect ratio “{wh_ratio_value}”", "error")
                     return
 
+                try:
+                    seed_int = int(seed_value) if seed_value is not None else -1
+                except (TypeError, ValueError):
+                    seed_int = -1
+
                 if FAKE_TEST:
                     logger.info("FAKE_TEST mode active — bypassing real API call")
                     yield from _fake_generation_iter(prompt, wh_ratio_value)
                     return
 
                 logger.info("Creating API request")
-                task_id = create_request(prompt, wh_ratio_value)
+                task_id = create_request(
+                    prompt,
+                    wh_ratio_value,
+                    enable_prompt_refine=bool(enable_prompt_refine_value),
+                    seed=seed_int,
+                )
                 yield None, _status_html(f"Request submitted · Task {task_id[:8]}…", "running")
 
                 start_time = time.time()
@@ -921,7 +1042,7 @@ def create_ui():
             html = _status_html(_queue_text(waiting, running), "running")
             return None, html, True, html
 
-        def _generate_wrapped(prompt_value, wh_ratio_value):
+        def _generate_wrapped(prompt_value, wh_ratio_value, enable_prompt_refine_value, seed_value):
             """Click handler #2 — queued.
 
             Wraps the existing generator and, on the FIRST yield, also flips
@@ -929,7 +1050,9 @@ def create_ui():
             lets the generator's own `yield`s drive it (Generating XXs → ...).
             """
             first = True
-            for image, status_html in generate_with_status(prompt_value, wh_ratio_value):
+            for image, status_html in generate_with_status(
+                prompt_value, wh_ratio_value, enable_prompt_refine_value, seed_value
+            ):
                 if first:
                     first = False
                     yield image, status_html, False
@@ -944,7 +1067,7 @@ def create_ui():
             show_progress="hidden",
         ).then(
             fn=_generate_wrapped,
-            inputs=[prompt, wh_ratio],
+            inputs=[prompt, wh_ratio, enable_prompt_refine, seed],
             outputs=[output_image, status_msg, queued_state],
             show_progress="minimal",
             show_progress_on=[generate_btn],