Python 智能 PDF 文档助手开发指南

2.3 获取 OpenAI API Key

访问 OpenAI 官网

注册/登录账号

进入 API 管理页面

创建新的 API 密钥

复制并保存 Key

创建.env 文件

配置 API_KEY

# 创建 .env 文件
echo OPENAI_API_KEY=your_api_key_here > .env
echo OPENAI_BASE_URL=https://api.openai.com/v1 >> .env

📝 项目结构设计

pdf_ai_assistant/ main.py 主程序入口 config.py 配置文件 utils/ 工具函数 pdf_reader.py PDF 读取 ai_client.py AI 客户端 output.py 输出格式化 requirements.txt 依赖列表 .env 环境变量 README.md 项目说明

💻 核心代码实现

3.1 配置文件 (config.py)

""" 配置文件 - 管理所有配置项 """
import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

class Config:
    """应用配置类"""
    # OpenAI 配置
    OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
    OPENAI_BASE_URL = os.getenv("OPENAI_BASE_URL","https://api.openai.com/v1")
    OPENAI_MODEL = os.getenv("OPENAI_MODEL","gpt-3.5-turbo")
    
    # 应用配置
    MAX_FILE_SIZE = 10*1024*1024 # 10MB
    CHUNK_SIZE = 1000 # 每次处理的字符数
    MAX_TOKENS = 2000 # AI 最大响应长度
    
    # 颜色配置
    COLORS = {
        "header": "\033[95m", # 紫色
        "okblue": "\033[94m", # 蓝色
        "okgreen": "\033[92m", # 绿色
        "warning": "\033[93m", # 黄色
        "fail": "\033[91m", # 红色
        "endc": "\033[0m", # 结束
    }
    
    @classmethod
    def color_print(cls, color_type, message):
        """彩色打印"""
        color = cls.COLORS.get(color_type, "")
        print(f"{color}{message}{cls.COLORS['endc']}")

# 验证配置
if not Config.OPENAI_API_KEY:
    Config.color_print("fail", "❌ 请在.env 文件中设置 OPENAI_API_KEY")
    exit(1)

3.2 PDF 读取模块 (pdf_reader.py)

""" PDF 文档读取模块 """
import pdfplumber
from tqdm import tqdm
from config import Config

class PDFReader:
    """PDF 文档读取器"""
    def __init__(self, file_path):
        self.file_path = file_path
        self.content = ""
        self.page_count = 0

    def validate_file(self):
        """验证文件"""
        if not self.file_path.endswith('.pdf'):
            raise ValueError("❌ 仅支持 PDF 文件格式")
        import os
        file_size = os.path.getsize(self.file_path)
        if file_size > Config.MAX_FILE_SIZE:
            raise ValueError(f"❌ 文件过大，最大支持{Config.MAX_FILE_SIZE//1024//1024}MB")

    def read_pdf(self):
        """读取 PDF 内容"""
        try:
            self.validate_file()
            Config.color_print("okblue", f"📖 正在读取文件：{self.file_path}")
            with pdfplumber.open(self.file_path) as pdf:
                self.page_count = len(pdf.pages)
                Config.color_print("okgreen", f"📄 共 {self.page_count} 页")
                # 使用进度条
                for page in tqdm(pdf.pages, desc="读取进度", unit="页"):
                    text = page.extract_text()
                    if text:
                        self.content += text + "\n\n"
            Config.color_print("okgreen", f"✅ 读取完成！共 {len(self.content)} 字符")
            return self.content
        except Exception as e:
            Config.color_print("fail", f"❌ 读取失败：{str(e)}")
            raise

    def get_summary(self):
        """获取文档摘要（前 500 字）"""
        return self.content[:500]+"..." if len(self.content)>500 else self.content

3.3 AI 客户端模块 (ai_client.py)

""" OpenAI API 客户端模块 """
from openai import OpenAI
from config import Config
import time

