Update app.py

app.py

@@ -1,39 +1,48 @@
 import gradio as gr
-import os
-os.system("pip install PyMuPDF")
 import fitz
+from zipfile import ZipFile
+import io
+import os
+
+# Función para convertir PDF a CBZ
+def pdf_to_cbz(pdf_file):
+    try:
+        # 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
+
+        # 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 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_bytes = pix.tobytes("png")
+                img_filename = f"{page_num}.png"
+                zipf.writestr(img_filename, img_bytes)
+
+        # Cerrar el documento PDF
+        pdf_document.close()
 
-def pdf_to_zpl(pdf_bytes):
-    # Abre el PDF con PyMuPDF
-    doc = fitz.open("pdf", pdf_bytes)
-    zpl_commands = []
-
-    for page in doc:
-        # Extraer texto de cada página
-        text = page.get_text().strip()
-        # Convertir el texto a un comando ZPL básico
-        # Aquí puedes añadir la lógica para ajustar la posición y el formato según tu diseño de etiqueta
-        zpl_command = (
-            f"^XA\n"  # Comienzo del formato de etiqueta
-            f"^CF0,60\n"  # Selecciona la fuente y tamaño
-            f"^FO50,50^FD{text}^FS\n"  # Campo de datos (posición y texto)
-            f"^XZ"  # Fin del formato de etiqueta
-        )
-        zpl_commands.append(zpl_command)
-    
-    # Cierra el documento PDF
-    doc.close()
-
-    # Une todos los comandos ZPL en un solo string
-    zpl_result = '\n'.join(zpl_commands)
-    return zpl_result
+        # 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_zpl,
-    inputs=gr.File(type="binary", label="Cargar archivo PDF"),
-    outputs=gr.Textbox(label="Código ZPL"),
-    title="Conversor de PDF a ZPL",
-    description="Esta herramienta intenta convertir un archivo PDF en código ZPL."
+    fn=pdf_to_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."
 )
 
 if __name__ == "__main__":