Update app.py

app.py

@@ -1,43 +1,46 @@
 import gradio as gr
 import fitz
 from zipfile import ZipFile
+import io
 import os
-from tempfile import NamedTemporaryFile
 
+# Función para convertir PDF a CBZ
 def pdf_to_cbz(pdf_file):
-    if not pdf_file:
-        return create_error_file("No se cargó ningún archivo. Por favor, carga un archivo PDF.")
-
     try:
-        # Desempaqueta el archivo PDF
+        # Desempaquetar el archivo recibido
+        if pdf_file is None:
+            return "No se recibió ningún archivo. Por favor, carga un archivo PDF."
+
         pdf_filename, file_type, pdf_bytes = pdf_file
-    except ValueError:
-        return create_error_file("Error al procesar el archivo. Asegúrate de que sea un archivo PDF válido.")
 
-    doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+        # Crear un documento con los bytes del PDF
+        pdf_document = fitz.open("pdf", pdf_bytes)
+
+        # Crear un archivo temporal en memoria para el CBZ
+        cbz_temp_file = io.BytesIO()
 
-    with NamedTemporaryFile(delete=False, suffix=".cbz") as temp_cbz:
-        with ZipFile(temp_cbz.name, 'w') as zipf:
-            for page_num in range(len(doc)):
-                page = doc.load_page(page_num)
+        with ZipFile(cbz_temp_file, 'w') as zipf:
+            for page_num in range(len(pdf_document)):
+                page = pdf_document.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)
+                img_bytes = pix.tobytes("png")
+                img_filename = f"{page_num}.png"
+                zipf.writestr(img_filename, img_bytes)
 
-    doc.close()
-    
-    return temp_cbz.name
+        # Cerrar el documento PDF
+        pdf_document.close()
 
-def create_error_file(error_message):
-    with NamedTemporaryFile(delete=False, suffix=".txt") as temp_txt:
-        temp_txt.write(error_message.encode())
-    return temp_txt.name
+        # Preparar el archivo CBZ para ser devuelto
+        cbz_temp_file.seek(0)
+        return cbz_temp_file.read(), "application/zip", "comic.cbz"
+    except Exception as e:
+        return f"Error al procesar el archivo: {str(e)}"
 
+# Crear la interfaz de Gradio
 iface = gr.Interface(
     fn=pdf_to_cbz,
-    inputs=gr.File(type="binary", label="Cargar archivo PDF"),
-    outputs=gr.File(label="Descargar archivo CBZ"),
+    inputs=gr.File(type="file", label="Cargar archivo PDF"),
+    outputs=gr.File(type="file", label="Descargar archivo CBZ"),
     title="Conversor de PDF a CBZ",
     description="Esta herramienta convierte un archivo PDF en un archivo CBZ."
 )