Update app.py

app.py

@@ -227,7 +227,16 @@ def generate_poster(
 def create_interface():
     """Create Gradio interface"""
     
-    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    with gr.Blocks(
+        theme=gr.themes.Soft(),
+        css="""
+        .preserve-aspect-ratio img {
+            object-fit: contain !important;
+            max-height: 300px !important;
+            width: auto !important;
+        }
+        """
+    ) as demo:
         gr.HTML("<div style='text-align: center; width: 100%; margin: 20px 0;'><h1 style='margin: 0; font-size: 2.5em;'>PosterCraft-v1_RL</h1></div>")
 
         with gr.Row():
@@ -248,7 +257,7 @@ def create_interface():
 
             with gr.Column(scale=1):
                 gr.Markdown("### 2. Results")
-                image_output = gr.Image(label="Generated Image", type="pil", show_download_button=True, container=False)
+                image_output = gr.Image(label="Generated Image", type="pil", show_download_button=True, height=300, 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)