分享幾張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)
? ? # 使用系統(tǒng)默認的圖片瀏覽器打開圖片# 啟動系統(tǒng)默認的圖片瀏覽器打開圖片
? ? os.startfile(imgname)