Update app.py

app.py

@@ -1,10 +1,11 @@
+import spaces
+import gradio as gr
 import fitz  # PyMuPDF
 from PIL import Image
 import pytesseract
 import os
 import numpy as np
 import cv2
-import gradio as gr
 
 def clean_ocr_text(text):
     lines = text.splitlines()
@@ -68,8 +69,14 @@ def extract_text_markdown(doc, image_paths, page_index, seen_xrefs):
     markdown_output += "\n---\n\n"
     return markdown_output.strip()
 
-def convert(pdf_file_path):
-    doc = fitz.open(pdf_file_path)
+@spaces.GPU
+def convert(inputs):
+    # <- inputs es una lista, aunque solo tenga 1 archivo
+    pdf_file = inputs[0]
+    if not os.path.exists(pdf_file):
+        return "Archivo no encontrado", [], ""
+
+    doc = fitz.open(pdf_file)
     markdown_output = ""
     image_paths = []
     seen_xrefs = set()
@@ -128,16 +135,16 @@ def convert(pdf_file_path):
 
     return markdown_output.strip(), image_paths, markdown_path
 
-# --- Interfaz Gradio API-compatible ---
-iface = gr.Interface(
-    fn=convert,
-    inputs=gr.File(type="filepath", label="Archivo PDF"),
-    outputs=[
-        gr.Textbox(label="Markdown generado", lines=25),
-        gr.Gallery(label="Imágenes extraídas", type="file"),
-        gr.File(label="Descargar Markdown")
-    ],
-    title="Conversor PDF a Markdown"
-)
-
-iface.launch()
+# Interfaz Gradio
+with gr.Blocks() as demo:
+    with gr.Row():
+        pdf_input = gr.File(label="Sube tu PDF", type="filepath")
+        submit_btn = gr.Button("Procesar PDF")
+
+    markdown_output = gr.Textbox(label="Markdown estructurado", lines=25, interactive=True)
+    gallery_output = gr.Gallery(label="Imágenes extraídas", type="file")
+    download_md = gr.File(label="Descargar .md")
+
+    submit_btn.click(fn=convert, inputs=[pdf_input], outputs=[markdown_output, gallery_output, download_md])
+
+demo.launch()