Spaces:
Running
Running
import gradio as gr | |
from PIL import Image | |
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 | |
img = Image.open(image) | |
return img, f"Tamaño del archivo: {file_size:.2f} KB" | |
def convert_image_format(image, target_format): | |
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) | |
# 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 | |
# Devolver la ruta del archivo convertido 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: | |
gr.Markdown("### Conversor de formatos de imagen") | |
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") | |
# 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") | |
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() | |