python自动化、爬虫等脚本定制

Python ChatGPT自动化批量改写文章 基于GPT-3.5-turbo模型

ChatGPT admin 2年前 (2023-03-06) 1972次浏览 2个评论

一、实现原理:

1、导入Python所需要的模块

import-openai

前提:需要先安装openai模块

cmd窗口安装命令:pip install openai

2、Python读取我们要改写的文档Python-读取文档

3、将要改写的文章交给openai处理

python-chatgpt-openai

4、将改写后的文章输出到文档python-output

二、完整代码:

import openai

openai.api_key = "填写你的openai key"
title = '中国春节.txt' #要改写的文档文件名称,新建一个intput目录,文档放在intput目录下
print(title)
#这里要读取两次,第一次是计算循坏所需要的次数,第二次读取原文
f = open("./intput/{}".format(title),'r',encoding= 'utf-8') #
s = len(f.read()) // 400
f.close()
f = open("./intput/{}".format(title),'r',encoding= 'utf-8')

for i in range(s+1):
    content = f.read(400)
    print('这是原文:', content + "\n")

    conversation = [{"role": "system", "content": "You are a helpful assistant."}]

    prompt = '帮我改写以下文章:' + content #content为原文
    conversation.append({"role": "user", "content": prompt})
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=conversation,
        temperature=1,
        max_tokens=2048,
        top_p=0.9
    )
    conversation.append(
        {"role": "assistant", "content": response['choices'][0]['message']['content']})  # 将上一次会话信息返回给chatgpt,可省略

    message = response['choices'][0]['message']['content'] + "\n"
    print('这是改写后的文章:', message)
    #改写的结果都保存为output/xxx文件,先新建一个output目录
    with open("./output/{}".format(title), "a") as f2:
        f2.write(message)
        f2.close()

print('转换完成,请查看你的文章')
喜欢 (5)
发表我的评论
取消评论
表情 贴图 加粗 删除线 居中 斜体 签到
(2)个小伙伴在吐槽
  1. 博主,能否付费改一下?
    匿名2023-03-08 15:53 回复
    • 进群私信我一下:https://t.me/+WRKAeKTAlKgwOTM1
      admin2023-03-08 17:29 回复