a

app.py

@@ -45,7 +45,6 @@ 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
@@ -57,10 +56,7 @@ def flujo_principal(pdf_file, idioma="spa"):
         # Procesar con OCR
         crear_pdf_con_texto_incrustado(simplificado_pdf, output_pdf, idioma)
 
-        # 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
+        return output_pdf  # Devolver el archivo para descarga y vista previa
 
     except Exception as e:
         logger.error(f"Error durante el procesamiento del PDF: {str(e)}")
@@ -84,16 +80,15 @@ with gr.Blocks() as interfaz:
         pdf_vista = PDF(label="Visor PDF procesado", interactive=False)  # Salida usando PDF para visualización
         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],
-    )
+    def procesamiento_y_descarga(pdf_file, idioma):
+        """Realiza el procesamiento del PDF y devuelve tanto la vista previa como la ruta de descarga."""
+        archivo_procesado = flujo_principal(pdf_file, idioma)
+        return archivo_procesado, archivo_procesado
 
     boton_procesar.click(
-        fn=lambda x: x,
-        inputs=[pdf_vista],
-        outputs=[pdf_descarga]
+        fn=procesamiento_y_descarga,
+        inputs=[archivo_pdf, idioma_ocr],
+        outputs=[pdf_vista, pdf_descarga],
     )
 
 if __name__ == "__main__":