#!/usr/bin/env python3
"""인카다이렉트 M1-1 '정당한 대우' Google 배너 생성기
컨셉 #49 Spikes Trust Protection 스타일 — 기업 로비 배경
- 1200x628 (가로형)
- 1080x1080 (정사각형)
"""
import base64
import sys
from pathlib import Path
import requests
from playwright.sync_api import sync_playwright
sys.path.insert(0, "/home/jay/workspace/tools/ai-image-gen")
import gcloud_auth
OUTPUT_DIR = Path("/home/jay/workspace/output/google-ads/banners")
BG_PATH = OUTPUT_DIR / "bg_incar_fair.jpg"
OUT_1200 = OUTPUT_DIR / "incar-fair-1200x628.png"
OUT_1080 = OUTPUT_DIR / "incar-fair-1080x1080.png"
GEMINI_API_BASE = "https://generativelanguage.googleapis.com/v1beta"
BG_PROMPT = (
"Modern corporate headquarters lobby interior, "
"wide floor-to-ceiling glass windows overlooking city skyline, "
"clean polished marble floor with subtle reflections, "
"elegant indirect lighting from ceiling and walls, "
"one professional Korean businessman walking confidently away from camera, "
"business suit, confident upright posture, back view, "
"bright morning natural light streaming through glass, "
"warm golden hour light creating depth and atmosphere, "
"premium corporate aesthetic, photorealistic, "
"high-end Samsung Fire & Marine Insurance level advertisement quality, "
"deep navy and warm gold color tones, "
"spacious modern architecture, "
"no text, no watermarks, no graphics, no charts, no graphs, "
"no handshakes, no meeting rooms, no multiple people, "
"no icons, no illustrations, photo quality"
)
def generate_background() -> bool:
"""Gemini SA 토큰으로 배경 이미지를 생성합니다."""
print(" SA 토큰 획득 중...")
try:
sa_token = gcloud_auth.get_service_account_token(
"https://www.googleapis.com/auth/generative-language"
)
except Exception as e:
print(f" [오류] SA 토큰 획득 실패: {e}")
return False
headers = {
"Authorization": f"Bearer {sa_token}",
"Content-Type": "application/json",
}
models_to_try = [
"gemini-2.0-flash-preview-image-generation",
"gemini-3.1-flash-image-preview",
"gemini-3-pro-image-preview",
]
for model in models_to_try:
url = f"{GEMINI_API_BASE}/models/{model}:generateContent"
payload = {
"contents": [{"parts": [{"text": BG_PROMPT}]}],
"generationConfig": {"responseModalities": ["TEXT", "IMAGE"]},
}
print(f" 배경 생성 요청 중... (모델: {model})")
try:
resp = requests.post(url, headers=headers, json=payload, timeout=120)
except Exception as e:
print(f" [오류] 연결 실패: {e}")
continue
if resp.status_code == 200:
data = resp.json()
parts = data.get("candidates", [{}])[0].get("content", {}).get("parts", [])
for part in parts:
if "inlineData" in part:
img_bytes = base64.b64decode(part["inlineData"]["data"])
BG_PATH.write_bytes(img_bytes)
print(f" 배경 저장 완료: {BG_PATH} ({len(img_bytes):,} bytes)")
return True
reason = data.get("candidates", [{}])[0].get("finishReason", "?")
print(f" [오류] 이미지 파트 없음 (finishReason: {reason})")
else:
print(f" [HTTP {resp.status_code}] {resp.text[:300]}")
return False
# ─── 1200x628 가로형 HTML ──────────────────────────────────────────────────
HTML_1200 = """
코스닥 상장사
정당한 보상을
받고 계신가요?
투명한 수수료 구조 · 99% 정착률
조건 확인하기
코스닥 상장 인카금융서비스
"""
# ─── 1080x1080 정사각형 HTML ──────────────────────────────────────────────
HTML_1080 = """
코스닥 상장사
정당한 보상을
받고 계신가요?
투명한 수수료 구조 · 99% 정착률
조건 확인하기
"""
def render_html_to_png(html_content: str, output_path: Path, width: int, height: int) -> bool:
"""Playwright로 HTML을 PNG로 캡처합니다."""
html_file = OUTPUT_DIR / f"overlay_incar_{width}x{height}.html"
bg_url = f"file://{BG_PATH}" if BG_PATH.exists() and BG_PATH.stat().st_size > 0 else ""
html_content = html_content.replace("__BG_PATH__", bg_url)
html_file.write_text(html_content, encoding="utf-8")
with sync_playwright() as p:
browser = p.chromium.launch()
page = browser.new_page(viewport={"width": width, "height": height})
page.goto(f"file://{html_file}")
page.wait_for_timeout(3500) # 웹폰트 로딩 대기
page.screenshot(path=str(output_path), type="png")
browser.close()
if output_path.exists():
size_kb = output_path.stat().st_size / 1024
print(f" 저장 완료: {output_path} ({size_kb:.0f} KB)")
return True
return False
def main():
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
print("=" * 65)
print("인카다이렉트 M1-1 '정당한 대우' Google 배너 생성")
print("컨셉 #49 Spikes Trust Protection — 기업 로비 배경")
print("=" * 65)
print("\n[1/3] Gemini 배경 이미지 생성...")
bg_ok = generate_background()
if not bg_ok:
print(" [경고] 배경 생성 실패. 딥네이비 배경으로 진행합니다.")
print("\n[2/3] 1200x628 가로형 배너 렌더링...")
ok1 = render_html_to_png(HTML_1200, OUT_1200, 1200, 628)
print("\n[3/3] 1080x1080 정사각형 배너 렌더링...")
ok2 = render_html_to_png(HTML_1080, OUT_1080, 1080, 1080)
print("\n" + "=" * 65)
print("결과 요약")
print("=" * 65)
print(f" 1200x628 : {'OK' if ok1 else 'FAIL'} → {OUT_1200}")
print(f" 1080x1080: {'OK' if ok2 else 'FAIL'} → {OUT_1080}")
return 0 if (ok1 and ok2) else 1
if __name__ == "__main__":
sys.exit(main())