Update app.py

app.py

@@ -35,10 +35,10 @@ def optimize_image(image):
         webp_lossy_diff = original_size - webp_lossy_size
         
         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)"
+            lossless_img, f"Sin pérdida: {lossless_size:.2f} KB ({-lossless_diff:.2f} KB)", lossless_output_path,
+            lossy_img, f"Con pérdida: {lossy_size:.2f} KB ({-lossy_diff:.2f} KB)", lossy_output_path,
+            reduced_img, f"Reducción de resolución: {reduced_size:.2f} KB ({-reduced_diff:.2f} KB)", reduced_output_path,
+            webp_lossy_img, f"WebP con pérdida: {webp_lossy_size:.2f} KB ({-webp_lossy_diff:.2f} KB)", webp_lossy_output_path
         ]
 
 with gr.Blocks() as demo:
@@ -48,27 +48,40 @@ with gr.Blocks() as demo:
     optimize_button = gr.Button("Optimizar")
 
     with gr.Row():
-        optimized_output1 = gr.Image(label="Optimización sin pérdida")
-        optimized_size1 = gr.Text()
+        with gr.Column():
+            optimized_output1 = gr.Image(label="Optimización sin pérdida")
+            optimized_size1 = gr.Text(visible=False)
+            download_button1 = gr.Button("Descargar")
         
-        optimized_output2 = gr.Image(label="Optimización con pérdida")
-        optimized_size2 = gr.Text()
+        with gr.Column():
+            optimized_output2 = gr.Image(label="Optimización con pérdida")
+            optimized_size2 = gr.Text(visible=False)
+            download_button2 = gr.Button("Descargar")
 
-        optimized_output3 = gr.Image(label="Reducción de resolución")
-        optimized_size3 = gr.Text()
+        with gr.Column():
+            optimized_output3 = gr.Image(label="Reducción de resolución")
+            optimized_size3 = gr.Text(visible=False)
+            download_button3 = gr.Button("Descargar")
 
-        optimized_output4 = gr.Image(label="Optimización WebP con pérdida")
-        optimized_size4 = gr.Text()
+        with gr.Column():
+            optimized_output4 = gr.Image(label="Optimización WebP con pérdida")
+            optimized_size4 = gr.Text(visible=False)
+            download_button4 = gr.Button("Descargar")
 
     optimize_button.click(
         fn=optimize_image,
         inputs=image_input,
         outputs=[
-            optimized_output1, optimized_size1,
-            optimized_output2, optimized_size2,
-            optimized_output3, optimized_size3,
-            optimized_output4, optimized_size4
+            optimized_output1, optimized_size1, download_button1,
+            optimized_output2, optimized_size2, download_button2,
+            optimized_output3, optimized_size3, download_button3,
+            optimized_output4, optimized_size4, download_button4
         ]
     )
 
+    download_button1.click(lambda x: x, inputs=optimized_size1, outputs=download_button1)
+    download_button2.click(lambda x: x, inputs=optimized_size2, outputs=download_button2)
+    download_button3.click(lambda x: x, inputs=optimized_size3, outputs=download_button3)
+    download_button4.click(lambda x: x, inputs=optimized_size4, outputs=download_button4)
+
 demo.launch()