Update app.py

app.py

@@ -3,8 +3,19 @@ from PIL import Image
 import os
 import tempfile
 
+def get_image_info(image):
+    # Obtener la ruta del archivo
+    image_path = image.name
+    
+    # Obtener el tamaño del archivo en KB
+    file_size = os.path.getsize(image_path) / 1024  # Convertir a KB
+    
+    # Abrir la imagen para mostrarla
+    img = Image.open(image_path)
+    
+    return img, f"Tamaño del archivo: {file_size:.2f} KB"
+
 def convert_image_format(image, target_format):
-    # Crear un directorio temporal para guardar la imagen
     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()}")
@@ -15,8 +26,11 @@ def convert_image_format(image, target_format):
         # Guardar la imagen en el formato deseado
         img.save(output_name, format=target_format.upper())
 
-        # Devolver la ruta del archivo convertido
-        return output_name
+        # Calcular el tamaño del nuevo archivo
+        file_size = os.path.getsize(output_name) / 1024  # Convertir a KB
+
+        # Devolver la imagen convertida y su tamaño
+        return output_name, f"Tamaño del archivo convertido: {file_size:.2f} KB"
 
 # Interfaz de Gradio
 with gr.Blocks() as demo:
@@ -26,8 +40,15 @@ with gr.Blocks() as demo:
     format_dropdown = gr.Dropdown(label="Selecciona el formato de salida", choices=["jpg", "png", "webp"], value="webp")
     convert_button = gr.Button("Convertir")
 
-    output_gallery = gr.Gallery(label="Imagen convertida")
+    # Mostrar la imagen original y su tamaño
+    image_output = gr.Image(label="Imagen Original")
+    size_output = gr.Text(label="Información del Archivo")
+
+    # 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")
 
-    convert_button.click(fn=convert_image_format, inputs=[image_input, format_dropdown], outputs=output_gallery)
+    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])
 
 demo.launch()