#!/usr/bin/env python3
"""M3-1 서울대보험쌤 × 정당한 대우 Google 광고 배너 생성
- m3-1-1200x628.png (가로형)
- m3-1-1080x1080.png (정사각형)
"""
from __future__ import annotations
import base64
import json
import sys
import time
from pathlib import Path
import requests
from playwright.sync_api import sync_playwright
# ── 경로 설정 ────────────────────────────────────────────────────────────────
BASE_DIR = Path(__file__).parent
OUTPUT_DIR = Path("/home/jay/workspace/output/google-ads/banners/m3")
TMP_DIR = BASE_DIR / "output" / "v4-hybrid"
BG_PATH = TMP_DIR / "bg_m3_1.jpg"
# Gemini API 설정
GEMINI_API_BASE = "https://generativelanguage.googleapis.com/v1beta"
MODEL_ID = "gemini-3-pro-image-preview"
FALLBACK_MODEL_ID = "gemini-3.1-flash-image-preview"
# ── 배경 프롬프트 (디자인 디렉션 원문) ────────────────────────────────────────
BG_PROMPT = (
"Photographic scene of an elegant, warm private study in the evening. "
"Medium shot, slightly low angle looking across the room. A large "
"walnut-toned bookshelf fills most of the back wall — books neatly "
"organized, warm ambient lighting from integrated shelf lighting. "
"A solid walnut desk in the foreground, left side: a MacBook Pro closed, "
"a handwritten notebook open with a pen resting on it, and a small "
"glass of water. A warm reading lamp on the right edge of the desk "
"casting directional amber light. Through a window on the left wall, "
"a soft Seoul cityscape at dusk — lights beginning to appear, sky "
"in deep blue-orange gradient. The room feels lived-in but immaculate — "
"a space of concentrated expertise and quiet ambition. Camera: 50mm "
"equivalent, natural indoor lighting with warm color temperature. "
"No text. No logos. No people. No icons. No graphics."
)
# ── Gemini 배경 생성 ──────────────────────────────────────────────────────────
def get_auth_token() -> str:
"""gcloud_auth 모듈로 Bearer 토큰을 획득합니다."""
sys.path.insert(0, str(BASE_DIR))
import gcloud_auth
return gcloud_auth.get_access_token()
def _call_model(token: str, api_key: str | None, model: str, payload: dict) -> requests.Response:
"""SA 토큰 우선, 실패 시 API 키로 Gemini 호출."""
url_bearer = f"{GEMINI_API_BASE}/models/{model}:generateContent"
headers_bearer = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
resp = requests.post(url_bearer, headers=headers_bearer, json=payload, timeout=300)
if resp.status_code == 200:
return resp
# API 키 폴백
if api_key:
url_key = f"{GEMINI_API_BASE}/models/{model}:generateContent?key={api_key}"
headers_key = {"Content-Type": "application/json"}
return requests.post(url_key, headers=headers_key, json=payload, timeout=300)
return resp
def generate_background() -> Path:
"""Gemini API로 배경 이미지를 생성하고 경로를 반환합니다."""
# 이미 생성된 배경이 있으면 재사용
if BG_PATH.exists() and BG_PATH.stat().st_size > 50_000:
print(f"[배경] 기존 배경 이미지 재사용: {BG_PATH}")
return BG_PATH
print("[배경] Gemini API로 배경 이미지 생성 중...")
token = get_auth_token()
sys.path.insert(0, str(BASE_DIR))
import gcloud_auth as ga
api_key = ga.get_api_key("GEMINI_API_KEY")
payload = {
"contents": [{"parts": [{"text": BG_PROMPT}]}],
"generationConfig": {"responseModalities": ["IMAGE", "TEXT"]},
}
models_to_try = [MODEL_ID, FALLBACK_MODEL_ID]
last_error = None
for attempt, model in enumerate(models_to_try, 1):
print(f" 시도 {attempt}: 모델={model}")
try:
resp = _call_model(token, api_key, model, payload)
except Exception as e:
last_error = str(e)
print(f" 요청 실패: {e}")
continue
if resp.status_code in (400, 403, 404):
print(f" 모델 접근 실패 ({resp.status_code}: {resp.text[:200]})")
last_error = f"HTTP {resp.status_code}"
continue
if resp.status_code != 200:
print(f" HTTP {resp.status_code}: {resp.text[:300]}")
last_error = f"HTTP {resp.status_code}"
continue
data = resp.json()
candidates = data.get("candidates", [])
if not candidates:
print(f" 응답에 candidates 없음: {json.dumps(data)[:200]}")
last_error = "No candidates"
continue
parts = candidates[0].get("content", {}).get("parts", [])
image_part = next((p for p in parts if "inlineData" in p), None)
if image_part is None:
print(f" 이미지 없음. parts: {[list(p.keys()) for p in parts]}")
last_error = "No inlineData"
continue
mime = image_part["inlineData"].get("mimeType", "image/jpeg")
img_bytes = base64.b64decode(image_part["inlineData"]["data"])
BG_PATH.write_bytes(img_bytes)
print(f" [배경] 저장 완료: {BG_PATH} ({len(img_bytes):,} bytes, mime={mime})")
return BG_PATH
raise RuntimeError(f"배경 이미지 생성 실패: {last_error}")
# ── HTML 템플릿 ───────────────────────────────────────────────────────────────
def make_html_1200x628(bg_path: Path) -> str:
"""1200×628 가로형 배너 HTML 생성."""
bg_uri = f"file://{bg_path.resolve()}"
return f"""
서울대보험쌤 · TOP사업단
지점장이 직접
컨설팅하는 보험영업
AI 시스템으로 실적을 극대화
DB영업 | 멘토링 | 정착금 1,000만원
무료 상담 신청 →
"""
def make_html_1080x1080(bg_path: Path) -> str:
"""1080×1080 정사각형 배너 HTML 생성."""
bg_uri = f"file://{bg_path.resolve()}"
return f"""
서울대보험쌤 · TOP사업단
지점장이 직접
컨설팅하는 보험영업
AI 시스템으로 실적을 극대화
무료 상담 신청 →
"""
# ── Playwright 캡처 ───────────────────────────────────────────────────────────
def capture_html_to_png(html_content: str, width: int, height: int, output_path: Path) -> None:
"""HTML을 Playwright로 캡처하여 PNG로 저장합니다."""
tmp_html = TMP_DIR / f"_tmp_m3_1_{width}x{height}.html"
tmp_html.write_text(html_content, encoding="utf-8")
with sync_playwright() as p:
browser = p.chromium.launch()
try:
page = browser.new_page(viewport={"width": width, "height": height})
page.goto(f"file://{tmp_html.resolve()}", wait_until="networkidle")
page.wait_for_timeout(2000) # 폰트 로딩 대기
output_path.parent.mkdir(parents=True, exist_ok=True)
page.screenshot(path=str(output_path), type="png", clip={
"x": 0, "y": 0, "width": width, "height": height
})
print(f" [캡처] {output_path} ({output_path.stat().st_size / 1024:.0f} KB)")
finally:
browser.close()
tmp_html.unlink(missing_ok=True)
# ── 메인 ─────────────────────────────────────────────────────────────────────
def main():
print("=" * 60)
print("M3-1 서울대보험쌤 × 정당한 대우 배너 생성 시작")
print("=" * 60)
TMP_DIR.mkdir(parents=True, exist_ok=True)
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
# 1. 배경 이미지 생성
try:
bg_path = generate_background()
except Exception as e:
print(f"[오류] 배경 생성 실패: {e}")
print("[폴백] 다크 그라데이션 배경으로 대체합니다.")
bg_path = None
# 2. 1200x628 배너 생성
out_1200 = OUTPUT_DIR / "m3-1-1200x628.png"
print("\n[1] 1200×628 가로형 배너 생성 중...")
try:
if bg_path:
html_1200 = make_html_1200x628(bg_path)
else:
# 폴백: 배경 없이 다크 그라데이션
dummy = Path("/nonexistent_bg")
html_1200 = make_html_1200x628(dummy).replace(
f"url('file:///nonexistent_bg')",
"none",
).replace(
".bg-image {",
".bg-image { background: linear-gradient(135deg, #3E2723 0%, #5D3A28 50%, #3E2723 100%);",
)
capture_html_to_png(html_1200, 1200, 628, out_1200)
print(f" 완료: {out_1200}")
except Exception as e:
print(f" [오류] {e}")
raise
# 3. 1080x1080 배너 생성
out_1080 = OUTPUT_DIR / "m3-1-1080x1080.png"
print("\n[2] 1080×1080 정사각형 배너 생성 중...")
try:
if bg_path:
html_1080 = make_html_1080x1080(bg_path)
else:
dummy = Path("/nonexistent_bg")
html_1080 = make_html_1080x1080(dummy).replace(
f"url('file:///nonexistent_bg')",
"none",
).replace(
".bg-image {",
".bg-image { background: linear-gradient(160deg, #2C1810 0%, #4A2C1C 40%, #3E2723 100%);",
)
capture_html_to_png(html_1080, 1080, 1080, out_1080)
print(f" 완료: {out_1080}")
except Exception as e:
print(f" [오류] {e}")
raise
print("\n" + "=" * 60)
print("생성 완료!")
print(f" 1200×628 : {out_1200} ({out_1200.stat().st_size / 1024:.0f} KB)")
print(f" 1080×1080: {out_1080} ({out_1080.stat().st_size / 1024:.0f} KB)")
print("=" * 60)
if __name__ == "__main__":
main()