GPT 이미지 (리버스)

from openai import OpenAI
import os
import requests
import base64

client = OpenAI(
    api_key="AIHUBMIX_API_KEY", # AiHubMix API 키로 교체
    base_url="https://aihubmix.com/v1"
)

prompt = """
영화 [블랙 스완]의 포스터를 재디자인하세요, 3D 만화, 부드러운 렌더링, 밝은 톤, 2:3
"""

result = client.images.generate(
    model="gpt-4o-image-vip",
    prompt=prompt,
    n=1, # 생성할 이미지 수
    size="1024x1536", # 1024x1024 (정사각형), 1536x1024 (3:2 가로), 1024x1536 (2:3 세로), auto (기본값)
    #quality="high" # 리버스 모델에 대해 품질을 전달하지 마십시오. 내부적으로 바인딩됩니다 📍
)

print(result) # 전체 API 응답 인쇄, 리버스 API에 대해 사용량은 지원되지 않음 📍

# 출력 파일 이름 접두사 및 디렉토리 정의
output_dir = "." # 다른 디렉토리를 지정할 수 있습니다
file_prefix = "image_gen-reverse"

# 출력 디렉토리가 있는지 확인
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을 찾을 수 없어 저장을 건너뜁니다.")