Update app.py

app.py

@@ -1,61 +1,55 @@
 import gradio as gr
-from PIL import Image
+from PIL import Image, ImageOps
 import os
 import tempfile
 
-def get_image_info(image):
-    # Obtener el tamaño del archivo en KB
-    file_size = os.path.getsize(image.name) / 1024  # Convertir a KB
-    
-    # Abrir la imagen para mostrarla
+def optimize_image(image):
+    # Leer la imagen original
     img = Image.open(image)
     
-    return img, f"Tamaño del archivo: {file_size:.2f} KB"
-
-def convert_image_format(image, target_format):
+    # Ruta temporal para guardar las imágenes optimizadas
     with tempfile.TemporaryDirectory() as tmpdirname:
-        # Crear una ruta para el archivo de salida en el directorio temporal
-        output_name = os.path.join(tmpdirname, f"output_image.{target_format.lower()}")
-
-        # Abrir la imagen con PIL
-        img = Image.open(image)
-
-        # Si el formato es JPG, convertir a RGB si la imagen es RGBA o tiene transparencia
-        if target_format.lower() == "jpg":
-            img = img.convert("RGB")
-
-        # Guardar la imagen en el formato deseado
-        img.save(output_name, format=target_format.upper())
-
-        # Calcular el tamaño del nuevo archivo
-        file_size = os.path.getsize(output_name) / 1024  # Convertir a KB
+        # 1. Compresión sin pérdida
+        lossless_output_path = os.path.join(tmpdirname, "lossless.png")
+        img.save(lossless_output_path, format="PNG", optimize=True)
+        lossless_size = os.path.getsize(lossless_output_path) / 1024  # tamaño en KB
+        lossless_img = Image.open(lossless_output_path)
         
-        # Volver a abrir la imagen convertida para que Gradio la pueda mostrar
-        converted_img = Image.open(output_name)
-
-        # Convertir a un formato que Gradio pueda mostrar y manejar adecuadamente
-        converted_img = converted_img.convert("RGB")
-
-        # Retornar la imagen convertida para visualización en Gradio y su tamaño
-        return converted_img, f"Tamaño del archivo convertido: {file_size:.2f} KB"
+        # 2. Compresión con pérdida (reducción de calidad)
+        lossy_output_path = os.path.join(tmpdirname, "lossy.jpg")
+        img.save(lossy_output_path, format="JPEG", quality=50, optimize=True)
+        lossy_size = os.path.getsize(lossy_output_path) / 1024  # tamaño en KB
+        lossy_img = Image.open(lossy_output_path)
+        
+        # 3. Reducción de resolución
+        reduced_output_path = os.path.join(tmpdirname, "reduced_resolution.jpg")
+        reduced_img = img.resize((img.width // 2, img.height // 2), Image.ANTIALIAS)
+        reduced_img.save(reduced_output_path, format="JPEG", quality=85, optimize=True)
+        reduced_size = os.path.getsize(reduced_output_path) / 1024  # tamaño en KB
+        reduced_img = Image.open(reduced_output_path)
+        
+        # Retornar imágenes optimizadas y sus tamaños
+        return (lossless_img, f"Sin pérdida: {lossless_size:.2f} KB",
+                lossy_img, f"Con pérdida: {lossy_size:.2f} KB",
+                reduced_img, f"Reducción de resolución: {reduced_size:.2f} KB")
 
 # Interfaz de Gradio
 with gr.Blocks() as demo:
-    gr.Markdown("### Conversor de formatos de imagen")
+    gr.Markdown("### Optimización de imágenes para la web")
 
     image_input = gr.File(label="Sube tu imagen", file_types=['image'])
-    format_dropdown = gr.Dropdown(label="Selecciona el formato de salida", choices=["jpg", "png", "webp"], value="webp")
-    convert_button = gr.Button("Convertir")
+    optimize_button = gr.Button("Optimizar")
+
+    # Mostrar las imágenes optimizadas y sus tamaños
+    optimized_output1 = gr.Image(label="Optimización sin pérdida")
+    optimized_size1 = gr.Text(label="Tamaño del archivo sin pérdida")
 
-    # Mostrar la imagen original y su tamaño
-    image_output = gr.Image(label="Imagen Original")
-    size_output = gr.Text(label="Información del Archivo")
+    optimized_output2 = gr.Image(label="Optimización con pérdida")
+    optimized_size2 = gr.Text(label="Tamaño del archivo con pérdida")
 
-    # Mostrar la imagen convertida y su tamaño
-    converted_image_output = gr.Image(label="Imagen Convertida")
-    converted_size_output = gr.Text(label="Información del Archivo Convertido")
+    optimized_output3 = gr.Image(label="Reducción de resolución")
+    optimized_size3 = gr.Text(label="Tamaño del archivo con reducción de resolución")
 
-    image_input.change(fn=get_image_info, inputs=image_input, outputs=[image_output, size_output])
-    convert_button.click(fn=convert_image_format, inputs=[image_input, format_dropdown], outputs=[converted_image_output, converted_size_output])
+    optimize_button.click(fn=optimize_image, inputs=image_input, outputs=[optimized_output1, optimized_size1, optimized_output2, optimized_size2, optimized_output3, optimized_size3])
 
 demo.launch()