Update app.py

app.py

@@ -1,9 +1,11 @@
 import gradio as gr
-import fitz
+import fitz  # PyMuPDF
 import zipfile
 import tempfile
 import shutil
 import os
+from PIL import Image
+import io
 
 def cbz_to_pdf(cbz_file):
     try:
@@ -11,7 +13,7 @@ def cbz_to_pdf(cbz_file):
         if cbz_file is None:
             raise ValueError("No se recibió ningún archivo CBZ.")
 
-        # Crear un archivo temporal para guardar el archivo CBZ
+        # Crear un directorio temporal para el archivo CBZ
         temp_dir = tempfile.mkdtemp()
         cbz_temp_path = os.path.join(temp_dir, "temp.cbz")
         with open(cbz_temp_path, "wb") as f:
@@ -21,30 +23,32 @@ def cbz_to_pdf(cbz_file):
         pdf_filename = os.path.join(temp_dir, "output.pdf")
         pdf = fitz.open()
 
-        # Abrir el archivo CBZ desde el archivo temporal
+        # Abrir el archivo CBZ
         with zipfile.ZipFile(cbz_temp_path, 'r') as zipf:
-            # Ordenar los nombres de los archivos para mantener el orden de las páginas
+            # Ordenar los nombres de los archivos
             sorted_files = sorted(zipf.namelist())
 
-            # Agregar cada imagen al documento PDF
+            # Procesar cada imagen y añadirla al PDF
             for file_name in sorted_files:
                 image_bytes = zipf.read(file_name)
-                img = fitz.open(stream=image_bytes, filetype="png")
+                image = Image.open(io.BytesIO(image_bytes))
+                pdfbytes = io.BytesIO()
+                image.save(pdfbytes, format="PNG")
+                img = fitz.open("png", pdfbytes.getvalue())
                 pdf_page = pdf.new_page(width=img[0].width, height=img[0].height)
-                pdf_page.insert_image(pdf_page.rect, stream=image_bytes)
+                pdf_page.insert_image(pdf_page.rect, pixmap=img[0].get_pixmap())
 
-        # Guardar el PDF
+        # Guardar el archivo PDF
         pdf.save(pdf_filename)
         pdf.close()
 
-        # Limpiar: eliminar el directorio temporal y su contenido
-        shutil.rmtree(temp_dir)
-
+        # Devolver el archivo PDF
         return pdf_filename
 
     except Exception as e:
         return f"Error al procesar el archivo: {str(e)}"
 
+# Crear la interfaz de Gradio
 iface = gr.Interface(
     fn=cbz_to_pdf,
     inputs=gr.File(type="binary", label="Cargar archivo CBZ"),