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 = """
redesign poster of the movie [Black Swam], 3D cartoon, smooth render, bright tone, 2:3
"""

result = client.images.generate(
    model="gpt-4o-image-vip",
    prompt=prompt,
    n=1, # 單次數量
    size="1024x1536", # 1024x1024 (square), 1536x1024 (3:2 landscape), 1024x1536 (2:3 portrait), auto (default)
    #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} 張圖片時發生錯誤：{e}")
    else:
        print(f"第 {i} 張圖片數據中既無 b64_json 也無 url，跳過保存。")