Updated app.py

app.py

@@ -1,15 +1,14 @@
 import os
 import shutil
+import subprocess
 import requests
 import gradio as gr
 from urllib.parse import urlparse
-from marker.pdf import process_pdf
 
 def procesar_pdf(pdf_file=None, url_pdf=None):
     output_base = "./marker_output"
     os.makedirs(output_base, exist_ok=True)
 
-    # Determinar si se usa archivo o URL
     if url_pdf:
         parsed = urlparse(url_pdf)
         file_name = os.path.basename(parsed.path) or "documento.pdf"
@@ -21,10 +20,8 @@ def procesar_pdf(pdf_file=None, url_pdf=None):
             response = requests.get(url_pdf)
             if response.status_code != 200 or b"%PDF" not in response.content[:1024]:
                 return "La URL proporcionada no contiene un archivo PDF válido."
-
             with open(file_path, "wb") as f:
                 f.write(response.content)
-
         except Exception as e:
             return f"Error al descargar el PDF desde la URL: {str(e)}"
 
@@ -39,20 +36,19 @@ def procesar_pdf(pdf_file=None, url_pdf=None):
     else:
         return "Por favor, proporciona una URL o sube un archivo PDF."
 
-    # Crear carpeta de salida y limpiar si ya existe
     name_wo_ext = os.path.splitext(os.path.basename(file_path))[0]
     output_dir = os.path.join(output_base, name_wo_ext)
     if os.path.exists(output_dir):
         shutil.rmtree(output_dir)
+    os.makedirs(output_dir, exist_ok=True)
 
     try:
-        process_pdf(
+        subprocess.run([
+            "marker_single",
             file_path,
-            output_dir=output_dir,
-            extract_images=False,
-            extract_tables=False,
-            extract_html=True
-        )
+            "--output_format", "markdown",
+            "--output_dir", output_dir
+        ], check=True)
 
         md_path = os.path.join(output_dir, f"{name_wo_ext}.md")
         if os.path.exists(md_path):
@@ -61,10 +57,11 @@ def procesar_pdf(pdf_file=None, url_pdf=None):
         else:
             return "No se generó el archivo Markdown. Es posible que el PDF esté dañado o sea incompatible."
 
+    except subprocess.CalledProcessError as e:
+        return f"Error ejecutando Marker: {str(e)}"
     except Exception as e:
-        return f"Error al ejecutar Marker: {str(e)}"
+        return f"Error inesperado: {str(e)}"
 
-# Interfaz Gradio
 demo = gr.Interface(
     fn=procesar_pdf,
     inputs=[