PDF-Extract-Kit代码实例：与Flask框架集成

1. 引言

1.1 业务场景描述

在现代文档处理系统中，PDF文件的智能信息提取已成为科研、教育、出版等领域的核心需求。然而，现有的通用OCR工具往往难以满足对复杂版式（如公式、表格）的高精度识别要求。为此，PDF-Extract-Kit应运而生——这是一个由开发者“科哥”基于深度学习技术二次开发构建的PDF智能提取工具箱，集成了布局检测、公式识别、OCR文字提取和表格解析等多项功能。

本文将重点介绍如何将PDF-Extract-Kit的核心能力通过Flask 框架进行 Web 化封装，实现一个可远程调用、支持多用户并发访问的服务接口，从而为后续集成到企业级文档管理系统提供技术支撑。

1.2 痛点分析

传统本地运行的 GUI 工具存在以下局限性： - 难以与其他系统对接（如内容管理系统 CMS） - 不支持自动化批处理流程 - 多人协作时需重复部署环境 - 缺乏统一的日志与权限管理机制

因此，将 PDF-Extract-Kit 封装为 RESTful API 接口服务，是提升其工程化价值的关键一步。

1.3 方案预告

本文将展示如何使用 Flask 构建一个轻量级 Web 服务，完成以下目标： - 实现 PDF 文件上传与异步处理 - 调用 PDF-Extract-Kit 各模块（布局检测、OCR、公式识别等） - 返回结构化结果 JSON 及可视化图像 - 支持跨域请求（CORS），便于前端集成

2. 技术方案选型

2.1 为什么选择 Flask？

对比项	Flask	FastAPI	Django
轻量化	✅ 极简设计	✅ 异步支持	❌ 功能臃肿
学习成本	✅ 低	⚠️ 中等	⚠️ 较高
扩展性	✅ 插件丰富	✅ 自动生成文档	✅ 完整MVC
性能	⚠️ 同步为主	✅ 原生异步	⚠️ 同步阻塞
适用场景	快速原型/微服务	高并发API	全栈应用

📌结论：对于PDF-Extract-Kit这类以模型推理为主的任务，I/O 主要集中在文件读写与模型加载，Flask 的简洁性和生态成熟度使其成为最佳选择。

2.2 核心依赖组件

Flask==2.3.3 Werkzeug==2.3.7 Pillow==9.5.0 pdf2image==1.16.3 numpy==1.24.3 torch==2.0.1 # 若使用YOLOv8或LaTeX识别模型

此外，项目需预先安装PDF-Extract-Kit的本地 Python 包或直接引用其源码目录。

3. 实现步骤详解

3.1 目录结构规划

为保证工程清晰，建议采用如下结构：

flask-pdf-extract/ ├── app.py # Flask主程序 ├── config.py # 配置参数 ├── utils/ │ ├── pdf_converter.py # PDF转图像 │ └── result_handler.py # 结果打包返回 ├── modules/ │ ├── layout_detector.py # 布局检测封装 │ ├── formula_recognizer.py # 公式识别封装 │ └── table_parser.py # 表格解析封装 ├── uploads/ # 用户上传文件 ├── outputs/ # 处理结果输出 └── static/ # 可视化图片存放

3.2 Flask 主服务搭建

核心代码实现

# app.py from flask import Flask, request, jsonify, send_from_directory from werkzeug.utils import secure_filename import os import uuid from datetime import datetime from modules.layout_detector import run_layout_detection from modules.formula_recognizer import recognize_formulas from modules.table_parser import parse_table app = Flask(__name__) app.config['UPLOAD_FOLDER'] = 'uploads' app.config['OUTPUT_FOLDER'] = 'outputs' app.config['ALLOWED_EXTENSIONS'] = {'pdf', 'png', 'jpg', 'jpeg'} def allowed_file(filename): return '.' in filename and \ filename.rsplit('.', 1)[1].lower() in app.config['ALLOWED_EXTENSIONS'] @app.route('/api/v1/layout/detect', methods=['POST']) def api_layout_detect(): if 'file' not in request.files: return jsonify({'error': 'No file uploaded'}), 400 file = request.files['file'] if file.filename == '': return jsonify({'error': 'Empty filename'}), 400 if file and allowed_file(file.filename): filename = secure_filename(file.filename) unique_id = str(uuid.uuid4())[:8] save_name = f"{unique_id}_{filename}" file_path = os.path.join(app.config['UPLOAD_FOLDER'], save_name) file.save(file_path) try: # 调用PDF-Extract-Kit的布局检测模块 result_json, image_path = run_layout_detection(file_path) return jsonify({ 'status': 'success', 'task_id': unique_id, 'timestamp': datetime.now().isoformat(), 'result': result_json, 'visual_image_url': f"/static/{os.path.basename(image_path)}" }) except Exception as e: return jsonify({'error': str(e)}), 500 return jsonify({'error': 'Invalid file type'}), 400 @app.route('/static/<filename>') def serve_image(filename): return send_from_directory('static', filename) if __name__ == '__main__': os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True) os.makedirs(app.config['OUTPUT_FOLDER'], exist_ok=True) os.makedirs('static', exist_ok=True) app.run(host='0.0.0.0', port=5000, debug=False)

代码解析

使用secure_filename防止路径穿越攻击
生成唯一task_id用于追踪任务
封装异常捕获，避免服务崩溃
提供/static/<filename>接口供前端获取可视化结果图

3.3 模块化功能封装示例

布局检测模块封装

# modules/layout_detector.py import subprocess import json import os def run_layout_detection(input_path: str): """ 调用PDF-Extract-Kit的布局检测功能 返回: (json_result_dict, visual_image_path) """ output_dir = "outputs/layout_detection" os.makedirs(output_dir, exist_ok=True) cmd = [ "python", "webui/app.py", "--input", input_path, "--output", output_dir, "--task", "layout", "--img_size", "1024", "--conf_thres", "0.25" ] result = subprocess.run(cmd, capture_output=True, text=True) if result.returncode != 0: raise RuntimeError(f"Layout detection failed: {result.stderr}") # 假设输出为固定命名格式 json_file = os.path.join(output_dir, "layout_result.json") img_file = os.path.join(output_dir, "layout_visual.png") if not os.path.exists(json_file): raise FileNotFoundError("Layout result JSON not found") with open(json_file, 'r', encoding='utf-8') as f: result_data = json.load(f) # 复制可视化图片到static目录以便Web访问 static_img = f"static/layout_{os.path.basename(input_path)}.png" from shutil import copyfile copyfile(img_file, static_img) return result_data, static_img

💡说明：此处通过subprocess调用原生webui/app.py脚本，也可改为直接导入其内部函数以提高效率。

3.4 前后端交互设计

请求示例（cURL）

curl -X POST http://localhost:5000/api/v1/layout/detect \ -F "file=@sample.pdf" \ -H "Content-Type: multipart/form-data"

成功响应示例

{ "status": "success", "task_id": "a1b2c3d4", "timestamp": "2025-04-05T10:23:45.123456", "result": { "elements": [ {"type": "text", "bbox": [100, 120, 300, 150], "text": "Introduction"}, {"type": "table", "bbox": [80, 200, 500, 400]} ] }, "visual_image_url": "/static/layout_sample.pdf.png" }