GPT Image

from openai import OpenAI
import base64
import os

client = OpenAI(
    api_key="AIHUBMIX_API_KEY", # 換成你在後台生成的 Key "sk-***"
    base_url="https://aihubmix.com/v1"
)

prompt = """redesign poster of the movie [Black Swan], 3D cartoon, smooth render, bright tone, 2:3 portrait."""

result = client.images.generate(
    model="gpt-image-1",
    prompt=prompt,
    n=1, # 單次出圖數量，最多 10 張
    size="1024x1536", # 1024x1024 (square), 1536x1024 (3:2 landscape), 1024x1536 (2:3 portrait), auto (default) 
    quality="high", # high, medium, low, auto (default)
    moderation="low", # low, auto (default) 需要升級 openai 包 📍
    background="auto", # transparent, opaque, auto (default)
)

print(result.usage)

# 定義文件名前綴和保存目錄
output_dir = "." # 可以指定其他目錄
file_prefix = "image_gen"

# 確保輸出目錄存在
os.makedirs(output_dir, exist_ok=True)

# 遍歷所有返回的圖片數據
for i, image_data in enumerate(result.data):
    image_base64 = image_data.b64_json
    if image_base64: # 確保 b64_json 不為空
        image_bytes = base64.b64decode(image_base64)

        # --- 文件名衝突處理邏輯開始 ---
        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

        # 使用找到的唯一 file_path 保存圖片到文件
        with open(file_path, "wb") as f:
            f.write(image_bytes)
        print(f"圖片已保存至：{file_path}")
    else:
        print(f"第 {i} 張圖片數據為空，跳過保存。")

from openai import OpenAI
import base64
import os

client = OpenAI(
    api_key="AIHUBMIX_API_KEY", # 換成你在後台生成的 Key "sk-***"
    base_url="https://aihubmix.com/v1"
)

prompt = """redesign poster of the movie [Black Swan], 3D cartoon, smooth render, bright tone, 2:3 portrait."""

result = client.images.edit(
    model="gpt-image-1",
    image=open("yourpath/edit.jpg", "rb"), #多參考圖應使用 [列表，]
    n=2, # 單次數量
    prompt=prompt,
    size="1024x1536", # 1024x1024 (square), 1536x1024 (3:2 landscape), 1024x1536 (2:3 portrait), auto (default)
    # moderation="low", # edit 不支持 moderation
    quality="high" # high, medium, low, auto (default)
)

print(result.usage)

# 定義文件名前綴和保存目錄
output_dir = "." # 可以指定其他目錄
file_prefix = "image_edit" # 修改文件名前綴

# 確保輸出目錄存在
os.makedirs(output_dir, exist_ok=True)

# --- 遍歷 API 返回的每張圖片數據 ---
for i, image_item in enumerate(result.data):
    image_base64 = image_item.b64_json
    if image_base64 is None:
        print(f"警告：第 {i+1} 張圖片沒有返回 base64 數據，跳過保存。")
        continue # 如果没有 b64_json 數據，跳到下張圖片

    image_bytes = base64.b64decode(image_base64)

    # --- 為當前圖片尋找不衝突的文件名 ---
    current_index = 0 # 每次都從 0 開始檢查，或者維護一個全局遞增的索引
    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

    # 使用找到的唯一 file_path 保存當前圖片到文件
    try:
        with open(file_path, "wb") as f:
            f.write(image_bytes)
        print(f"第 {i+1} 張編輯後的圖片已保存至：{file_path}")
    except Exception as e:
        print(f"保存第 {i+1} 張圖片時出錯 ({file_path}): {e}")

Usage(input_tokens=150, input_tokens_details=UsageInputTokensDetails(image_tokens=0, text_tokens=150), output_tokens=6240, total_tokens=6390)
圖片已保存至：./image_gen_14.png

Usage(input_tokens=992, input_tokens_details=UsageInputTokensDetails(image_tokens=646, text_tokens=346), output_tokens=12480, total_tokens=13472)
第 1 張編輯後的圖片已保存至：./image_edit_1.png
第 2 張編輯後的圖片已保存至：./image_edit_2.png