a

app.py

@@ -41,30 +41,17 @@ def flujo_principal(pdf_file, idioma="spa"):
     if not pdf_file:
         raise gr.Error("No se subió ningún archivo.")
 
-    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
+    with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as reparado_pdf:
+        reparar_pdf(pdf_file, reparado_pdf.name)
 
-    try:
-        # Reparar el PDF
-        reparar_pdf(pdf_file, reparado_pdf)
-
-        # Simplificar el PDF
-        simplificar_pdf(reparado_pdf, simplificado_pdf)
-
-        # Procesar con OCR
-        crear_pdf_con_texto_incrustado(simplificado_pdf, output_pdf, idioma)
-
-        return output_pdf, output_pdf # Devolver dos veces para visor y descarga
+    with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as simplificado_pdf:
+        simplificar_pdf(reparado_pdf.name, simplificado_pdf.name)
+        os.remove(reparado_pdf.name)
 
-    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)
+    with tempfile.NamedTemporaryFile(suffix=".pdf") as output_pdf:
+        crear_pdf_con_texto_incrustado(simplificado_pdf.name, output_pdf.name, idioma)
+        os.remove(simplificado_pdf.name)
+        return output_pdf.name, output_pdf.name  # Retorna la ruta dos veces
 
 
 # Interfaz Gradio
@@ -77,8 +64,8 @@ with gr.Blocks() as interfaz:
         boton_procesar = gr.Button("Procesar OCR")
 
     with gr.Row():
-        pdf_vista = PDF(label="Visor PDF")  # Corrección: usar PDF
-        pdf_descarga = gr.File(label="Descargar PDF procesado", visible=False)
+        pdf_vista = PDF(label="Visor PDF")
+        pdf_descarga = gr.File(label="Descargar PDF procesado", visible=True)
 
     boton_procesar.click(
         fn=flujo_principal,