使用 Python 总结 PDF —— PDF 总结 API 教程

前提条件

• Python 3.8+ 以及 requests 库（pip install requests）。
• API 密钥：创建一个免费账户并从控制台复制你的密钥。建议将其存储在环境变量中，而不是直接硬编码在代码里：

export PDF_SUMMARIZE_KEY="sk_live_..."

在 Python 中生成你的第一个 PDF 总结

import os
import requests

API_URL = "https://pdf-summarize.com/api/summarize.php"
API_KEY = os.environ["PDF_SUMMARIZE_KEY"]

def summarize_pdf(path, length="medium", fmt="bullets", lang="auto"):
    with open(path, "rb") as f:
        resp = requests.post(
            API_URL,
            headers={"Authorization": f"Bearer {API_KEY}"},
            files={"file": (os.path.basename(path), f, "application/pdf")},
            data={"length": length, "format": fmt, "outputLang": lang, "includeKeywords": "1"},
            timeout=180,
        )
    data = resp.json()
    if resp.status_code != 200 or not data.get("success"):
        raise RuntimeError(f"{resp.status_code}: {data.get('error', data)}")
    return data

result = summarize_pdf("annual-report.pdf")
print(result["summary"])
print("页数：", result["page_count"], "| 核心要点：", result.get("keywords"))

响应内容

{
  "success": true,
  "summary": "- 收入增长 6% 至...",
  "keywords": "revenue, margin, guidance, ...",
  "page_count": 48,
  "original_length": 182340,
  "summary_length": 1210,
  "processing_time": 6.214,
  "requests_remaining": 4987
}

requests_remaining 是你当月剩余的额度（-1 表示无限）。

流式传输总结

对于长文档或交互式应用，可以在生成总结时进行流式传输：

import json

def stream_summary(path):
    with open(path, "rb") as f:
        with requests.post(
            API_URL + "?stream=1",
            headers={"Authorization": f"Bearer {API_KEY}"},
            files={"file": (os.path.basename(path), f, "application/pdf")},
            data={"length": "long", "format": "mixed"},
            stream=True, timeout=300,
        ) as resp:
            resp.raise_for_status()
            for line in resp.iter_lines(decode_unicode=True):
                if not line or not line.startswith("data: "):
                    continue
                payload = line[6:]
                if payload == "[DONE]":
                    break
                event = json.loads(payload)
                if "error" in event:
                    raise RuntimeError(event["error"])
                if event.get("type") == "summary":
                    print(event["content"], end="", flush=True)

stream_summary("research-paper.pdf")

批量处理 PDF 文件夹

import pathlib, time

out_dir = pathlib.Path("summaries")
out_dir.mkdir(exist_ok=True)

for pdf in sorted(pathlib.Path("pdfs").glob("*.pdf")):
    target = out_dir / (pdf.stem + ".md")
    if target.exists():
        continue                      # 支持断点续传
    for attempt in range(3):
        try:
            data = summarize_pdf(str(pdf), length="long", fmt="mixed")
            target.write_text(data["summary"], encoding="utf-8")
            print("成功", pdf.name, data["page_count"], "页")
            break
        except RuntimeError as e:
            if str(e).startswith("429"):
                time.sleep(30 * (attempt + 1))   # 触及速率限制：退避重试
            else:
                print("失败", pdf.name, e)
                break

错误处理

提示

• 为长 PDF 设置慷慨的超时时间（2-3 分钟）。
• 通过文件哈希缓存结果，避免为同一文档重复付费。
• 将密钥保存在服务器端 —— 切勿将其打包在前端或移动端应用中。
• 完整参数参考：请查阅 API 文档。

常见问题