#!/usr/bin/env python3
"""컨셉 #07 Impact Number v2 - hybrid-image 파이프라인
Gemini 배경 생성 + Playwright HTML 오버레이 합성
"""
from __future__ import annotations
import base64
import json
import shutil
import sys
import time
from pathlib import Path
from typing import Any
import requests
# ─── 경로 설정 ────────────────────────────────────────────────────────────────
SCRIPT_DIR = Path(__file__).parent
sys.path.insert(0, str(SCRIPT_DIR))
OUTPUT_DIR = Path("/home/jay/workspace/output/meta-ads/concept-catalog/07-impact-number")
BG_PATH = OUTPUT_DIR / "bg_v2_impact.jpg"
SAMPLE_PATH = OUTPUT_DIR / "sample-v2.png"
FINAL_PATH = OUTPUT_DIR / "07-impact-number-v2.png"
HTML_PATH = OUTPUT_DIR / "overlay_impact_v2.html"
FONT_DIR = Path.home() / ".local/share/fonts/Pretendard"
# ─── Gemini 배경 프롬프트 ──────────────────────────────────────────────────────
BG_PROMPT = (
"Very dark navy blue background (#121928), minimal abstract composition. "
"Subtle golden radial glow emanating from center, soft warm amber-gold light diffusion. "
"Pure dark gradient background, no patterns, no text, no objects, no people. "
"Just deep dark navy with a gentle golden center glow. "
"Clean, simple, elegant. 1080x1080 square format."
)
# ─── HTML 오버레이 템플릿 ──────────────────────────────────────────────────────
HTML_TEMPLATE = """
신입 정착지원금 최대
1,000만원
2026년 7월, 조건이 바뀝니다
지금 상담 신청하기
"""
# ─── Gemini 배경 생성 ──────────────────────────────────────────────────────────
def generate_background(token: str) -> bool:
"""Gemini API로 배경 이미지 생성 후 BG_PATH에 저장."""
models = ["gemini-2.0-flash-preview-image-generation", "gemini-3-pro-image-preview", "gemini-3.1-flash-image-preview"]
api_base = "https://generativelanguage.googleapis.com/v1beta"
headers = {
"Authorization": f"Bearer {token}",
"Content-Type": "application/json",
}
payload: dict[str, Any] = {
"contents": [{"parts": [{"text": BG_PROMPT}]}],
"generationConfig": {"responseModalities": ["IMAGE", "TEXT"]},
}
for model_id in models:
url = f"{api_base}/models/{model_id}:generateContent"
print(f"[배경] Gemini 모델 시도: {model_id}")
try:
resp = requests.post(url, headers=headers, json=payload, timeout=120)
if resp.status_code in (403, 404):
print(f" → HTTP {resp.status_code}, 다음 모델로 시도...")
continue
resp.raise_for_status()
data = resp.json()
candidates = data.get("candidates", [])
if not candidates:
print(f" → 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" → 이미지 파트 없음, 다음 모델로...")
continue
image_bytes = base64.b64decode(image_part["inlineData"]["data"])
BG_PATH.write_bytes(image_bytes)
print(f" → 배경 저장: {BG_PATH} ({len(image_bytes):,} bytes)")
return True
except Exception as e:
print(f" → 오류: {e}")
continue
return False
# ─── Playwright 캡처 ──────────────────────────────────────────────────────────
def capture_overlay() -> None:
"""HTML 오버레이를 Playwright로 캡처하여 SAMPLE_PATH에 저장."""
from playwright.sync_api import sync_playwright
# HTML 파일 저장
html_content = HTML_TEMPLATE.format(
font_dir=str(FONT_DIR),
bg_path=str(BG_PATH) if BG_PATH.exists() else "",
)
HTML_PATH.write_text(html_content, encoding="utf-8")
print(f"[HTML] 템플릿 저장: {HTML_PATH}")
with sync_playwright() as p:
browser = p.chromium.launch()
try:
page = browser.new_page(viewport={"width": 1080, "height": 1080})
page.goto(f"file://{HTML_PATH.resolve()}", wait_until="networkidle")
# 폰트 렌더링 대기
page.wait_for_timeout(2000)
SAMPLE_PATH.parent.mkdir(parents=True, exist_ok=True)
page.screenshot(path=str(SAMPLE_PATH), type="png")
print(f"[캡처] {SAMPLE_PATH} ({SAMPLE_PATH.stat().st_size:,} bytes)")
finally:
browser.close()
# ─── 메인 ─────────────────────────────────────────────────────────────────────
def main() -> None:
import gcloud_auth
print("=" * 60)
print("컨셉 #07 Impact Number v2 — hybrid-image 생성")
print("=" * 60)
# 1. 인증
print("\n[인증] 토큰 획득 중...")
try:
token = gcloud_auth.get_access_token()
print(f"[인증] 토큰 획득 성공 ({len(token)} chars)")
except Exception as e:
print(f"[인증 실패] {e}")
print("[폴백] 배경 없이 단색 배경으로 진행합니다.")
token = None
# 2. 배경 생성
if token:
print("\n[배경] Gemini 이미지 생성 중...")
bg_ok = generate_background(token)
if not bg_ok:
print("[배경] 모든 모델 실패 — 단색 배경 폴백")
else:
bg_ok = False
# 배경이 없으면 HTML에서 CSS 단색 배경으로 처리 (canvas에 background: #121928)
if not bg_ok:
# 빈 BG_PATH 처리 — HTML 템플릿의 background-image가 빈 값이면 CSS background-color로 폴백
print("[배경] CSS 단색 다크 배경 + 황금 글로우로 대체합니다.")
# 3. Playwright 캡처
print("\n[캡처] HTML 오버레이 렌더링 중...")
capture_overlay()
# 4. 복사
shutil.copy2(SAMPLE_PATH, FINAL_PATH)
print(f"[복사] {FINAL_PATH}")
print("\n" + "=" * 60)
print("생성 완료!")
print(f" sample-v2.png : {SAMPLE_PATH}")
print(f" 07-impact-number-v2.png : {FINAL_PATH}")
print("=" * 60)
if __name__ == "__main__":
main()