python+ai大模型api实现一句话让ai编辑excel、ppt、word等电子文档代码-python代码-BFW代码库

python+ai大模型api实现一句话让ai编辑excel、ppt、word等电子文档代码

代码语言：python

所属分类：其他

代码描述：python+ai大模型api实现一句话让ai编辑excel、ppt、word等电子文档代码

代码标签： python ai 大模型 api 一句话让 ai 编辑 excel ppt word 电子文档

下面为部分代码预览，完整代码请点击下载或在bfwstudio webide中打开

import os
import openai
from dotenv import load_dotenv
import traceback # 用于打印详细错误信息

# 加载环境变量 (包含 OPENAI_API_KEY)
load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")

if not openai.api_key:
    print("错误：请设置 OPENAI_API_KEY 环境变量。")
    exit()

# --- LLM 调用函数 ---
def generate_modification_code(file_path: str, file_type: str, requirement: str, output_path: str) -> str | None:
    """
    调用 OpenAI GPT 模型生成用于修改文件的 Python 代码。

    Args:
        file_path: 输入文件的路径。
        file_type: 文件类型 ('excel', 'word', 'ppt')。
        requirement: 用户描述的修改需求。
        output_path: 期望的输出文件路径。

    Returns:
        生成的 Python 代码字符串，如果失败则返回 None。
    """
    print(f"--- 正在为 {file_type} 文件生成代码以满足需求: {requirement[:100]}... ---")

    # 根据文件类型选择合适的库和基本指令
    library_map = {
        'excel': 'openpyxl',
        'word': 'docx (from python-docx)',
        'ppt': 'pptx (from python-pptx)'
    }
    if file_type not in library_map:
        print(f"错误：不支持的文件类型 '{file_type}'")
        return None

    library_name = library_map[file_type]

    prompt = f"""
你是一个 Python 编程助手。请根据用户的需求，生成一段完整的 Python 代码来修改给定的文件。

文件类型: {file_type}
使用的主要库: {library_name}
输入文件路径变量名: input_file_path (值为: '{file_path}')
输出文件路径变量名: output_file_path (值为: '{output_path}')
用户修改需求:
{requirement}

请生成可以直接执行的 Python 代码，完成上述修改任务。代码需要：
1. 导入必要的库 (.........完整代码请登录后点击上方下载按钮下载查看