Update app.py

app.py

@@ -119,28 +119,26 @@ async def generate_variations(prompt, num_variants, use_enhanced, style):
 def image_viewer():
     images = list_saved_images()
     if images:
-        if 'current_image_index' not in st.session_state:
-            st.session_state['current_image_index'] = 0
-
-        current_index = st.session_state['current_image_index']
-        image_file = images[current_index]
-        st.image(str(image_file), use_column_width=True, caption=image_file.name)
-
-        st.write("### Galería")
-        thumbnail_cols = st.columns(len(images))
-        for i, image_file in enumerate(images):
-            with thumbnail_cols[i]:
-                if st.button("", key=f"thumb_{i}"):
+        st.write("### Galería de Imágenes")
+        thumbnails = []
+        
+        for image_file in images:
+            with Image.open(image_file) as img:
+                min_side = min(img.size)
+                cropped_img = img.crop(((img.width - min_side) // 2, (img.height - min_side) // 2,
+                                        (img.width + min_side) // 2, (img.height + min_side) // 2))
+                thumbnail = cropped_img.resize((100, 100))
+                thumbnails.append((image_file.name, thumbnail))
+        
+        cols = st.columns(8) 
+        
+        for i, (name, thumbnail) in enumerate(thumbnails):
+            col = cols[i % 8]
+            with col:
+                st.image(thumbnail, use_column_width=False, width=100, caption=name)
+                if st.button("Ver", key=f"view_{i}"):
                     st.session_state['current_image_index'] = i
-                st.image(str(image_file), width=100, caption=image_file.name)
-
-        col1, col2, col3 = st.columns([1, 2, 1])
-        with col1:
-            if st.button("⬅️", key="prev_image"):
-                st.session_state['current_image_index'] = (current_index - 1) % len(images)
-        with col3:
-            if st.button("➡️", key="next_image"):
-                st.session_state['current_image_index'] = (current_index + 1) % len(images)
+                    st.image(str(images[i]), caption=name)
     else:
         st.info("No hay imágenes guardadas.")