feat: 添加 Web 界面和项目文档

新增功能:
- 添加 Streamlit Web 界面 (src/app.py),支持批量上传和在线预览
- 添加 README.md,包含项目介绍、部署方案和配置要求

依赖更新:
- 锁定 PaddleOCR 2.x 版本以确保稳定性
- 新增 streamlit 依赖

部署方案:
- 内网服务器部署
- Docker 容器化部署
- systemd 系统服务

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
root
2026-02-12 07:19:06 +00:00
parent e2cb608845
commit 647a04d132
3 changed files with 237 additions and 2 deletions

146
README.md Normal file
View File

@@ -0,0 +1,146 @@
# 信封信息提取系统
工厂环境下信封背面信息的自动化提取与结构化录入工具。
## 功能特性
- 自动识别信封图片中的文字信息
- 结构化提取:编号、邮编、地址、联系人、电话
- 支持批量处理,结果导出为 Excel
- 提供 Web 界面,操作简单
## 系统要求
| 项目 | 最低配置 | 推荐配置 |
|------|----------|----------|
| CPU | 4 核 | 8 核 |
| 内存 | 4 GB | 8 GB |
| 硬盘 | 2 GB | 5 GB |
| 系统 | Ubuntu 20.04 / Windows 10 | Ubuntu 22.04 |
| Python | 3.8 | 3.10 |
## 快速开始
### 1. 安装依赖
```bash
# Ubuntu 需要安装系统依赖
sudo apt-get install -y libgl1-mesa-glx libglib2.0-0
# 安装 Python 依赖
pip install -r requirements.txt
```
### 2. 运行方式
**命令行批处理**
```bash
# 将图片放入 data/input/ 目录
python src/main.py
# 结果保存在 data/output/result.xlsx
```
**Web 界面**
```bash
streamlit run src/app.py --server.port 8501
# 浏览器访问 http://localhost:8501
```
## 部署方案
### 方案一:内网服务器部署(推荐)
适合多人使用,有内网环境的工厂。
```bash
# 启动服务(监听所有网卡)
streamlit run src/app.py --server.address 0.0.0.0 --server.port 8501
# 工人通过浏览器访问: http://服务器IP:8501
```
### 方案二Docker 容器化部署
适合需要隔离环境或快速部署的场景。
```bash
# 构建镜像
docker build -t envelope-ocr .
# 运行容器
docker run -d -p 8501:8501 --name envelope-ocr envelope-ocr
```
Dockerfile:
```dockerfile
FROM python:3.10-slim
RUN apt-get update && apt-get install -y libgl1-mesa-glx libglib2.0-0 && rm -rf /var/lib/apt/lists/*
WORKDIR /app
COPY . .
RUN pip install --no-cache-dir -r requirements.txt
EXPOSE 8501
CMD ["streamlit", "run", "src/app.py", "--server.address", "0.0.0.0"]
```
### 方案三:系统服务(开机自启)
适合长期稳定运行的生产环境。
创建服务文件 `/etc/systemd/system/envelope-ocr.service`:
```ini
[Unit]
Description=Envelope OCR Service
After=network.target
[Service]
User=www-data
WorkingDirectory=/opt/post-ocr
ExecStart=/usr/bin/streamlit run src/app.py --server.address 0.0.0.0 --server.port 8501
Restart=always
RestartSec=5
[Install]
WantedBy=multi-user.target
```
启用服务:
```bash
sudo systemctl daemon-reload
sudo systemctl enable envelope-ocr
sudo systemctl start envelope-ocr
```
## 目录结构
```
post-ocr/
├── data/
│ ├── input/ # 原始图片存放处
│ └── output/ # 结果 Excel 及处理日志
├── src/
│ ├── main.py # 命令行入口
│ ├── app.py # Web 界面
│ └── processor.py # 核心处理逻辑
├── requirements.txt
└── README.md
```
## 技术栈
- OCR 引擎: PaddleOCR 2.10 (PP-OCRv4)
- Web 框架: Streamlit
- 数据处理: Pandas
## 常见问题
**Q: 识别准确率不高怎么办?**
- 确保图片清晰、光线充足
- 避免图片倾斜或模糊
- 手写字体识别率较低,建议使用印刷体
**Q: 处理速度慢?**
- 首次运行需下载模型(约 200MB
- 有 GPU 可安装 paddlepaddle-gpu 加速
- 批量处理时建议使用命令行模式

View File

@@ -1,6 +1,7 @@
paddleocr
paddlepaddle
paddleocr>=2.6,<3
paddlepaddle>=2.5,<3
pandas
openpyxl
pydantic
tqdm
streamlit

88
src/app.py Normal file
View File

@@ -0,0 +1,88 @@
import os
import tempfile
import pandas as pd
import streamlit as st
from paddleocr import PaddleOCR
from processor import extract_info, save_to_excel
os.environ["PADDLE_PDX_DISABLE_MODEL_SOURCE_CHECK"] = "True"
st.set_page_config(page_title="信封信息提取系统", page_icon="📮", layout="wide")
st.title("📮 信封信息提取系统")
@st.cache_resource
def load_ocr():
return PaddleOCR(use_textline_orientation=True, lang="ch", show_log=False)
ocr = load_ocr()
def process_image(image_file):
"""处理单张图片"""
with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp:
tmp.write(image_file.getvalue())
tmp_path = tmp.name
try:
result = ocr.ocr(tmp_path, cls=False)
ocr_texts = []
if result and result[0]:
for line in result[0]:
if line and len(line) >= 2:
ocr_texts.append(line[1][0])
return extract_info(ocr_texts), ocr_texts
finally:
os.unlink(tmp_path)
# 文件上传
uploaded_files = st.file_uploader(
"上传信封图片(支持批量)",
type=["jpg", "jpeg", "png", "bmp"],
accept_multiple_files=True,
)
if uploaded_files:
all_records = []
progress = st.progress(0)
status = st.empty()
for i, file in enumerate(uploaded_files):
status.text(f"正在处理: {file.name}")
record, raw_texts = process_image(file)
record["文件名"] = file.name
all_records.append(record)
progress.progress((i + 1) / len(uploaded_files))
status.text("处理完成!")
# 显示结果表格
df = pd.DataFrame(all_records)
cols = ["文件名", "编号", "邮编", "地址", "联系人/单位名", "电话"]
df = df.reindex(columns=cols)
st.subheader("📋 提取结果")
st.dataframe(df, use_container_width=True)
# 下载按钮
output_path = tempfile.mktemp(suffix=".xlsx")
df.to_excel(output_path, index=False)
with open(output_path, "rb") as f:
st.download_button(
label="📥 下载 Excel",
data=f,
file_name="信封提取结果.xlsx",
mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
)
os.unlink(output_path)
# 预览图片和识别详情
with st.expander("🔍 查看识别详情"):
cols = st.columns(min(3, len(uploaded_files)))
for i, file in enumerate(uploaded_files):
with cols[i % 3]:
st.image(file, caption=file.name, use_container_width=True)
st.json(all_records[i])