Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from PIL import Image
 import os
 import tempfile
+from transformers import pipeline
 
 def optimize_image(image, png_optimize, jpeg_quality, jpeg_resolution, webp_quality):
     img = Image.open(image)
@@ -52,91 +53,63 @@ def optimize_image(image, png_optimize, jpeg_quality, jpeg_resolution, webp_qual
         webp_lossy_img, f"WebP con pérdida: {webp_lossy_size:.2f} KB \n(diferencia: {-webp_lossy_diff:.2f} KB)", webp_lossy_output_path
     ]
 
-with gr.Blocks() as demo:
-    gr.Markdown("### Optimización de imágenes para la web")
-
-    image_input = gr.File(label="Sube tu imagen", file_types=['image'])
-    optimize_button = gr.Button("Optimizar")
+# Función de ejemplo para aplicar un modelo de super-resolución
+def super_res_image(image):
+    esrgan = pipeline("image-super-resolution", model="microsoft/real-esrgan")
+    super_res_img = esrgan(image)
+    return super_res_img
 
-    with gr.Row():
-        with gr.Column():
-            optimized_output1 = gr.Image(label="Optimización sin pérdida")
-            png_optimize = gr.Checkbox(label="Optimizar PNG", value=True)
-            download_button1 = gr.File(label="Descargar", visible=True)
-            optimized_size1 = gr.Text(value="", interactive=False, show_label=False)
-
-        with gr.Column():
-            optimized_output2 = gr.Image(label="Optimización con pérdida (JPEG)")
-            jpeg_quality = gr.Slider(label="Calidad JPEG", minimum=10, maximum=100, value=75, step=1)
-            download_button2 = gr.File(label="Descargar", visible=True)
-            optimized_size2 = gr.Text(value="", interactive=False, show_label=False)
-            
-        with gr.Column():
-            optimized_output3 = gr.Image(label="Reducción de resolución (JPEG)")
-            jpeg_resolution = gr.Slider(label="Resolución JPEG (%)", minimum=10, maximum=100, value=50, step=1)
-            download_button3 = gr.File(label="Descargar", visible=True)
-            optimized_size3 = gr.Text(value="", interactive=False, show_label=False)
+with gr.Blocks() as demo:
+    with gr.Tab("Optimización Tradicional"):
+        image_input = gr.File(label="Sube tu imagen", file_types=['image'])
+        optimize_button = gr.Button("Optimizar")
 
-        with gr.Column():
-            optimized_output4 = gr.Image(label="Optimización WebP con pérdida")
-            webp_quality = gr.Slider(label="Calidad WebP", minimum=10, maximum=100, value=75, step=1)
-            download_button4 = gr.File(label="Descargar", visible=True)
-            optimized_size4 = gr.Text(value="", interactive=False, show_label=False)
+        with gr.Row():
+            with gr.Column():
+                optimized_output1 = gr.Image(label="Optimización sin pérdida")
+                png_optimize = gr.Checkbox(label="Optimizar PNG", value=True)
+                download_button1 = gr.File(label="Descargar", visible=True)
+                optimized_size1 = gr.Text(value="", interactive=False, show_label=False)
 
-    # Conectar cada control a la función de optimización
-    optimize_button.click(
-        fn=optimize_image,
-        inputs=[image_input, png_optimize, jpeg_quality, jpeg_resolution, webp_quality],
-        outputs=[
-            optimized_output1, optimized_size1, download_button1,
-            optimized_output2, optimized_size2, download_button2,
-            optimized_output3, optimized_size3, download_button3,
-            optimized_output4, optimized_size4, download_button4
-        ]
-    )
+            with gr.Column():
+                optimized_output2 = gr.Image(label="Optimización con pérdida (JPEG)")
+                jpeg_quality = gr.Slider(label="Calidad JPEG", minimum=10, maximum=100, value=50, step=1)
+                download_button2 = gr.File(label="Descargar", visible=True)
+                optimized_size2 = gr.Text(value="", interactive=False, show_label=False)
+                
+            with gr.Column():
+                optimized_output3 = gr.Image(label="Reducción de resolución (JPEG)")
+                jpeg_resolution = gr.Slider(label="Resolución JPEG (%)", minimum=10, maximum=100, value=50, step=1)
+                download_button3 = gr.File(label="Descargar", visible=True)
+                optimized_size3 = gr.Text(value="", interactive=False, show_label=False)
 
-    png_optimize.change(
-        fn=optimize_image,
-        inputs=[image_input, png_optimize, jpeg_quality, jpeg_resolution, webp_quality],
-        outputs=[
-            optimized_output1, optimized_size1, download_button1,
-            optimized_output2, optimized_size2, download_button2,
-            optimized_output3, optimized_size3, download_button3,
-            optimized_output4, optimized_size4, download_button4
-        ]
-    )
+            with gr.Column():
+                optimized_output4 = gr.Image(label="Optimización WebP con pérdida")
+                webp_quality = gr.Slider(label="Calidad WebP", minimum=10, maximum=100, value=50, step=1)
+                download_button4 = gr.File(label="Descargar", visible=True)
+                optimized_size4 = gr.Text(value="", interactive=False, show_label=False)
 
-    jpeg_quality.change(
-        fn=optimize_image,
-        inputs=[image_input, png_optimize, jpeg_quality, jpeg_resolution, webp_quality],
-        outputs=[
-            optimized_output1, optimized_size1, download_button1,
-            optimized_output2, optimized_size2, download_button2,
-            optimized_output3, optimized_size3, download_button3,
-            optimized_output4, optimized_size4, download_button4
-        ]
-    )
+        # Conectar cada control a la función de optimización
+        optimize_button.click(
+            fn=optimize_image,
+            inputs=[image_input, png_optimize, jpeg_quality, jpeg_resolution, webp_quality],
+            outputs=[
+                optimized_output1, optimized_size1, download_button1,
+                optimized_output2, optimized_size2, download_button2,
+                optimized_output3, optimized_size3, download_button3,
+                optimized_output4, optimized_size4, download_button4
+            ]
+        )
 
-    jpeg_resolution.change(
-        fn=optimize_image,
-        inputs=[image_input, png_optimize, jpeg_quality, jpeg_resolution, webp_quality],
-        outputs=[
-            optimized_output1, optimized_size1, download_button1,
-            optimized_output2, optimized_size2, download_button2,
-            optimized_output3, optimized_size3, download_button3,
-            optimized_output4, optimized_size4, download_button4
-        ]
-    )
+    with gr.Tab("Optimización con Modelos de Hugging Face"):
+        hf_image_input = gr.File(label="Sube tu imagen para optimización avanzada", file_types=['image'])
+        super_res_output = gr.Image(label="Super-Resolution Result")
+        hf_button = gr.Button("Aplicar Super-Resolution")
 
-    webp_quality.change(
-        fn=optimize_image,
-        inputs=[image_input, png_optimize, jpeg_quality, jpeg_resolution, webp_quality],
-        outputs=[
-            optimized_output1, optimized_size1, download_button1,
-            optimized_output2, optimized_size2, download_button2,
-            optimized_output3, optimized_size3, download_button3,
-            optimized_output4, optimized_size4, download_button4
-        ]
-    )
+        hf_button.click(
+            fn=super_res_image,
+            inputs=hf_image_input,
+            outputs=super_res_output
+        )
 
 demo.launch()