app.py

import streamlit as st
from PIL import Image, ImageFile, UnidentifiedImageError
from io import BytesIO

ImageFile.LOAD_TRUNCATED_IMAGES = True

st.title("Reductor de tamaño de imágenes a PNG")

# Permitimos la carga de múltiples imágenes a la vez
uploaded_files = st.file_uploader("Elige las imágenes", type=["png", "jpg", "jpeg", "webp", "tiff", "bmp"], accept_multiple_files=True)

MAX_SIZE = 1 * 1024 * 1024  # 1 MB

def compress_image(image, initial_compression=1, step=1):
    compression = initial_compression
    output_data = None
    with BytesIO() as output:
        while compression <= 9:
            output.seek(0)  # Limpiar el BytesIO
            image.save(output, format="PNG", compress_level=compression)
            if len(output.getvalue()) <= MAX_SIZE:
                output_data = output.getvalue()
                break
            compression += step

    if output_data is None:  # Si no se pudo comprimir lo suficiente
        with BytesIO() as output:
            image.save(output, format="PNG", compress_level=9)
            output_data = output.getvalue()
    
    compressed_image = Image.open(BytesIO(output_data))
    return compressed_image, output_data

if uploaded_files:
    for uploaded_file in uploaded_files:
        image = Image.open(uploaded_file)
        
        # Comprime la imagen
        compressed_image, compressed_data = compress_image(image)
        
        # Botón de descarga individual
        st.download_button(
            label="Descargar imagen comprimida",
            data=compressed_data,
            file_name=f"compressed_image.png",
            mime="image/png"
        )
        
        # Muestra la imagen comprimida en un tamaño reducido
        st.image(compressed_image, width=150)