【Python】ChatGPTで画像を生成する方法｜OpenAI APIを用いた画像処理

2023年3月16日

こんにちは、DXCEL WAVEの運営者(@dxcelwave)です！

こんな方におすすめ！

AI画像生成に興味がある
PythonコードとChatGPTを用いて画像を生成する方法が知りたい

【OpenAI】ChatGPTとは？

ChatGPTとは、OpenAIが開発した対話型チャットボットのモデルです。ChatGPTの名前は、GPT-3という第3世代の生成言語モデルに由来しています。

ChatGPTのモデルには、人工知能（AI）が搭載されており、人間の発話に対して自然なやり取りを可能にしています。また、英語をはじめ、中国語、日本語、フランス語など複数言語を認識し、人間らしく応答できるのも特徴的です。

さらに、ChatGPTではチャットの他に、画像生成など近年多様な機能がリリースされてます。以下、ChatGPTで利用可能な機能一覧を示します。

チャット機能（対話・文章生成）
テキストから画像を生成する機能
オーディオを文字起こしする機能

【参考】OpenAIが提供するAPI利用

本記事ではChatGPTを用いたPythonプログラミングについて解説します。その際、Open AIが提供するAPI情報が必要になります。「ChatGPTの概要」および「API情報の取得手順」については、こちらの記事で詳しく解説しています。

【事前準備】OpenAIのPythonライブラリをインストール

ChatGPT APIをPythonで利用する場合、openaiライブラリを活用します。

コマンドプロンプト（WIndows）またはターミナル（Mac）を開き、以下を実行し、ライブラリをインストールしましょう。

pip install openai

【実践】PythonでChatGPT APIを呼び出し画像生成

Pythonプログラムを記述し、ChatGPT APIを活用した画像生成方法について解説します。

PythonライブラリとAPI情報の読込

はじめに、Pythonプログラムの先頭にライブラリとAPI認証情報を記述します。

前述で取得したOrganization IDとシークレットキーをそれぞれ入力しましょう。

import openai
import requests
import json

openai.organization = "<Organization IDをここに入力>"
openai.api_key      = "<APIシークレットキーをここに入力>"

画像生成関数の作成

続いて、ChatGPT APIを呼び出し画像を生成するための関数を記述します。次のコードを記述しましょう。

コード

# 画像生成
def create_image_from_text(text):

    # 応答設定
    response = openai.Image.create(
                  prompt = text,             # 画像生成に用いる説明文章
                  n = 1,                     # 何枚の画像を生成するか
                  size = '512x512',          # 画像サイズ
                  response_format = "url"    # API応答のフォーマット

                )

    # API応答から画像URLを指定
    image_url = response['data'][0]['url']
    
    # 画像をローカルに保存
    image_data = requests.get(image_url).content
    with open("chat-gpt-generated-image.jpg", "wb") as f:
        f.write(image_data)
        
    return image_url

引数情報

openai.Image.create()メソッドに説明文章を渡し、画像を生成します。その際、次の引数を指定します。

スクロールできます

引数名	概要
prompt	画像生成に用いる説明文章
n	何枚の画像を生成するか
size	生成生成サイズ。`256x256`, `512x512`, `1024x1024`が指定可能。
response_format	APIの応答フォーマット。`url`または`b64_json`が指定可能。

画像生成の実行

前述の画像生成関数に入力テキスト（text）を渡し、ChatGPT APIを起動＆実行します。

コード

# 生成するイメージを文章で定義
text = "椅子の上の猫"

# 画像生成
img = create_image_from_text(text)

出力イメージ

上記のコードでは、「椅子の上の猫」の画像を生成するよう命令しています。コードを実行すると、上図のような出力結果が得られました。説明文章通りの画像が生成されていることが分かりますね。

【まとめ】OpenAI ChatGPT APIを利用したPythonコード全量

最後に、本日ご紹介したPythonコードを全量示します。

import openai
import requests
import json

openai.organization = "<Organization IDをここに入力>"
openai.api_key      = "<APIシークレットキーをここに入力>"

# ================================================
# 画像生成
# ================================================

def create_image_from_text(text):
    
    # 応答設定 
    response = openai.Image.create(
                  prompt = text,             # 画像生成に用いる説明文章
                  n = 1,                     # 何枚の画像を生成するか
                  size = '512x512',          # 画像サイズ
                  response_format = "url"    # API応答のフォーマット
                )

    # API応答から画像URLを指定
    image_url = response['data'][0]['url']
    
    # 画像をローカルに保存
    image_data = requests.get(image_url).content
    with open("chat-gpt-generated-image.jpg", "wb") as f:
        f.write(image_data)
        
    return image_url


# ================================================
# 関数実行
# ================================================

# 生成するイメージを文章で定義
text = "椅子の上の猫"

# 画像生成
img = create_image_from_text(text)