Update app.py

app.py

@@ -16,6 +16,16 @@ if not os.path.exists(SAVE_DIR):
 def contar_imagenes(directorio):
     return len([f for f in os.listdir(directorio) if os.path.isfile(os.path.join(directorio, f))])
 
+# Función para eliminar todas las imágenes
+def eliminar_imagenes(directorio):
+    for f in os.listdir(directorio):
+        file_path = os.path.join(directorio, f)
+        try:
+            if os.path.isfile(file_path):
+                os.unlink(file_path)
+        except Exception as e:
+            st.error(f"No se pudo eliminar {file_path}. Error: {e}")
+
 # Número máximo de imágenes permitidas
 MAX_IMAGENES = 2
 num_imagenes_actuales = contar_imagenes(SAVE_DIR)
@@ -55,6 +65,7 @@ if num_imagenes_actuales < MAX_IMAGENES:
                 image.save(file_path)
                 st.success(f"Imagen guardada con éxito en {file_path}.")
                 num_imagenes_actuales += 1
+                st.experimental_rerun()
 
     # Para subir la imagen
     uploaded_file = st.file_uploader("O sube una imagen...", type=["jpg", "jpeg", "png"])
@@ -73,5 +84,10 @@ if num_imagenes_actuales < MAX_IMAGENES:
         image.save(file_path)
         st.success(f"Imagen guardada con éxito en {file_path}.")
         num_imagenes_actuales += 1
+        st.experimental_rerun()
 else:
     st.warning(f"Has alcanzado el límite máximo de {MAX_IMAGENES} imágenes. No puedes subir más imágenes.")
+    if st.button('Eliminar todas las imágenes'):
+        eliminar_imagenes(SAVE_DIR)
+        st.success("Todas las imágenes han sido eliminadas.")
+        st.experimental_rerun()