跳转至

kg-book-tool

将电子书（PDF / EPUB / DOCX / PPTX / XLSX / HTML / Markdown / 图片）自动转换为交互式知识图谱。

PyPI version Python License: MIT Downloads CI Release Docs

特性

  • 📚 多格式支持 — PDF / EPUB / DOCX / PPTX / XLSX / HTML / Markdown / TXT
  • 🖼️ OCR 回退 — 自动检测图像型 PDF 并调用 EasyOCR（中文支持）
  • 🔤 编码自动检测 — UTF-8 / GBK / GB2312 / GB18030 / Big5 / Latin-1
  • 📖 参考文献识别 — 自动将"参考文献"小节的编号条目提取为实体
  • 🌐 桌面应用 — Tkinter 原生 GUI
  • 🎯 三视图阅读器 — 实体列表、思维导图、原文对照
  • 🪄 LLM 摘要 — 集成 Ollama，支持 qwen2.5 等模型

安装

pip install kg-book-tool

快速开始

# 命令行
kg-build book.pdf --output data.json --js data.js

# 启用 LLM 摘要
kg-build book.epub --summary --model qwen2.5:7b

from kg_core import KnowledgeGraphBuilder

builder = KnowledgeGraphBuilder("book.pdf", model_name="qwen2.5:3b")
builder.load_book()
builder.extract_entities()
builder.extract_definitions_and_paragraphs()
builder.build_relations()
data = builder.build()

builder.save_json("data.json")
builder.save_js("data.js")
builder.save_md("data.md", book_name="我的书")

链接

许可

MIT License