Update app.py

app.py

@@ -310,6 +310,12 @@ button.primary:hover {
     text-align: left !important;
     justify-content: flex-start !important;
 }
+
+.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:
@@ -345,7 +351,7 @@ button.primary:hover {
 
                 with gr.Column(scale=1, elem_classes="result-section"):
                     gr.HTML('<div class="section-header"><h3>🎨 2. Results</h3></div>')
-                    image_output = gr.Image(label="🖼️ Generated Image", type="pil", show_download_button=True, height=512, container=False, scale=None)
+                    image_output = gr.Image(label="🖼️ Generated Image", type="pil", show_download_button=True, height=512, container=False,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)