使用DALL-E 2生成图片需要先安装OpenAI API和相关的Python库。以下是生成图片的基本代码:
import openai
import requests
from requests.structures import CaseInsensitiveDict
import json
openai.api_key = "YOUR_API_KEY"
image_prompt = "a portrait of a cat"
openai_model = "image-alpha-001"
image_size = 256
def generate_image(prompt, model, size):
headers = CaseInsensitiveDict()
headers["Content-Type"] = "application/json"
headers["Authorization"] = f"Bearer {openai.api_key}"
data = """
{
"""
data += f'"model": "{model}",'
data += f'"prompt": "{prompt}",'
data += f'"num_images":1,'
data += f'"size":"{size}"'
data += """
}
"""
resp = requests.post("https://api.openai.com/v1/images/generations", headers=headers, data=data)
if resp.status_code != 200:
raise ValueError("Failed to generate image")
response_text = json.loads(resp.text)
image_url = response_text['data'][0]['url']
return image_url
image_url = generate_image(image_prompt, openai_model, image_size)
print("Image URL:", image_url)
这里使用了OpenAI API的“images”功能,通过post请求将生成图片的参数传递给API,获取url链接后,你可以将生成图片的代码进一步完善,实现更多自定义参数的功能,例如生成多张图片、调整图片大小、添加图像元素等。