首先需要安装openai模块
pip install openai
其次需要去官网申请一个key
登陆点击右上角头像,选择View API keys,然后点击下图按钮创建key即可。
1、聊天接口调用代码
import os import openai openai.api_key = "sk-oahsWxJYvJYb6iX 你的key!!!
Qggoq3vimkl7IKMOGCkm" askInfo = input('--- ask: ') # 聊天 response = openai.
Completion.create( model="text-davinci-003", prompt=askInfo, temperature=0.7,
max_tokens=2560, top_p=1, frequency_penalty=0, presence_penalty=0 ) print(
response)
2、文字生成图片
下面代码除了需要openai模块外,还需要安装其他模块,主要用于返回的图片直接显示。
import os import openai import requests as req from PIL import Image from io
import BytesIO openai.api_key = "sk-oahsWxJYvJYb6iX 你的key!!!
Qggoq3vimkl7IKMOGCkm" askInfo = input('--- ask: ') # 绘制图片 response = openai.
Image.create( prompt=askInfo, n=1, size="512x512" ) print(response) image_url =
response['data'][0]['url'] print(image_url) # 打开链接中的图片 response = req.get(
image_url) image = Image.open(BytesIO(response.content)) image.show()