はじめに
がちもとさんアドベントカレンダー7日目の記事です。
今日は、ダリーさん（DALLE3）に画像生成してもらいました。
開発環境
- Windows 11 PC
- Python 3.11
導入
1．ライブラリのインストール
pip install openai
pip install requests
2．APIキーを発行
3．プログラムを作成
dalle3.py
from openai import OpenAI
import requests
import os
client = OpenAI(api_key="<INSERT-YOUR-API-KEY>")
response = client.images.generate(
model="dall-e-3",
prompt="a white siamese cat",
size="1024x1024",
quality="standard", # "hd"
n=1,
)
image_url = response.data[0].url
print(image_url)
url_parts = image_url.split('?')
file_name = url_parts[0].split('/')[-1]
response = requests.get(image_url)
if response.status_code == 200:
with open(file_name, "wb") as f:
f.write(response.content)
print("画像をダウンロードして保存しました。")
else:
print("画像のダウンロードに失敗しました。")
コード解説
このコードは、OpenAIのDALL·Eモデルを使用して、指定したテキストプロンプトに基づいて画像を生成し、その画像をダウンロードして保存するPythonプログラムです。以下、コードの簡単な説明です。
1．OpenAIのAPIキーをセットアップします。
- api_keyには、OpenAIのAPIを使用するためのキーを指定します。
2．client.images.generateメソッドを使用して画像を生成します。
- modelには、使用するDALL·Eモデルの種類を指定します（ここでは"dall-e-3"を使用しています）。
- promptには、生成する画像に関するテキストのプロンプトを指定します（ここでは「a white siamese cat」を指定しています）。
- sizeには、生成される画像のサイズを指定します（ここでは1024x1024ピクセル）。
- qualityには、画像の品質を指定します（ここでは「standard」を使用していますが、「hd」も選択可能です）。
- nには、生成する画像の数を指定します（ここでは1つ）。
3．生成された画像のURLを取得します。
4．画像のURLからファイル名を取得します。
5．requests.getを使用して、生成された画像のURLから画像データをダウンロードします。
6．ダウンロードが成功した場合は、ファイルに保存します。
このコードを実行すると、指定したテキストプロンプトに基づいて生成された画像がダウンロードされ、保存されます。
| standard | hd |
|---|---|
 |
 |
お疲れさまでした。