DALL-E

from openai import OpenAI
import base64
import requests

client = OpenAI()

# 기본 이미지 생성
def generate_image(prompt, size="1024x1024", quality="standard"):
    """
    DALL-E 3로 이미지 생성
    size: "1024x1024", "1792x1024", "1024x1792"
    quality: "standard", "hd"
    """
    response = client.images.generate(
        model="dall-e-3",
        prompt=prompt,
        size=size,
        quality=quality,
        n=1  # DALL-E 3는 1개만 지원
    )

    return response.data[0].url, response.data[0].revised_prompt

# 사용 예시
image_url, revised_prompt = generate_image(
    prompt="미래적인 서울 도심의 밤 풍경, 네온 불빛이 반사되는 빗물",
    size="1792x1024",
    quality="hd"
)
print(f"생성된 이미지: {image_url}")
print(f"확장된 프롬프트: {revised_prompt}")

# 이미지 편집 (DALL-E 2 사용)
def edit_image(image_path, mask_path, prompt):
    """
    마스크 영역을 프롬프트에 따라 수정
    """
    with open(image_path, "rb") as image_file:
        with open(mask_path, "rb") as mask_file:
            response = client.images.edit(
                model="dall-e-2",
                image=image_file,
                mask=mask_file,
                prompt=prompt,
                n=1,
                size="1024x1024"
            )
    return response.data[0].url

# 이미지 변형 (DALL-E 2 사용)
def create_variation(image_path):
    """
    원본 이미지의 변형 버전 생성
    """
    with open(image_path, "rb") as image_file:
        response = client.images.create_variation(
            model="dall-e-2",
            image=image_file,
            n=2,
            size="1024x1024"
        )
    return [img.url for img in response.data]

# 이미지 다운로드 유틸리티
def download_image(url, filename):
    response = requests.get(url)
    with open(filename, "wb") as f:
        f.write(response.content)
    print(f"저장 완료: {filename}")

# 배치 생성 (비용 효율적)
prompts = [
    "AI 로봇이 책을 읽는 모습, 미니멀 일러스트",
    "데이터 시각화 대시보드, 다크 테마 UI",
    "머신러닝 뉴런 네트워크, 추상적 3D 렌더링"
]

for i, prompt in enumerate(prompts):
    url, _ = generate_image(prompt, quality="standard")
    download_image(url, f"output_{i}.png")