自动提取pdf公式 ➕ 输出 LaTeX

# 创建打包脚本的主内容
script_content = """
from doc2x.extract_formula import extract_formula_imgs
from pix2text import Pix2Text
from PIL import Image
import osdef main():pdf_path = "your_file.pdf"  # 将你的PDF命名为 your_file.pdf 并放在同目录下output_dir = "formulas"os.makedirs(output_dir, exist_ok=True)print("👉 正在提取公式图像...")extract_formula_imgs(pdf_path, output_dir, dpi=300)print("✅ 公式图像提取完成，开始识别 LaTeX 公式...")p2t = Pix2Text()with open("extracted_formulas.txt", "w", encoding="utf-8") as f_out:for fname in sorted(os.listdir(output_dir)):if fname.endswith(".png"):img_path = os.path.join(output_dir, fname)latex = p2t(Image.open(img_path))print(f"{fname} → {latex}")f_out.write(f"{fname}: {latex}\\n")print("\\n🎉 所有公式识别完成，结果已保存至 extracted_formulas.txt")if __name__ == "__main__":main()
"""requirements_content = """
doc2x[pix2text]
"""# 写入打包文件
script_path = "/mnt/data/extract_formulas.py"
requirements_path = "/mnt/data/requirements.txt"with open(script_path, "w", encoding="utf-8") as f:f.write(script_content.strip())with open(requirements_path, "w", encoding="utf-8") as f:f.write(requirements_content.strip())script_path, requirements_path

# 创建打包脚本的主内容 extract_formulas.py

依赖环境文件：requirements.txt

doc2x[pix2text]

用方法（只需3步）：

1. 将你的 PDF 重命名为：your_file.pdf，放在脚本同一目录下

2. 安装环境：

   pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

3. 运行脚本：

   python extract_formulas.py

完成后会自动：

• 提取所有数学公式区域图像

• 识别为 LaTeX 格式

• 保存结果到：extracted_formulas.txt

转换成GUI工具

# 将脚本转换为简单的 tkinter GUI 应用程序
gui_script = """
import os
import tkinter as tk
from tkinter import filedialog, messagebox
from doc2x.extract_formula import extract_formula_imgs
from pix2text import Pix2Text
from PIL import Imagedef select_pdf():file_path = filedialog.askopenfilename(filetypes=[("PDF Files", "*.pdf")])if file_path:entry_pdf.delete(0, tk.END)entry_pdf.insert(0, file_path)def run_extraction():pdf_path = entry_pdf.get()if not os.path.isfile(pdf_path) or not pdf_path.endswith(".pdf"):messagebox.showerror("错误", "请选择有效的 PDF 文件。")returnoutput_dir = os.path.join(os.path.dirname(pdf_path), "formulas")os.makedirs(output_dir, exist_ok=True)messagebox.showinfo("提示", "正在提取公式图像，请稍候...")extract_formula_imgs(pdf_path, output_dir, dpi=300)p2t = Pix2Text()output_file = os.path.join(os.path.dirname(pdf_path), "extracted_formulas.txt")with open(output_file, "w", encoding="utf-8") as f_out:for fname in sorted(os.listdir(output_dir)):if fname.endswith(".png"):img_path = os.path.join(output_dir, fname)latex = p2t(Image.open(img_path))f_out.write(f"{fname}: {latex}\\n")messagebox.showinfo("完成", f"🎉 公式提取完成！结果保存在\\n{output_file}")# GUI 界面
root = tk.Tk()
root.title("PDF 公式提取工具")
root.geometry("500x150")tk.Label(root, text="选择 PDF 文件：").pack(pady=5)
entry_pdf = tk.Entry(root, width=50)
entry_pdf.pack(padx=10)
tk.Button(root, text="浏览", command=select_pdf).pack(pady=5)tk.Button(root, text="开始提取公式", command=run_extraction, bg="#4CAF50", fg="white").pack(pady=10)root.mainloop()
"""gui_script_path = "/mnt/data/pdf_formula_gui.py"
with open(gui_script_path, "w", encoding="utf-8") as f:f.write(gui_script.strip())gui_script_path

• 安装依赖（首次运行）：

  pip install doc2x[pix2text] -i https://pypi.tuna.tsinghua.edu.cn/simple

• 双击运行或使用命令：

  python pdf_formula_gui.py

• 在弹出的窗口中：

  • 选择你的 PDF 文件

  • 点击【开始提取公式】

  • 将这个图形界面程序打包成 .exe 可执行文件的详细教程，适合 Windows 用户👇

    ---

    ✅ 步骤一：安装打包工具 PyInstaller

    打开命令行，运行：

    pip install pyinstaller

    ---

    ✅ 步骤二：使用 PyInstaller 打包为 EXE

    在包含 pdf_formula_gui.py 文件的文件夹中运行以下命令：

    pyinstaller --onefile --windowed pdf_formula_gui.py

    参数解释：

  • --onefile：打包成单个 .exe 文件

  • --windowed：不显示命令行窗口（适合 GUI 应用）

  • 你可以双击它，像普通软件一样运行！

    ---

    ✅ 步骤三：获取打包好的 EXE 文件

    dist/pdf_formula_gui.exe

    自动生成：LaTeX 识别结果（extracted_formulas.txt）

自动安装依赖 .bat 文件

你还可以创建一个 install_and_run.bat 文件，内容如下：

@echo off pip install doc2x[pix2text] -i https://pypi.tuna.tsinghua.edu.cn/simple start pdf_formula_gui.exe

这样就可以发给别人用了，一键安装依赖并启动程序 🪄