一、实现原理:
1、导入Python所需要的模块
前提:需要先安装openai模块
cmd窗口安装命令:pip install openai
3、将要改写的文章交给openai处理
二、完整代码:
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('转换完成,请查看你的文章')