Update app.py

app.py

@@ -1,43 +1,33 @@
 import gradio as gr
-import tempfile
+import os
+import fitz
 from zipfile import ZipFile
-from PIL import Image
-from pdf2image import convert_from_bytes
+from io import BytesIO
 
 def pdf_to_cbz(pdf_file):
-    if pdf_file is None:
-        raise ValueError("No se recibió ningún archivo PDF.")
-
-    # pdf_file es una tupla de (nombre, tipo, bytes)
     pdf_filename, file_type, pdf_bytes = pdf_file
-
-    # Convertir las páginas del PDF en imágenes directamente desde los bytes
-    images = convert_from_bytes(pdf_bytes)
-
-    # Crear un archivo CBZ temporal
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".cbz") as temp_cbz:
-        with ZipFile(temp_cbz.name, 'w') as zipf:
-            for i, img in enumerate(images):
-                img_filename = f"page_{i+1}.png"
-                img_path = os.path.join(tempfile.gettempdir(), img_filename)
-                img.save(img_path, "PNG")
-                zipf.write(img_path, img_filename)
-                os.remove(img_path)
-        cbz_temp_file = temp_cbz.name
-
-    # Leer el archivo CBZ y devolver los bytes
-    with open(cbz_temp_file, 'rb') as cbz_file:
-        cbz_bytes = cbz_file.read()
-
-    # Limpiar el archivo temporal CBZ
-    os.remove(cbz_temp_file)
-
-    return cbz_bytes
+    doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+    cbz_temp_file = BytesIO()
+
+    with ZipFile(cbz_temp_file, 'w') as zipf:
+        for page_num in range(len(doc)):
+            page = doc.load_page(page_num)
+            pix = page.get_pixmap()
+            img_data = pix.tobytes("png")
+            img_filename = f"page_{page_num + 1}.png"
+            zipf.writestr(img_filename, img_data)
+    doc.close()
+    
+    cbz_temp_file.seek(0)
+    return cbz_temp_file, "comic.cbz"
 
 iface = gr.Interface(
     fn=pdf_to_cbz,
     inputs=gr.File(type="binary", label="Cargar archivo PDF"),
-    outputs=gr.File(label="Descargar archivo CBZ")
+    outputs=gr.File(label="Descargar archivo CBZ"),
+    title="Conversor de PDF a CBZ",
+    description="Esta herramienta convierte un archivo PDF en un archivo CBZ."
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()