Update app.py

app.py

@@ -76,27 +76,41 @@ footer {display: none !important}
 #gallery {
     width: 100% !important;
     max-width: 100% !important;
-    overflow: hidden !important;
+    overflow: visible !important;
 }
 #gallery > div {
     width: 100% !important;
     max-width: none !important;
-    transform: scale(2) !important;
-    transform-origin: left top !important;
 }
-.gallery-container {
+#gallery > div > div {
+    width: 100% !important;
     display: grid !important;
     grid-template-columns: repeat(5, 1fr) !important;
     gap: 16px !important;
     padding: 16px !important;
+}
+.gallery-container {
     background: rgba(255, 255, 255, 0.05);
     border-radius: 8px;
     margin-top: 10px;
     width: 100% !important;
-    max-width: none !important;
     box-sizing: border-box !important;
 }
 /* Force gallery items to maintain aspect ratio */
+.gallery-item {
+    width: 100% !important;
+    aspect-ratio: 1 !important;
+    overflow: hidden !important;
+    border-radius: 4px !important;
+}
+.gallery-item img {
+    width: 100% !important;
+    height: 100% !important;
+    object-fit: cover !important;
+    border-radius: 4px !important;
+    transition: transform 0.2s;
+}
+/* Force gallery items to maintain aspect ratio */
 .gallery-item {
     width: 100% !important;
     aspect-ratio: 1 !important;
@@ -145,7 +159,6 @@ footer {display: none !important}
 }
 """
 
-
 def save_image(image):
     """Save the generated image and return the path"""
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -262,7 +275,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 elem_id="output-image",
                 elem_classes=["output-image", "fixed-width"]
             )
-
             gallery = gr.Gallery(
                 label="Generated Images Gallery",
                 show_label=True,
@@ -272,7 +284,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 height="auto",
                 object_fit="cover",
                 elem_classes=["gallery-container", "fixed-width"]
-            )    
+            )
+
+
 
             
             # Load existing gallery images on startup