Update app.py

app.py

@@ -7,6 +7,9 @@ def optimize_image(image):
     # Leer la imagen original
     img = Image.open(image)
     
+    # Calcular el tamaño del archivo original
+    original_size = os.path.getsize(image.name) / 1024  # tamaño en KB
+    
     # Ruta temporal para guardar las imágenes optimizadas
     with tempfile.TemporaryDirectory() as tmpdirname:
         # 1. Compresión sin pérdida
@@ -14,12 +17,14 @@ def optimize_image(image):
         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)
+        lossless_diff = original_size - lossless_size
         
         # 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)
+        lossy_diff = original_size - lossy_size
         
         # 3. Reducción de resolución
         reduced_output_path = os.path.join(tmpdirname, "reduced_resolution.jpg")
@@ -27,18 +32,20 @@ def optimize_image(image):
         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)
+        reduced_diff = original_size - reduced_size
 
         # 4. Compresión con WebP (con pérdida)
         webp_lossy_output_path = os.path.join(tmpdirname, "lossy.webp")
         img.save(webp_lossy_output_path, format="WEBP", quality=50, optimize=True)
         webp_lossy_size = os.path.getsize(webp_lossy_output_path) / 1024  # tamaño en KB
         webp_lossy_img = Image.open(webp_lossy_output_path)
+        webp_lossy_diff = original_size - webp_lossy_size
         
-        # 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",
-                webp_lossy_img, f"WebP con pérdida: {webp_lossy_size:.2f} KB")
+        # Retornar imágenes optimizadas y sus tamaños con la diferencia
+        return (lossless_img, f"Sin pérdida: {lossless_size:.2f} KB (diferencia: {-lossless_diff:.2f} KB)",
+                lossy_img, f"Con pérdida: {lossy_size:.2f} KB (diferencia: {-lossy_diff:.2f} KB)",
+                reduced_img, f"Reducción de resolución: {reduced_size:.2f} KB (diferencia: {-reduced_diff:.2f} KB)",
+                webp_lossy_img, f"WebP con pérdida: {webp_lossy_size:.2f} KB (diferencia: {-webp_lossy_diff:.2f} KB)")
 
 # Interfaz de Gradio
 with gr.Blocks() as demo: