Merlintxu commited on
Commit
216e8c5
1 Parent(s): 4c45b50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -20,14 +20,24 @@ def convert_image_format(image, target_format):
20
  # Abrir la imagen con PIL
21
  img = Image.open(image)
22
 
 
 
 
 
23
  # Guardar la imagen en el formato deseado
24
  img.save(output_name, format=target_format.upper())
25
 
26
  # Calcular el tamaño del nuevo archivo
27
  file_size = os.path.getsize(output_name) / 1024 # Convertir a KB
 
 
 
 
 
 
28
 
29
- # Devolver la ruta del archivo convertido y su tamaño
30
- return output_name, f"Tamaño del archivo convertido: {file_size:.2f} KB"
31
 
32
  # Interfaz de Gradio
33
  with gr.Blocks() as demo:
 
20
  # Abrir la imagen con PIL
21
  img = Image.open(image)
22
 
23
+ # Si el formato es JPG, convertir a RGB si la imagen es RGBA o tiene transparencia
24
+ if target_format.lower() == "jpg":
25
+ img = img.convert("RGB")
26
+
27
  # Guardar la imagen en el formato deseado
28
  img.save(output_name, format=target_format.upper())
29
 
30
  # Calcular el tamaño del nuevo archivo
31
  file_size = os.path.getsize(output_name) / 1024 # Convertir a KB
32
+
33
+ # Volver a abrir la imagen convertida para que Gradio la pueda mostrar
34
+ converted_img = Image.open(output_name)
35
+
36
+ # Convertir a un formato que Gradio pueda mostrar y manejar adecuadamente
37
+ converted_img = converted_img.convert("RGB")
38
 
39
+ # Retornar la imagen convertida para visualización en Gradio y su tamaño
40
+ return converted_img, f"Tamaño del archivo convertido: {file_size:.2f} KB"
41
 
42
  # Interfaz de Gradio
43
  with gr.Blocks() as demo: