Nano Banana 2 API 가격과 사용법: 이미지 생성·편집

"""Nano Banana 2(gemini-3.1-flash-image)로 이미지를 생성하거나 편집합니다."""
import argparse
import base64
import os
import sys
from pathlib import Path

import requests

URL = "https://api.laozhang.ai/v1beta/models/gemini-3.1-flash-image:generateContent"
RATIOS = ["1:1", "1:4", "1:8", "2:3", "3:2", "3:4", "4:1", "4:3", "4:5", "5:4", "8:1", "9:16", "16:9", "21:9"]
SIZES = ["512", "1K", "2K", "4K"]
MIME_TYPES = {".png": "image/png", ".jpg": "image/jpeg", ".jpeg": "image/jpeg", ".webp": "image/webp"}
EXTENSIONS = {"image/png": "png", "image/jpeg": "jpg", "image/webp": "webp"}

parser = argparse.ArgumentParser(description="이미지를 생성하거나, 전달한 이미지를 편집합니다.")
parser.add_argument("prompt")
parser.add_argument("images", nargs="*", help="참조 이미지(PNG, JPEG, WebP) 최대 14장")
parser.add_argument("--ratio", choices=RATIOS, help="종횡비. 생략하면 1:1이거나 입력 이미지를 따릅니다")
parser.add_argument("--size", default="1K", choices=SIZES)
parser.add_argument("--out", default="nano-banana-2", help="확장자를 뺀 출력 파일 이름")
args = parser.parse_args()
if len(args.images) > 14:
    sys.exit("참조 이미지는 최대 14장까지 보낼 수 있습니다.")

parts = [{"text": args.prompt}]
for path in args.images:
    mime = MIME_TYPES.get(Path(path).suffix.lower())
    if mime is None:
        sys.exit(f"PNG, JPEG, WebP 파일을 사용하세요: {path}")
    data = base64.b64encode(Path(path).read_bytes()).decode("ascii")
    parts.append({"inlineData": {"mimeType": mime, "data": data}})

image_config = {"imageSize": args.size}
if args.ratio:
    image_config["aspectRatio"] = args.ratio

response = requests.post(
    URL,
    headers={"Authorization": f"Bearer {os.environ['LAOZHANG_API_KEY']}"},
    json={
        "contents": [{"role": "user", "parts": parts}],
        "generationConfig": {"responseModalities": ["IMAGE"], "imageConfig": image_config},
    },
    timeout=300,
)
if response.status_code != 200:
    sys.exit(f"HTTP {response.status_code}: {response.text[:500]}")

result = response.json()
images = [
    part["inlineData"]
    for candidate in result.get("candidates") or []
    for part in (candidate.get("content") or {}).get("parts") or []
    if "inlineData" in part and not part.get("thought")
]
if not images:
    reasons = [candidate.get("finishReason") for candidate in result.get("candidates") or []]
    sys.exit(f"반환된 이미지가 없습니다: {result.get('promptFeedback') or reasons}")
for number, image in enumerate(images, start=1):
    suffix = "" if number == 1 else f"-{number}"
    output = Path(f"{args.out}{suffix}.{EXTENSIONS.get(image['mimeType'], 'bin')}")
    output.write_bytes(base64.b64decode(image["data"]))
    print("저장됨:", output)

python nano_banana_2.py "Generate an image of a night market food stall with a hand-painted sign that reads \"NOODLES\", warm lantern light." --ratio 16:9 --size 4K --out nb2-night-market

# 이미지 한 장 편집
python nano_banana_2.py "이 낮 거리 사진을 네온이 비치는 비 오는 밤 장면으로 바꿔 주세요. 건물은 그대로 유지하세요." street.jpg --ratio 3:2 --size 2K --out nb2-street

# 참조 이미지 세 장 합성: 제품, 장면, 분위기를 정할 세 번째 이미지
python nano_banana_2.py "첫 번째 이미지의 제품을 두 번째 이미지의 테이블 위에 놓고, 세 번째 이미지의 화풍으로 그려 주세요." product.png scene.jpg style.jpg --ratio 3:2 --size 2K --out nb2-composite