From 5b385dc546467e08543eafdc81c8ae955a6c5ba5 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Mon, 1 Dec 2025 10:48:55 -0800
Subject: [PATCH] Make the image galleries taller

---
 css/main.css                   | 8 ++++++++
 modules/ui_image_generation.py | 4 ++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/css/main.css b/css/main.css
index 565aaf49..317a31f1 100644
--- a/css/main.css
+++ b/css/main.css
@@ -1681,3 +1681,11 @@ button#swap-height-width {
     right: 0;
     border: 0;
 }
+
+#image-output-gallery {
+    height: calc(100vh - 105px);
+}
+
+#image-history-gallery {
+    height: calc(100vh - 139px);
+}
diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py
index fe0c8120..6f251b0e 100644
--- a/modules/ui_image_generation.py
+++ b/modules/ui_image_generation.py
@@ -145,7 +145,7 @@ def create_ui():
 
                     with gr.Column(scale=6, min_width=500):
                         with gr.Column(elem_classes=["viewport-container"]):
-                            shared.gradio['image_output_gallery'] = gr.Gallery(label="Output", show_label=False, columns=2, rows=2, height="80vh", object_fit="contain", preview=True)
+                            shared.gradio['image_output_gallery'] = gr.Gallery(label="Output", show_label=False, columns=2, rows=2, height="80vh", object_fit="contain", preview=True, elem_id="image-output-gallery")
                         with gr.Row():
                             shared.gradio['image_used_seed'] = gr.Markdown(label="Info", interactive=False)
 
@@ -153,7 +153,7 @@ def create_ui():
             with gr.TabItem("Gallery"):
                 with gr.Row():
                     shared.gradio['image_refresh_history'] = gr.Button("🔄 Refresh Gallery", elem_classes="refresh-button")
-                shared.gradio['image_history_gallery'] = gr.Gallery(label="History", show_label=False, columns=6, object_fit="cover", height="auto", allow_preview=True)
+                shared.gradio['image_history_gallery'] = gr.Gallery(label="History", show_label=False, columns=6, object_fit="cover", height="auto", allow_preview=True, elem_id="image-history-gallery")
 
             # TAB 3: MODEL
             with gr.TabItem("Model"):