a

app.py

@@ -5,6 +5,7 @@ import os
 import tempfile
 import shlex
 from gradio_pdf import PDF
+import shutil
 
 # Configuración de logs
 logger = logging.getLogger(__name__)
@@ -44,6 +45,7 @@ def flujo_principal(pdf_file, idioma="spa"):
     reparado_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf").name
     simplificado_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf").name
     output_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf").name
+    final_output_path = "/mnt/data/pdf_con_ocr.pdf"  # Ubicación fija para descarga en Hugging Face Space
 
     try:
         # Reparar el PDF
@@ -55,11 +57,19 @@ def flujo_principal(pdf_file, idioma="spa"):
         # Procesar con OCR
         crear_pdf_con_texto_incrustado(simplificado_pdf, output_pdf, idioma)
 
-        return output_pdf  # Devolver el PDF final con OCR
+        # Mover el archivo final a una ubicación persistente
+        shutil.move(output_pdf, final_output_path)
+
+        return final_output_path  # Devolver el archivo para visualización y descarga
 
     except Exception as e:
         logger.error(f"Error durante el procesamiento del PDF: {str(e)}")
         raise gr.Error(f"Error al procesar el PDF: {str(e)}")
+    finally:
+        # Limpiar archivos temporales no necesarios
+        for temp_file in [reparado_pdf, simplificado_pdf]:
+            if os.path.exists(temp_file):
+                os.remove(temp_file)
 
 # Interfaz Gradio
 with gr.Blocks() as interfaz:
@@ -72,20 +82,19 @@ with gr.Blocks() as interfaz:
 
     with gr.Row():
         pdf_vista = PDF(label="Visor PDF procesado", interactive=False)  # Salida usando PDF para visualización
-        boton_procesar.click(
-            fn=flujo_principal,
-            inputs=[archivo_pdf, idioma_ocr],
-            outputs=[pdf_vista],
-        )
-
-    with gr.Row():
-        gr.Markdown("### Descargar PDF procesado con OCR")
-        pdf_descarga = gr.File(label="Descargar PDF con OCR", interactive=False)
-        boton_procesar.click(
-            fn=lambda x: x,
-            inputs=[pdf_vista],
-            outputs=[pdf_descarga]
-        )
+        pdf_descarga = gr.File(label="Descargar PDF procesado con OCR", interactive=False)
+
+    boton_procesar.click(
+        fn=flujo_principal,
+        inputs=[archivo_pdf, idioma_ocr],
+        outputs=[pdf_vista],
+    )
+
+    boton_procesar.click(
+        fn=lambda x: x,
+        inputs=[pdf_vista],
+        outputs=[pdf_descarga]
+    )
 
 if __name__ == "__main__":
     interfaz.launch()