#!/usr/bin/env python3
"""M1-1 '정당한 대우' Google 광고 배너 생성기.
Gemini API로 포토리얼 기업 로비 배경을 생성하고,
HTML/CSS + Playwright로 한글 텍스트를 정밀하게 오버레이합니다.
출력:
- /home/jay/workspace/output/google-ads/banners/m1/m1-1-1200x628.png
- /home/jay/workspace/output/google-ads/banners/m1/m1-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
# ─── 경로 설정 ────────────────────────────────────────────────────────────────
TOOL_DIR = Path(__file__).parent
sys.path.insert(0, str(TOOL_DIR))
import gcloud_auth # noqa: E402
from gen_config import CTA_MIN_PX, SUBHEAD_MIN_PX, WORKSPACE_ROOT
# ── 디자인 토큰 ──
_CTA_PX = CTA_MIN_PX
_SIZE_44PX = 44
_SIZE_48PX = 48
_SIZE_58PX = 58
_SUBHEAD_PX = SUBHEAD_MIN_PX
_LH_48PX = 48
_LH_52PX = 52
_LH_68PX = 68
_LH_1_2 = 1.2
_LH_1_4 = 1.4
OUTPUT_DIR = WORKSPACE_ROOT / "output/google-ads/banners/m1"
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
TMP_DIR = TOOL_DIR / "output" / "v4-hybrid" / "m1-1-fair"
TMP_DIR.mkdir(parents=True, exist_ok=True)
# ─── Gemini API 설정 ──────────────────────────────────────────────────────────
GEMINI_API_BASE = "https://generativelanguage.googleapis.com/v1beta"
GEMINI_SCOPE = "https://www.googleapis.com/auth/generative-language"
MODEL_ID = "gemini-2.0-flash-preview-image-generation"
FALLBACK_MODEL_ID = "gemini-2.5-flash-image"
# ─── 배경 프롬프트 ────────────────────────────────────────────────────────────
BG_PROMPT = (
"Photographic scene of a modern corporate headquarters lobby in Seoul. "
"Wide shot, shot from low angle (waist height), looking slightly upward. "
"Floor-to-ceiling glass walls revealing city skyline in the background. "
"Clean white marble floor with subtle dark veining. Warm indirect lighting "
"from recessed ceiling fixtures casting a soft glow. Early morning light "
"filtering through the glass, golden hour quality. One male professional "
"in his mid-30s, rear view only — no face visible — wearing a well-fitted "
"dark navy suit, walking confidently toward the elevator bank in the "
"middle distance. The person occupies roughly the right third of the frame, "
"leaving the left two-thirds open for text overlay. Shallow depth of field, "
"the background city slightly blurred. Shot on full-frame DSLR quality, "
"f/2.8, natural cinematic color grading. Corporate, trustworthy, aspirational."
)
# ─── HTML 템플릿 ──────────────────────────────────────────────────────────────
HTML_1200x628 = f"""
코스닥 상장사 ◆ 인카금융서비스
정당한 보상을
받고 계신가요?
투명한 수수료 구조 | 정착률 99%
조건 확인하기 →
"""
HTML_1080x1080 = f"""
코스닥 상장사 ◆ 인카금융서비스
정당한 보상을
받고 계신가요?
투명한 수수료 구조 | 정착률 99%
경력직 정착금 최대 50%
조건 확인하기 →
"""
# ─── Gemini 이미지 생성 ───────────────────────────────────────────────────────
def generate_bg_image(token: str, output_path: Path) -> Path:
"""Gemini API로 배경 이미지를 생성하고 저장합니다."""
print(f"[배경] Gemini API로 기업 로비 배경 생성 중... (모델: {MODEL_ID})")
start = time.time()
url = f"{GEMINI_API_BASE}/models/{MODEL_ID}:generateContent"
headers = {
"Authorization": f"Bearer {token}",
"Content-Type": "application/json",
}
payload = {
"contents": [{"parts": [{"text": BG_PROMPT}]}],
"generationConfig": {"responseModalities": ["IMAGE", "TEXT"]},
}
response = requests.post(url, headers=headers, json=payload, timeout=300)
# fallback
if response.status_code in (400, 403, 404):
print(f"[배경] 모델 {MODEL_ID} 실패 (HTTP {response.status_code}). Fallback: {FALLBACK_MODEL_ID}")
url2 = f"{GEMINI_API_BASE}/models/{FALLBACK_MODEL_ID}:generateContent"
response = requests.post(url2, headers=headers, json=payload, timeout=300)
if not response.ok:
print(f"[배경] API 오류: {response.status_code} - {response.text[:400]}")
response.raise_for_status()
data = response.json()
candidates = data.get("candidates", [])
if not candidates:
raise RuntimeError(f"응답에 candidates 없음: {json.dumps(data)[:300]}")
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:
text_parts = [p.get("text", "") for p in parts if "text" in p]
raise RuntimeError(f"이미지 데이터 없음. 텍스트: {text_parts[:2]}")
mime_type = image_part["inlineData"].get("mimeType", "image/jpeg")
image_bytes = base64.b64decode(image_part["inlineData"]["data"])
ext = ".jpg" if "jpeg" in mime_type else ".png"
final_path = output_path.with_suffix(ext)
final_path.write_bytes(image_bytes)
elapsed = time.time() - start
print(f"[배경] 완료: {final_path.name} ({len(image_bytes):,} bytes, {elapsed:.1f}초)")
return final_path
# ─── Playwright HTML → PNG 캡처 ───────────────────────────────────────────────
def capture_html_to_png(
html_content: str,
bg_path: Path,
output_path: Path,
width: int,
height: int,
) -> None:
"""HTML 템플릿에 배경 이미지 URL을 삽입하고 Playwright로 PNG 캡처합니다."""
bg_url = f"file://{bg_path.resolve()}"
html_filled = html_content.format(bg_url=bg_url)
# 임시 HTML 파일 저장
tmp_html = TMP_DIR / f"tmp_{width}x{height}.html"
tmp_html.write_text(html_filled, encoding="utf-8")
print(f"[캡처] Playwright로 {width}x{height} 캡처 중...")
with sync_playwright() as p:
browser = p.chromium.launch(args=["--no-sandbox", "--disable-gpu"])
try:
page = browser.new_page(viewport={"width": width, "height": height})
page.goto(f"file://{tmp_html.resolve()}", wait_until="networkidle")
# 폰트 로딩 대기
page.wait_for_timeout(2500)
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}")
finally:
browser.close()
tmp_html.unlink(missing_ok=True)
# ─── 메인 ─────────────────────────────────────────────────────────────────────
def main() -> None:
print("=" * 60)
print("M1-1 '정당한 대우' Google 배너 생성 시작")
print("=" * 60)
# 1. 인증 토큰 획득
print("\n[인증] SA 토큰 획득 중...")
try:
token = gcloud_auth.get_access_token()
print(f"[인증] 토큰 획득 성공 (길이: {len(token)} chars)")
except Exception as e:
print(f"[인증] 실패: {e}")
raise SystemExit(1) from e
# 2. Gemini 배경 이미지 생성 (공유 배경) — 이미 존재하면 재사용
print("\n[배경] 배경 이미지 확인 중...")
bg_tmp_path = TMP_DIR / "bg_m1_1_fair"
existing = list(TMP_DIR.glob("bg_m1_1_fair.*"))
if existing:
bg_path = existing[0]
print(f"[배경] 기존 이미지 재사용: {bg_path.name}")
else:
try:
bg_path = generate_bg_image(token, bg_tmp_path)
except Exception as e:
print(f"[배경] 생성 실패: {e}")
raise SystemExit(1) from e
# 3. 배너 1: 1200x628 가로형
print("\n[배너 1] 1200x628 가로형 생성 중...")
output_1200 = OUTPUT_DIR / "m1-1-1200x628.png"
try:
capture_html_to_png(
html_content=HTML_1200x628,
bg_path=bg_path,
output_path=output_1200,
width=1200,
height=628,
)
size_kb = output_1200.stat().st_size / 1024
print(f"[배너 1] 완료: {output_1200} ({size_kb:.0f} KB)")
except Exception as e:
print(f"[배너 1] 실패: {e}")
import traceback; traceback.print_exc()
# 4. 배너 2: 1080x1080 정사각형
print("\n[배너 2] 1080x1080 정사각형 생성 중...")
output_1080 = OUTPUT_DIR / "m1-1-1080x1080.png"
try:
capture_html_to_png(
html_content=HTML_1080x1080,
bg_path=bg_path,
output_path=output_1080,
width=1080,
height=1080,
)
size_kb = output_1080.stat().st_size / 1024
print(f"[배너 2] 완료: {output_1080} ({size_kb:.0f} KB)")
except Exception as e:
print(f"[배너 2] 실패: {e}")
import traceback; traceback.print_exc()
print("\n" + "=" * 60)
print("생성 완료")
print(f" 배너 1: {output_1200}")
print(f" 배너 2: {output_1080}")
print("=" * 60)
if __name__ == "__main__":
main()