class AIClient:
    """AI 助手客户端"""
    def __init__(self):
        self.client = OpenAI(
            api_key=Config.OPENAI_API_KEY,
            base_url=Config.OPENAI_BASE_URL
        )

    def ask(self, question, context=""):
        """向 AI 提问"""
        try:
            # 构建消息
            messages = [
                {"role":"system","content":"你是一个专业的文档分析助手。"},
                {"role":"user","content":f"文档内容：\n{context}\n\n问题：{question}"}
            ]
            Config.color_print("okblue","🤖 AI 正在思考...")
            # 调用 API
            start_time = time.time()
            response = self.client.chat.completions.create(
                model=Config.OPENAI_MODEL,
                messages=messages,
                max_tokens=Config.MAX_TOKENS,
                stream=True # 启用流式输出
            )
            # 流式输出
            answer = ""
            Config.color_print("okgreen","📝 AI 回答：")
            for chunk in response:
                if chunk.choices[0].delta.content:
                    content = chunk.choices[0].delta.content
                    answer += content
                    print(content, end="", flush=True)
            print() # 换行
            elapsed = time.time()- start_time
            Config.color_print("okblue",f"⏱️ 耗时：{elapsed:.2f}秒")
            return answer
        except Exception as e:
            Config.color_print("fail",f"❌ API 调用失败：{str(e)}")
            raise

    def summarize(self, content):
        """生成文档摘要"""
        prompt = f"请为以下文档生成一个简洁的摘要（不超过 200 字）：\n\n{content[:2000]}"
        return self.ask(prompt, "")

    def extract_keywords(self, content):
        """提取关键词"""
        prompt = f"请从以下文档中提取 5-10 个关键词：\n\n{content[:2000]}"
        return self.ask(prompt, "")

3.4 主程序入口 (main.py)

""" PDF 智能助手主程序 """
import argparse
import sys
from pdf_reader import PDFReader
from ai_client import AIClient
from config import Config

def print_banner():
    """打印欢迎界面"""
    banner = """ ╔═══════════════════════════════════════╗ ║ 📚 PDF 智能文档助手 v1.0 📚 ║ ║ Powered by OpenAI GPT ║ ╚═══════════════════════════════════════╝ """
    Config.color_print("header", banner)

def interactive_mode(reader, ai_client):
    """交互模式"""
    Config.color_print("okgreen","\n🎯 进入交互模式（输入'quit'退出）")
    context = reader.content[:3000] # 使用前 3000 字作为上下文
    while True:
        try:
            question = input("\n💬 请输入问题：").strip()
            if question.lower() in ['quit','exit','q']:
                Config.color_print("warning","👋 再见！")
                break
            if not question:
                continue
            ai_client.ask(question, context)
        except KeyboardInterrupt:
            Config.color_print("warning","\n👋 用户取消，再见！")
            break
        except Exception as e:
            Config.color_print("fail",f"❌ 错误：{str(e)}")

def main():
    """主函数"""
    parser = argparse.ArgumentParser(description="PDF 智能文档助手")
    parser.add_argument("file",help="PDF 文件路径")
    parser.add_argument("--summarize", action="store_true",help="生成文档摘要")
    parser.add_argument("--keywords", action="store_true",help="提取关键词")
    parser.add_argument("--ask", metavar="QUESTION",help="向 AI 提问")
    args = parser.parse_args()

    print_banner()
    try:
        # 读取 PDF
        reader = PDFReader(args.file)
        content = reader.read_pdf()
        # 初始化 AI 客户端
        ai_client = AIClient()
        # 执行相应功能
        if args.summarize:
            ai_client.summarize(content)
        elif args.keywords:
            ai_client.extract_keywords(content)
        elif args.ask:
            ai_client.ask(args.ask, content[:3000])
        else:
            # 进入交互模式
            interactive_mode(reader, ai_client)
    except Exception as e:
        Config.color_print("fail",f"\n❌ 程序异常：{str(e)}")
        sys.exit(1)

if __name__ == "__main__":
    main()

📊 项目功能流程图

