Update app.py

app.py

@@ -12,6 +12,7 @@ from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
 from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 from gradio.themes import Soft
 from gradio.themes.utils import colors, fonts, sizes
+from typing import Iterable
 
 # --- Custom Theme Definition ---
 colors.orange_red = colors.Color(
@@ -22,9 +23,30 @@ colors.orange_red = colors.Color(
 )
 
 class OrangeRedTheme(Soft):
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(
+        self,
+        *,
+        primary_hue: colors.Color | str = colors.gray,
+        secondary_hue: colors.Color | str = colors.orange_red, # Use the new color
+        neutral_hue: colors.Color | str = colors.slate,
+        text_size: sizes.Size | str = sizes.text_lg,
+        font: fonts.Font | str | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("Outfit"), "Arial", "sans-serif",
+        ),
+        font_mono: fonts.Font | str | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("IBM Plex Mono"), "ui-monospace", "monospace",
+        ),
+    ):
+        super().__init__(
+            primary_hue=primary_hue,
+            secondary_hue=secondary_hue,
+            neutral_hue=neutral_hue,
+            text_size=text_size,
+            font=font,
+            font_mono=font_mono,
+        )
         super().set(
+            background_fill_primary="*primary_50",
             button_primary_background_fill="linear-gradient(90deg, *secondary_500, *secondary_600)",
             button_primary_background_fill_hover="linear-gradient(90deg, *secondary_600, *secondary_700)",
             button_primary_text_color="white",
@@ -93,15 +115,8 @@ def infer(input_image, prompt, lora_adapter, seed=42, randomize_seed=True, guida
         num_images_per_prompt=1,
     ).images[0]
 
-    return (original_image, result), seed, gr.Button(visible=True)
+    return result, seed, gr.Button(visible=True)
 
-@spaces.GPU
-def infer_example(input_image, prompt, lora_adapter):
-    """
-    Wrapper function for gr.Examples to call the main infer logic for the slider.
-    """
-    (original_image, generated_image), seed, _ = infer(input_image, prompt, lora_adapter, upscale_image=False)
-    return (original_image, generated_image), seed
 
 # --- UI ---
 css = """
@@ -119,7 +134,7 @@ with gr.Blocks(css=css, theme=orange_red_theme) as demo:
 
         with gr.Row():
             with gr.Column():
-                input_image = gr.Image(label="Upload Image", type="pil", height="300")
+                input_image = gr.Image(label="Upload Image", type="pil", height=400)
                 with gr.Row():
                     prompt = gr.Text(
                         label="Edit Prompt",
@@ -147,8 +162,8 @@ with gr.Blocks(css=css, theme=orange_red_theme) as demo:
                     randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
                     guidance_scale = gr.Slider(
                         label="Guidance Scale",
-                        minimum=1,
-                        maximum=10,
+                        minimum=1.0,
+                        maximum=10.0,
                         step=0.1,
                         value=1.0,
                     )
@@ -161,7 +176,7 @@ with gr.Blocks(css=css, theme=orange_red_theme) as demo:
                     )
 
             with gr.Column():
-                output_slider = gr.Image(label="Output Image", show_label=True, interactive=False, format="png")
+                output_image = gr.Image(label="Output Image", show_label=True, interactive=False, format="png", height=400)
                 reuse_button = gr.Button("Reuse this image", visible=False)
 
         gr.Examples(
@@ -172,9 +187,9 @@ with gr.Blocks(css=css, theme=orange_red_theme) as demo:
                 ["examples/dark.jpg", "Relight the image using soft, diffused lighting that simulates sunlight filtering through curtains.", "Relight"],
             ],
             inputs=[input_image, prompt, lora_adapter],
-            outputs=[output_slider, seed],
+            outputs=[output_image, seed],
             fn=infer,
-            cache_examples=False,
+            cache_examples="lazy",
             label="Examples"
         )
 
@@ -182,12 +197,12 @@ with gr.Blocks(css=css, theme=orange_red_theme) as demo:
         triggers=[run_button.click, prompt.submit],
         fn=infer,
         inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps],
-        outputs=[output_slider, seed, reuse_button]
+        outputs=[output_image, seed, reuse_button]
     )
 
     reuse_button.click(
-        fn=lambda images: images[1] if isinstance(images, (list, tuple)) and len(images) > 1 else images,
-        inputs=[output_slider],
+        fn=lambda image: image,
+        inputs=[output_image],
         outputs=[input_image]
     )