a

app.py

@@ -4,10 +4,11 @@ import logging
 from PyPDF2 import PdfReader
 import tempfile
 import os
+import shlex  # Importa para manejar nombres de archivo con espacios
 
-# Configuración de logs (mejor usar un logger)
+# Configuración de logs
 logger = logging.getLogger(__name__)
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
 def ejecutar_comando(comando):
     """Ejecuta un comando de shell y maneja errores."""
@@ -23,13 +24,12 @@ def ejecutar_comando(comando):
 def crear_pdf_con_texto_incrustado(pdf_original, archivo_salida, idioma="spa"):
     """Procesa un PDF con OCR usando OCRmyPDF."""
     try:
-        ejecutar_comando(
-            f"ocrmypdf -l {idioma} --force-ocr --deskew --clean --output-type pdf {pdf_original} {archivo_salida}"
-        )
-    except RuntimeError as e:  # Capturar el error específico
+        # Usa shlex.quote para manejar espacios en los nombres de archivo
+        comando = f"ocrmypdf -l {idioma} --force-ocr --deskew --clean --output-type pdf {shlex.quote(pdf_original)} {shlex.quote(archivo_salida)}"
+        ejecutar_comando(comando)
+    except RuntimeError as e:
         raise gr.Error(str(e))  # Mostrar el error en la interfaz de Gradio
 
-
 def leer_pdf(pdf_path):
     """Extrae texto de un archivo PDF."""
     try:
@@ -46,25 +46,27 @@ def flujo_principal(pdf_file, idioma="spa"):
     if not pdf_file:
         raise gr.Error("No se subió ningún archivo.")
 
-    # Usa la ruta del archivo directamente
+    # Asegúrate de que los nombres de archivo sean seguros
     input_pdf = pdf_file.name
+    input_pdf_fixed = input_pdf.replace(" ", "_")  # Renombra temporalmente para evitar problemas con espacios
+    os.rename(input_pdf, input_pdf_fixed)
 
     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_output:
         output_pdf = temp_output.name
 
-    texto_original = leer_pdf(input_pdf)
+    texto_original = leer_pdf(input_pdf_fixed)
 
     try:
-        crear_pdf_con_texto_incrustado(input_pdf, output_pdf, idioma)
+        crear_pdf_con_texto_incrustado(input_pdf_fixed, output_pdf, idioma)
         texto_ocr = leer_pdf(output_pdf)
-        return gr.File(input_pdf, label="PDF Original"), texto_original, gr.File(output_pdf, label="PDF con OCR"), texto_ocr
+        return gr.File(input_pdf_fixed, label="PDF Original"), texto_original, gr.File(output_pdf, label="PDF con OCR"), texto_ocr
     except gr.Error as e:
-        # Limpieza en caso de error
         if os.path.exists(output_pdf):
             os.remove(output_pdf)
-        raise e  # Re-lanzar la excepción para que Gradio la maneje
+        raise e
     finally:
-        # Limpieza, incluso si hay excepciones
+        if os.path.exists(input_pdf_fixed):
+            os.remove(input_pdf_fixed)
         if os.path.exists(output_pdf):
             os.remove(output_pdf)
 
@@ -91,4 +93,4 @@ with gr.Blocks() as interfaz:
         outputs=[pdf_original_vista, texto_original, pdf_ocr_vista, texto_ocr]
     )
 
-interfaz.launch(share=True)
+interfaz.launch(share=True)