有参数 无参数 失败 成功 是 是 是 quit 问题 程序启动 检查命令行参数 执行指定功能 读取 PDF 文件 文件验证 显示错误信息 提取文本内容 初始化 AI 客户端 --summarize? --keywords? --ask? 生成摘要 提取关键词 单次提问 进入交互模式 等待用户输入 程序结束 AI 处理

🎯 使用示例

4.1 生成文档摘要

python main.py document.pdf --summarize

4.2 提取关键词

python main.py document.pdf --keywords

4.3 单次提问

python main.py document.pdf --ask"这篇文章的主要观点是什么？"

4.4 交互模式

python main.py document.pdf

交互模式示例输出：

💬 请输入问题：这篇文章讲了什么？
🤖 AI 正在思考...
📝 AI 回答：
这篇文章主要介绍了人工智能的发展历程和应用场景...
⏱️ 耗时：2.35 秒
💬 请输入问题：作者提到了哪些关键技术？
🤖 AI 正在思考...
📝 AI 回答：
作者主要提到了以下几项关键技术：
1. 深度学习
2. 自然语言处理
3. 计算机视觉 ...
⏱️ 耗时：1.98 秒
💬 请输入问题：quit
👋 再见！

📈 AI 小工具开发技能分布

25% 20% 20% 15% 10% 10% 开发 AI 小工具所需技能分布 Python 编程基础 API 调用与集成 数据处理能力 错误处理与调试 用户体验设计 文档与测试

🚀 进阶功能扩展

5.1 添加批量处理功能

def batch_process(file_list, func):
    """批量处理多个文件"""
    results = []
    for file in tqdm(file_list, desc="批量处理"):
        try:
            reader = PDFReader(file)
            content = reader.read_pdf()
            result = func(content)
            results.append({"file":file,"result": result,"status":"success"})
        except Exception as e:
            results.append({"file":file,"result":str(e),"status":"failed"})
    return results

5.2 添加导出功能

def export_to_markdown(content, output_file):
    """导出为 Markdown 格式"""
    with open(output_file,'w', encoding='utf-8') as f:
        f.write("# 文档摘要\n\n")
        f.write(content)
    Config.color_print("okgreen",f"✅ 已导出到：{output_file}")

5.3 添加记忆功能

class ConversationMemory:
    """对话记忆管理"""
    def __init__(self, max_history=5):
        self.history = []
        self.max_history = max_history

    def add(self, question, answer):
        """添加对话记录"""
        self.history.append({"question": question,"answer": answer })
        # 保留最近 N 条记录
        if len(self.history)> self.max_history:
            self.history = self.history[-self.max_history:]

    def get_context(self):
        """获取上下文"""
        context = ""
        for item in self.history:
            context += f"Q: {item['question']}\nA: {item['answer']}\n\n"
        return context

💡 开发经验总结

常见问题及解决方案

最佳实践

# 1. 使用类型注解
def process_file(file_path:str)->dict:
    """处理文件并返回结果字典"""
    pass

# 2. 添加异常处理
try:
    result = risky_operation()
except SpecificError as e:
    logger.error(f"操作失败：{e}")
    # 执行恢复操作
finally:
    cleanup()

# 3. 使用日志记录
import logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s')

# 4. 编写单元测试
import unittest
class TestPDFReader(unittest.TestCase):
    def test_validate_file(self):
        reader = PDFReader("test.pdf")
        # 测试代码...

🎓 结语

恭喜你！🎉

如果你已经跟随这篇文章完成了项目，那么你已经：

✅ 学会了如何调用 OpenAI API ✅ 掌握了 PDF 文件处理技巧 ✅ 了解了命令行工具的开发流程 ✅ 获得了一个实用的 AI 小工具

下一步建议：

1. 尝试添加新功能（如多格式支持、语音交互）
2. 优化用户界面（如使用 GUI 框架）
3. 部署到云端（如使用 FastAPI 构建 Web 服务）
4. 开发更多 AI 小工具（如图片识别、语音助手）

记住：最好的学习方式就是动手实践！