Update app.py

app.py

@@ -99,6 +99,31 @@ def gen(prompts, width, height, model_name, num_variants=1, use_enhanced=True):
         st.error(f"Error al generar imágenes: {e}")
     return images
 
+def display_gallery():
+    st.header("Galería de Imágenes Guardadas")
+    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"FaceSwap", key=f"select_{i}_{image_file.name}"):
+                    st.session_state['generated_image_path'] = str(image_file)
+                    st.success("Imagen seleccionada")
+
+                if st.button(f"Borrar", key=f"delete_{i}_{image_file.name}"):
+                    if os.path.exists(image_file):
+                        os.remove(image_file)
+                        st.success("Imagen borrada")
+                        display_gallery()
+                    else:
+                        st.warning("La imagen no existe.")
+    else:
+        st.info("No hay imágenes guardadas.")
+
 def improve_prompt(prompt):
     try:
         instructions = [