Update app.py

app.py

@@ -169,8 +169,12 @@ def generate(
     return generated_image, control_image, seed
 
 # --- 5. UI Definition ---
-with gr.Blocks(css="footer {display: none !important;}") as demo:
-    gr.Markdown("# Qwen-Image with Union ControlNet (Curated Preprocessors)")
+css = '''
+.fillable{max-width: 960px !important}
+'''
+
+with gr.Blocks(css=css, theme=gr.themes.Citrus()) as demo:
+    gr.Markdown("# Qwen-Image with Union ControlNet")
     gr.Markdown(
         "Generate images using a curated set of stable preprocessors. "
         "Choose a conditioning type, upload an image, and write a prompt."
@@ -178,7 +182,7 @@ with gr.Blocks(css="footer {display: none !important;}") as demo:
 
     with gr.Row():
         with gr.Column(scale=1):
-            input_image = gr.Image(type="pil", label="Input Image", height=512)
+            input_image = gr.Image(type="pil", label="Input Image")
             prompt = gr.Textbox(label="Prompt", placeholder="A detailed description of the desired image...")
             conditioning = gr.Radio(
                 choices=["Canny", "Soft Edge", "Depth", "Pose"],
@@ -202,8 +206,8 @@ with gr.Blocks(css="footer {display: none !important;}") as demo:
                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
         with gr.Column(scale=1):
-            control_image_output = gr.Image(label="Control Image (Preprocessor Output)", interactive=False, height=512)
-            generated_image_output = gr.Image(label="Generated Image", interactive=False, height=512)
+            generated_image_output = gr.Image(label="Generated Image", interactive=False)
+            control_image_output = gr.Image(label="Control Image (Preprocessor Output)", interactive=False)
             used_seed = gr.Number(label="Used Seed", interactive=False)
 
     gr.Examples(