a

app.py

@@ -3,6 +3,7 @@ import subprocess
 import logging
 import tempfile
 import os
+import shutil
 
 # Configurar logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -19,21 +20,33 @@ def ejecutar_comando(comando, mensaje_exito="", mensaje_error=""):
         logging.error(error_message)
         raise RuntimeError(error_message)
 
+def reparar_pdf(pdf_entrada, pdf_reparado):
+    """Repara un PDF usando qpdf."""
+    comando = f"qpdf --linearize '{pdf_entrada}' '{pdf_reparado}'"
+    ejecutar_comando(
+        comando,
+        mensaje_exito="Archivo PDF reparado con éxito",
+        mensaje_error="Error al reparar el PDF"
+    )
+
 def procesar_pdf_con_ocr(pdf_subido, idioma="spa"):
     """Procesa un PDF subido con OCR y devuelve la ruta del archivo procesado."""
     if not pdf_subido:
         raise gr.Error("No se subió ningún archivo.")
 
     try:
-        input_pdf = pdf_subido.name
+        # Copiar el archivo subido a un nombre seguro
+        safe_input_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf").name
+        shutil.copy(pdf_subido.name, safe_input_pdf)
+
+        reparado_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf").name
         output_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf").name
 
-        # Verificar que el archivo de entrada es un PDF
-        if not input_pdf.endswith(".pdf"):
-            raise gr.Error("El archivo subido no es un PDF válido.")
+        # Reparar el PDF
+        reparar_pdf(safe_input_pdf, reparado_pdf)
 
-        # Ejecutar ocrmypdf
-        comando = f"ocrmypdf -l {idioma} --force-ocr --deskew --output-type pdf '{input_pdf}' '{output_pdf}'"
+        # Ejecutar OCR
+        comando = f"ocrmypdf -l {idioma} --force-ocr --deskew --output-type pdf '{reparado_pdf}' '{output_pdf}'"
         logging.info(f"Ejecutando comando: {comando}")
         ejecutar_comando(
             comando,