Update app.py

app.py

@@ -9,11 +9,7 @@ import cv2
 
 def clean_ocr_text(text):
     lines = text.splitlines()
-    cleaned_lines = []
-    for line in lines:
-        line = line.strip()
-        if line and not line.isspace():
-            cleaned_lines.append(line)
+    cleaned_lines = [line.strip() for line in lines if line.strip()]
     return "\n".join(cleaned_lines)
 
 def extract_text_markdown(doc, image_paths, page_index, seen_xrefs):
@@ -70,14 +66,12 @@ def extract_text_markdown(doc, image_paths, page_index, seen_xrefs):
     return markdown_output.strip()
 
 @spaces.GPU
-def convert(inputs):
-    # <- inputs es una lista, aunque solo tenga 1 archivo
-    pdf_file = inputs[0]
-    print(f"Entradas recibidas: {inputs}")
-    if not inputs or not os.path.exists(inputs[0]):
-        return "Archivo no encontrado o inválido.", [], ""
-
-    doc = fitz.open(pdf_file)
+def convert(pdf_bytes):
+    temp_pdf_path = "/tmp/uploaded_file.pdf"
+    with open(temp_pdf_path, "wb") as f:
+        f.write(pdf_bytes)
+
+    doc = fitz.open(temp_pdf_path)
     markdown_output = ""
     image_paths = []
     seen_xrefs = set()
@@ -136,10 +130,9 @@ def convert(inputs):
 
     return markdown_output.strip(), image_paths, markdown_path
 
-# Interfaz Gradio
 with gr.Blocks() as demo:
     with gr.Row():
-        pdf_input = gr.File(label="Sube tu PDF", type="filepath")
+        pdf_input = gr.File(label="Sube tu PDF", type="binary")  # <-- Aquí el cambio importante
         submit_btn = gr.Button("Procesar PDF")
 
     markdown_output = gr.Textbox(label="Markdown estructurado", lines=25, interactive=True)