分享几张Openai生成的图片

输入的中文，翻译使用的openai的text-davinci-003.

import requests

from requests.structures import CaseInsensitiveDict

import openai

import json

import datetime

# 导入os

import os

from dotenv import load_dotenv

def openai_translate(text):

    text="Translate this into English:\n\n"+text+"\n\n"

    response = openai.Completion.create(

        engine="text-davinci-003",

        prompt=text,

        temperature=0.9,

        max_tokens=1024,

        top_p=1,

        frequency_penalty=0,

        presence_penalty=0.6,

    )

    return response['choices'][0]['text']

def generate_image(question):

    prompt = openai_translate(question)

    response = openai.Image.create(

        prompt=prompt,

        n=1,

        size="1024x1024"

        )

    image_url = response['data'][0]['url']

    return image_url

def download_image(url, file_name):

    response = requests.get(url)

    if response.status_code == 200:

        with open(file_name, 'wb') as f:

            f.write(response.content)

if __name__ == "__main__":

    # 加载配置文件

    load_dotenv()

    # 读取 API key

    openai.api_key = os.getenv("KEY")

    # 生成图像

    question = input()

    image_url = generate_image(question)

    now=datetime.datetime.now().strftime("%Y%m%d%H%M") #当前时间作为文件名

    imgname='d:\desktop\image%s.png'%now

    download_image(image_url, imgname)

    # 使用系统默认的图片浏览器打开图片# 启动系统默认的图片浏览器打开图片

    os.startfile(imgname)