Update app.py

app.py

@@ -336,6 +336,12 @@ button.primary:hover {
     max-width: 100%;
     object-fit: contain;
 }
+
+.preserve-aspect-ratio img {
+    object-fit: contain !important;
+    width: auto !important;
+    max-height: 512px !important;
+}
 """
     
     with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
@@ -373,8 +379,8 @@ button.primary:hover {
 
                 with gr.Column(scale=1, elem_classes="result-section"):
                     gr.HTML('<div class="section-header"><h3>🎨 2. Results</h3></div>')
-                    with gr.Box(elem_classes=["image-output-container"]):
-                        image_output = gr.Image(label="🖼️ Generated Image", type="pil", show_download_button=True, show_label=False, elem_classes=["image-output"])
+                    with gr.Group(elem_classes=["image-output-container"]):
+                        image_output = gr.Image(label="🖼️ Generated Image", type="pil", show_download_button=True, height=512, elem_classes=["preserve-aspect-ratio"])
                     recapped_prompt_output = gr.Textbox(label="📝 Final Prompt Used", lines=5, interactive=False)
                     status_output = gr.Textbox(label="📊 Status Log", lines=4, interactive=False)