基线:全量源码首提(D1 版本控制落地,含第一轮优化 B1-B4/A1/A4/缩放修复)
This commit is contained in:
dqb-dev
@@ -0,0 +1,231 @@
|
||||
// OcrSettingsForm:OCR 引擎与 Key 设置(百度 OCR / 智谱 GLM-OCR / 千帆 / UApiPro / Umi-OCR 本地)
|
||||
// 保存 → exe 旁 baidu_ocr.json(engine/api_key/secret_key/glm_api_key/qianfan_api_key/uapis_api_key/umi_url)
|
||||
// + 热更新 OcrClient + 作废百度 token 缓存。
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Windows.Forms;
|
||||
using System.Web.Script.Serialization;
|
||||
|
||||
namespace pdfbt
|
||||
{
|
||||
public partial class OcrSettingsForm : Form
|
||||
{
|
||||
private readonly OcrClient _ocr;
|
||||
|
||||
public OcrSettingsForm(OcrClient ocr)
|
||||
{
|
||||
InitializeComponent();
|
||||
_ocr = ocr;
|
||||
|
||||
// 回填现有配置(默认 Key 在 Designer 的 Text 里,密码遮罩显示;保存过则文件值优先)
|
||||
string engine = "baidu";
|
||||
try
|
||||
{
|
||||
string cfgPath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "baidu_ocr.json");
|
||||
if (File.Exists(cfgPath))
|
||||
{
|
||||
var jsv = new JavaScriptSerializer().Deserialize<Dictionary<string, string>>(File.ReadAllText(cfgPath));
|
||||
if (jsv != null)
|
||||
{
|
||||
if (jsv.ContainsKey("engine")) engine = jsv["engine"] ?? "baidu";
|
||||
txtApiKey.Text = jsv.ContainsKey("api_key") ? (jsv["api_key"] ?? "") : "";
|
||||
txtSecretKey.Text = jsv.ContainsKey("secret_key") ? (jsv["secret_key"] ?? "") : "";
|
||||
txtGlmKey.Text = jsv.ContainsKey("glm_api_key") ? (jsv["glm_api_key"] ?? "") : "";
|
||||
txtQianfanKey.Text = jsv.ContainsKey("qianfan_api_key") ? (jsv["qianfan_api_key"] ?? "") : "";
|
||||
txtUapisKey.Text = jsv.ContainsKey("uapis_api_key") ? (jsv["uapis_api_key"] ?? "") : "";
|
||||
txtUmiUrl.Text = jsv.ContainsKey("umi_url") && !string.IsNullOrWhiteSpace(jsv["umi_url"])
|
||||
? jsv["umi_url"] : "http://127.0.0.1:1224";
|
||||
}
|
||||
}
|
||||
}
|
||||
catch { }
|
||||
|
||||
cboEngine.SelectedIndex =
|
||||
engine == "glm" ? 1 :
|
||||
engine == "qianfan" ? 2 :
|
||||
engine == "uapis" ? 3 :
|
||||
engine == "umi" ? 4 : 0;
|
||||
UpdateEnginePanels();
|
||||
}
|
||||
|
||||
/// <summary>按引擎下拉显示对应的输入区(百度=两格;智谱/千帆/UApiPro=Key一格;Umi-OCR=服务地址一格)。</summary>
|
||||
private void UpdateEnginePanels()
|
||||
{
|
||||
int sel = cboEngine.SelectedIndex;
|
||||
bool glm = sel == 1;
|
||||
bool qf = sel == 2;
|
||||
bool ua = sel == 3;
|
||||
bool umi = sel == 4;
|
||||
bool baidu = !glm && !qf && !ua && !umi;
|
||||
lblApiKey.Visible = baidu;
|
||||
txtApiKey.Visible = baidu;
|
||||
lblSecretKey.Visible = baidu;
|
||||
txtSecretKey.Visible = baidu;
|
||||
lblGlmKey.Visible = glm;
|
||||
txtGlmKey.Visible = glm;
|
||||
lblQianfanKey.Visible = qf;
|
||||
txtQianfanKey.Visible = qf;
|
||||
lblUapisKey.Visible = ua;
|
||||
txtUapisKey.Visible = ua;
|
||||
lblUmiUrl.Visible = umi;
|
||||
txtUmiUrl.Visible = umi;
|
||||
btnUmi下载.Visible = umi;
|
||||
btnUmi检测.Visible = umi;
|
||||
lblUmi状态.Visible = umi;
|
||||
if (umi) RefreshUmiStatus();
|
||||
}
|
||||
|
||||
/// <summary>刷新 Umi 本体三态状态显示。</summary>
|
||||
private void RefreshUmiStatus()
|
||||
{
|
||||
try { lblUmi状态.Text = UmiOcrHost.DetectText(); } catch { }
|
||||
}
|
||||
|
||||
/// <summary>检测 OCR 本体:刷新状态(未安装/已安装未运行/运行中)。</summary>
|
||||
private void btnUmi检测_Click(object sender, EventArgs e)
|
||||
{
|
||||
RefreshUmiStatus();
|
||||
MessageBox.Show(this, UmiOcrHost.DetectText(), "检测 OCR 本体",
|
||||
MessageBoxButtons.OK, MessageBoxIcon.Information);
|
||||
}
|
||||
|
||||
/// <summary>下载 OCR 本体:网盘直链(~179MB)→解压到程序目录 Umi-OCR\→验证→删 zip。</summary>
|
||||
private void btnUmi下载_Click(object sender, EventArgs e)
|
||||
{
|
||||
if (UmiOcrHost.IsInstalled)
|
||||
{
|
||||
var r = MessageBox.Show(this, "OCR 本体已存在,重新下载将覆盖安装。继续吗?", "下载 OCR 本体",
|
||||
MessageBoxButtons.OKCancel, MessageBoxIcon.Question);
|
||||
if (r != DialogResult.OK) return;
|
||||
}
|
||||
btnUmi下载.Enabled = false;
|
||||
btnUmi检测.Enabled = false;
|
||||
lblUmi状态.ForeColor = System.Drawing.Color.DimGray;
|
||||
lblUmi状态.Text = "正在下载… 0%";
|
||||
int lastPct = -1;
|
||||
UmiOcrHost.DownloadAndInstallAsync(pct =>
|
||||
{
|
||||
if (pct == lastPct) return;
|
||||
lastPct = pct;
|
||||
try { this.BeginInvoke((MethodInvoker)delegate
|
||||
{
|
||||
lblUmi状态.Text = pct >= 100 ? "下载完成,正在解压安装…" : "正在下载… " + pct + "%";
|
||||
}); } catch { }
|
||||
}).ContinueWith(t =>
|
||||
{
|
||||
try { this.BeginInvoke((MethodInvoker)delegate
|
||||
{
|
||||
btnUmi下载.Enabled = true;
|
||||
btnUmi检测.Enabled = true;
|
||||
lblUmi状态.ForeColor = System.Drawing.SystemColors.GrayText;
|
||||
if (t.IsFaulted)
|
||||
{
|
||||
Exception root = t.Exception;
|
||||
while (root is AggregateException && root.InnerException != null) root = root.InnerException;
|
||||
lblUmi状态.Text = "下载/安装失败: " + root.Message;
|
||||
MessageBox.Show(this, "下载/安装失败:\n" + root.Message, "下载 OCR 本体",
|
||||
MessageBoxButtons.OK, MessageBoxIcon.Warning);
|
||||
}
|
||||
else
|
||||
{
|
||||
RefreshUmiStatus();
|
||||
MessageBox.Show(this, "OCR 本体安装完成!\n" + UmiOcrHost.DetectText() +
|
||||
"\n\n选 Umi-OCR 引擎保存后即可使用(识别时自动启动)。",
|
||||
"下载 OCR 本体", MessageBoxButtons.OK, MessageBoxIcon.Information);
|
||||
}
|
||||
}); } catch { }
|
||||
});
|
||||
}
|
||||
|
||||
private void cboEngine_SelectedIndexChanged(object sender, EventArgs e)
|
||||
{
|
||||
UpdateEnginePanels();
|
||||
}
|
||||
|
||||
/// <summary>帮助:全部引擎的 Key 获取方式与本地 OCR 设置流程。</summary>
|
||||
private void btn帮助_Click(object sender, EventArgs e)
|
||||
{
|
||||
string help =
|
||||
"════════ OCR 识别引擎使用帮助 ════════\r\n" +
|
||||
"\r\n" +
|
||||
"【1. 百度 OCR 高精度(含位置)】 默认引擎,已内置 Key 可直接用\r\n" +
|
||||
" · 获取自己的 Key(更稳,免费额度足够日常):\r\n" +
|
||||
" ① 打开 console.baidu.com 登录百度账号\r\n" +
|
||||
" ② 进入\"产品服务 / 文字识别\"→ 创建应用\r\n" +
|
||||
" ③ 勾选\"文字识别\"全部接口,创建完成\r\n" +
|
||||
" ④ 在应用详情里复制 API Key 和 Secret Key 填入本窗口\r\n" +
|
||||
"\r\n" +
|
||||
"【2. 智谱 GLM-OCR(版面解析/手写强)】\r\n" +
|
||||
" · 到 open.bigmodel.cn 注册并实名 → API Keys 页面新建密钥\r\n" +
|
||||
" · 把密钥填入\"智谱 API Key\"\r\n" +
|
||||
" · 计费约 0.2 元/百万 tokens,适合手写体/复杂版面\r\n" +
|
||||
"\r\n" +
|
||||
"【3. 百度 PaddleOCR-VL(千帆版面解析)】\r\n" +
|
||||
" · 到 console.bce.baidu.com → 千帆 ModelBuilder → 应用管理\r\n" +
|
||||
" · 创建应用后复制 API Key 填入\"千帆 API Key\"\r\n" +
|
||||
" · 需在千帆控制台开通 PaddleOCR-VL 模型服务\r\n" +
|
||||
"\r\n" +
|
||||
"【4. UApiPro 通用OCR(免费积分/带位置)】\r\n" +
|
||||
" · 到 uapis.cn 注册账号 → 控制台 → API Keys → 新建\r\n" +
|
||||
" · 把 Key 填入\"UApiPro Key\"\r\n" +
|
||||
" · 注册送免费积分(约 4 积分/次识别),小量使用零成本\r\n" +
|
||||
"\r\n" +
|
||||
"【5. Umi-OCR 本地引擎(离线免费,无限次)】\r\n" +
|
||||
" · 本地引擎无需任何 Key,识别全在本机完成,数据不出电脑\r\n" +
|
||||
" · 设置流程:\r\n" +
|
||||
" ① 选本引擎后点\"下载 OCR 本体\"(约 179MB,自动下载安装)\r\n" +
|
||||
" ② 点\"检测 OCR 本体\"确认状态为\"运行中\"或\"已安装\"\r\n" +
|
||||
" ③ 点\"保存\"完成,之后识别时自动启动、关窗自动关闭\r\n" +
|
||||
" · 服务地址保持默认 http://127.0.0.1:1224 即可\r\n" +
|
||||
" (只有你在 Umi-OCR 软件里改过 HTTP 端口才需要改这里)\r\n" +
|
||||
"\r\n" +
|
||||
"════════════════════════════════\r\n" +
|
||||
"提示: Key 全部以密码方式显示,点\"保存\"后立即生效;\r\n" +
|
||||
"所有 Key 统一保存在程序目录 baidu_ocr.json。";
|
||||
MessageBox.Show(this, help, "OCR 使用帮助",
|
||||
MessageBoxButtons.OK, MessageBoxIcon.Information);
|
||||
}
|
||||
|
||||
private void btn确定_Click(object sender, EventArgs e)
|
||||
{
|
||||
try
|
||||
{
|
||||
string cfgPath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "baidu_ocr.json");
|
||||
string engineValue =
|
||||
cboEngine.SelectedIndex == 1 ? "glm" :
|
||||
cboEngine.SelectedIndex == 2 ? "qianfan" :
|
||||
cboEngine.SelectedIndex == 3 ? "uapis" :
|
||||
cboEngine.SelectedIndex == 4 ? "umi" : "baidu";
|
||||
var jsv = new JavaScriptSerializer();
|
||||
File.WriteAllText(cfgPath, jsv.Serialize(new Dictionary<string, string>
|
||||
{
|
||||
{ "engine", engineValue },
|
||||
{ "api_key", txtApiKey.Text.Trim() },
|
||||
{ "secret_key", txtSecretKey.Text.Trim() },
|
||||
{ "glm_api_key", txtGlmKey.Text.Trim() },
|
||||
{ "qianfan_api_key", txtQianfanKey.Text.Trim() },
|
||||
{ "uapis_api_key", txtUapisKey.Text.Trim() },
|
||||
{ "umi_url", string.IsNullOrWhiteSpace(txtUmiUrl.Text.Trim()) ? "http://127.0.0.1:1224" : txtUmiUrl.Text.Trim() }
|
||||
}));
|
||||
// 热更新引擎/Key + 作废百度 token 缓存(下次识别重新换取)
|
||||
if (_ocr != null)
|
||||
{
|
||||
_ocr.LoadConfig();
|
||||
try { File.Delete(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "baidu_token.json")); } catch { }
|
||||
}
|
||||
// 选了 Umi-OCR 但还没装本体:提示引导下载(不阻断保存)
|
||||
if (engineValue == "umi" && !UmiOcrHost.IsInstalled)
|
||||
{
|
||||
MessageBox.Show(this, "Umi-OCR 本体尚未安装。\n请点\"下载 OCR 本体\"完成安装后即可使用。",
|
||||
"OCR 设置", MessageBoxButtons.OK, MessageBoxIcon.Information);
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
MessageBox.Show(this, "保存失败: " + ex.Message, "OCR 设置",
|
||||
MessageBoxButtons.OK, MessageBoxIcon.Warning);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user