GPT描画の先行体験（逆方向）

from openai import OpenAI
import os
import requests
import base64

client = OpenAI(
    api_key="AIHUBMIX_API_KEY", # あなたのAiHubMixキーに置き換えてください
    base_url="https://aihubmix.com/v1"
)

prompt = """
映画[ブラック・スワン]のポスターを再デザインしてください。3Dカートゥーン、スムーズなレンダリング、明るいトーン、2:3
"""

result = client.images.generate(
    model="gpt-4o-image-vip",
    prompt=prompt,
    n=1, # 1回あたりの数量
    size="1024x1536", # 1024x1024 (正方形), 1536x1024 (3:2横長), 1024x1536 (2:3縦長), auto (デフォルト)
    #quality="high" # 逆方向モデルは品質パラメータを渡すことはできません。モデルにバンドルされています 📍
)

print(result) # 完全なAPI応答を出力します。逆方向インターフェースはusageをサポートしていません 📍

# ファイル名のプレフィックスと保存ディレクトリを定義
output_dir = "." # 他のディレクトリを指定できます
file_prefix = "image_gen-逆方向"

# 出力ディレクトリが存在することを確認
os.makedirs(output_dir, exist_ok=True)

# 返されたすべての画像データを反復処理
for i, image_data in enumerate(result.data):
    # まずbase64文字列を処理
    if hasattr(image_data, "b64_json") and image_data.b64_json:
        image_bytes = base64.b64decode(image_data.b64_json)
        current_index = i
        while True:
            file_name = f"{file_prefix}_{current_index}.png"
            file_path = os.path.join(output_dir, file_name)
            if not os.path.exists(file_path):
                break
            current_index += 1
        with open(file_path, "wb") as f:
            f.write(image_bytes)
        print(f"画像は以下に保存されました：{file_path} (base64)")
    # 次にurlフィールドを処理（互換性のあるインターフェースが変更されるのを防ぐために堅牢な処理を維持）
    elif hasattr(image_data, "url") and image_data.url:
        try:
            response = requests.get(image_data.url, stream=True)
            response.raise_for_status()
            image_bytes = response.content
            current_index = i
            while True:
                file_name = f"{file_prefix}_{current_index}.png"
                file_path = os.path.join(output_dir, file_name)
                if not os.path.exists(file_path):
                    break
                current_index += 1
            with open(file_path, "wb") as f:
                f.write(image_bytes)
            print(f"画像は以下に保存されました：{file_path} (url)")
        except requests.exceptions.RequestException as e:
            print(f"画像 {i} のダウンロードに失敗しました：{e}")
        except Exception as e:
            print(f"画像 {i} の処理中にエラーが発生しました：{str(e)}")
    else:
        print(f"画像 {i} のデータにはb64_jsonもurlもありません。保存をスキップします。")