Update app.py

app.py

@@ -41,23 +41,6 @@ async def gen(prompts, width, height, model_name, num_variants=1, use_enhanced=T
 def list_saved_images():
     return sorted(DATA_PATH.glob("*.jpg"), key=os.path.getmtime, reverse=True)
 
-def display_gallery():
-    images = list_saved_images()
-    if images:
-        cols = st.columns(8)
-        for i, image_file in enumerate(images):
-            with cols[i % 8]:
-                st.image(str(image_file), caption=image_file.name, use_column_width=True)
-                prompt = get_prompt_for_image(image_file.name)
-                st.write(prompt[:300])
-
-                if st.button(f"Borrar", key=f"delete_{i}_{image_file.name}"):
-                    os.remove(image_file)
-                    st.success("Imagen borrada")
-                    display_gallery()
-    else:
-        st.info("No hay imágenes guardadas.")
-
 def save_prompt(prompt):
     with open(DATA_PATH / "prompts.txt", "a") as f:
         f.write(prompt + "\n")
@@ -133,6 +116,26 @@ async def generate_variations(prompt, num_variants, use_enhanced, style):
             prompts.add(f"{prompt}, estilo: {style}")
     return list(prompts)
 
+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)
+
+        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)
+    else:
+        st.info("No hay imágenes guardadas.")
+
 async def main():
     st.set_page_config(layout="wide")
 
@@ -170,8 +173,7 @@ async def main():
             except Exception as e:
                 st.error(f"Error al generar las imágenes: {str(e)}")
 
-    display_gallery()
-
+    image_viewer()
 
 if __name__ == "__main__":
     asyncio.run(main())