Update app.py

app.py

@@ -272,9 +272,19 @@ def load_and_apply_stack(extra_adapters: list[str], extra_weights: list[float]):
 
     return loaded, weights_out
 
+
+def clear_lora_stack():
+    """Reset all 6 LoRA dropdowns to 'None' and sliders to 0.75."""
+    updates = []
+    for _ in range(6):
+        updates.append(gr.update(value="None"))
+        updates.append(gr.update(value=0.75))
+    return updates
+
+
 # ── Inference ──────────────────────────────────────────────────────────────────
 MAX_SEED = np.iinfo(np.int32).max
-NEGATIVE_PROMPT = (
+DEFAULT_NEGATIVE_PROMPT = (
     "worst quality, low quality, bad anatomy, bad hands, text, error, "
     "missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, "
     "signature, watermark, username, blurry"
@@ -288,6 +298,7 @@ def infer(
     randomize_seed,
     guidance_scale,
     steps,
+    negative_prompt,
     *lora_params,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -339,7 +350,7 @@ def infer(
         result = pipe(
             image=image,
             prompt=prompt,
-            negative_prompt=NEGATIVE_PROMPT if guidance_scale > 1.0 else None,
+            negative_prompt=negative_prompt if guidance_scale > 1.0 else None,
             num_inference_steps=steps,
             generator=generator,
             true_cfg_scale=guidance_scale,
@@ -366,35 +377,40 @@ def infer(
 
 # ── UI ─────────────────────────────────────────────────────────────────────────
 css = """
-#col-container { margin: 0 auto; max-width: 960px; }
-#main-title h1 { font-size: 2.1em !important; }
-/* Preview defaults to cropping (cover-like); contain keeps the full image in frame */
-#col-container .image-frame,
-#col-container .image-frame > div,
-.contain-preview .image-frame,
-.contain-preview .image-frame > div {
-  overflow: visible !important;
-}
-#col-container .image-frame img,
+#col-container { margin: 0 auto; max-width: 980px; }
+#main-title h1 { font-size: 2.25em !important; letter-spacing: -0.02em; }
+.gr-button { transition: all 0.1s ease; }
+.gr-button:hover { transform: translateY(-1px); box-shadow: 0 4px 12px rgba(0,0,0,0.15); }
+/* Better image preview */
 .contain-preview .image-frame img {
   object-fit: contain !important;
-  height: auto !important;
-  max-height: min(75vh, 900px) !important;
+  max-height: min(78vh, 920px) !important;
   width: auto !important;
   max-width: 100% !important;
-  margin-inline: auto;
+  margin: 0 auto;
+  border-radius: 12px;
+  box-shadow: 0 10px 30px rgba(0,0,0,0.1);
 }
+/* LoRA rows alignment */
+.gr-row { align-items: center; }
 """
 
 LORA_NAMES = ["None"] + sorted(LORA_CONFIGS.keys())
 
 with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("# **Qwen-Image-Edit - 2509**", elem_id="main-title")
+        gr.Markdown("# **Qwen-Image-Edit • 2509**", elem_id="main-title")
         gr.Markdown(
-            "Base: `Qwen-Image-Edit-2509` + `Qwen-Image-Edit-Rapid-AIO-V4`"
-            "Add optional extra LoRAs below — selecting one auto-fills its trigger words."
+            "**Fast local LoRA-powered image editing** powered by `Qwen-Image-Edit-2509` + `Qwen-Image-Edit-Rapid-AIO-V4` + 50+ community LoRAs.<br>"
+            "Upload an image, describe the edit, optionally stack LoRAs, and hit **Edit Image**."
         )
+        with gr.Accordion("💡 Quick Tips", open=False):
+            gr.Markdown(
+                "- **LoRAs auto-add trigger words** to your prompt when selected.<br>"
+                "- Keep total LoRA strength under ~2.5 to avoid artifacts.<br>"
+                "- Use **Randomize Seed** for variations of the same edit.<br>"
+                "- For best results, start with 4-8 steps and CFG 1.0–2.0."
+            )
 
         with gr.Row(equal_height=False):
             with gr.Column():
@@ -419,9 +435,9 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
                 )
 
                 with gr.Accordion("➕ Extra LoRAs (optional)", open=False):
-                    gr.Markdown(
-                        "Stack up to 6 LoRAs."
-                    )
+                    with gr.Row():
+                        gr.Markdown("**Stack up to 6 LoRAs.** Select a LoRA to auto-fill trigger words in the prompt.")
+                        clear_btn = gr.Button("🧹 Clear All", size="sm", variant="secondary", scale=0)
                     lora_stack = []
                     for i in range(6):
                         with gr.Row():
@@ -430,6 +446,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
                                 value="None",
                                 label=f"LoRA {i + 1}",
                                 scale=3,
+                                interactive=True,
                             )
                             sl = gr.Slider(
                                 0.0, 1.5, value=0.75,
@@ -437,7 +454,19 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
                             )
                             lora_stack.extend([dd, sl])
 
+                    clear_btn.click(
+                        fn=clear_lora_stack,
+                        inputs=None,
+                        outputs=lora_stack,
+                    )
+
                 with gr.Accordion("⚙️ Advanced", open=False):
+                    negative_prompt = gr.Textbox(
+                        label="Negative Prompt",
+                        value=DEFAULT_NEGATIVE_PROMPT,
+                        lines=2,
+                        placeholder="Customize what to avoid...",
+                    )
                     seed = gr.Slider(
                         label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0
                     )
@@ -451,7 +480,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
 
     run_button.click(
         fn=infer,
-        inputs=[input_image, prompt, seed, randomize_seed, guidance_scale, steps]
+        inputs=[input_image, prompt, seed, randomize_seed, guidance_scale, steps, negative_prompt]
                 + lora_stack,
         outputs=[output_image, seed],
     )