#!/usr/bin/env python3
"""
Meta carousel slide A-5 (CTA) 생성 스크립트
Hybrid pipeline: Gemini background + HTML/CSS Playwright overlay
Output: /home/jay/workspace/output/meta-ads/a-group-v6/pilot/pilot-A5-cta.png
"""
import base64
import json
import sys
import time
from pathlib import Path
import requests
from playwright.sync_api import sync_playwright
from gen_config import WORKSPACE_ROOT
# ──────────────────────────────────────────────────────
# 경로 설정
# ──────────────────────────────────────────────────────
TOOL_DIR = Path(__file__).parent
OUTPUT_DIR = WORKSPACE_ROOT / "output/meta-ads/a-group-v6/pilot"
BG_PATH = OUTPUT_DIR / "pilot-A5-bg.jpg"
HTML_PATH = OUTPUT_DIR / "pilot-A5-overlay.html"
FINAL_PATH = OUTPUT_DIR / "pilot-A5-cta.png"
# gcloud_auth 모듈 경로 추가
sys.path.insert(0, str(TOOL_DIR))
import gcloud_auth
_SIZE_22PX = 22
_SIZE_26PX = 26
_SIZE_36PX = 36
_SIZE_46PX = 46
_SIZE_68PX = 68
_LH_1_22 = 1.22
_LH_1_65 = 1.65
# ──────────────────────────────────────────────────────
# Gemini API 배경 생성
# ──────────────────────────────────────────────────────
GEMINI_BG_PROMPT = """
Create a cinematic, premium insurance consultation advertisement background image.
COMPOSITION: Square 1:1 format (1080x1080px). Abstract architectural space.
SCENE: Dark, deep navy interior space — like the threshold of an important decision.
View from slightly below center, looking toward a wide-open double door or archway.
Through the opening: warm golden-amber volumetric light cascades in dramatic god-rays,
illuminating floating dust particles. The space beyond is warm, brilliant, inviting.
The foreground/corridor is deep navy (#1A2F5E) fading to almost black at corners.
LIGHTING: Extreme chiaroscuro. Deep navy darkness in corners and edges.
Brilliant golden-amber glow (#C9A84C tones) streaming through the center opening.
Volumetric light rays — like sunlight through cathedral glass. Cinematic depth.
Subtle radial glow emanating from center-top area of the image.
COLOR GRADE: Deep navy (#1A2F5E) dominant background. Warm gold (#C9A84C) light source center.
Rich navy-to-gold gradient across the image. No cold blues, no grey. Warm and vivid.
STYLE: Like Denis Villeneuve / Christopher Nolan cinematography. Ultra-premium.
Clean, abstract — NO people, NO faces, NO text, NO logos, NO UI elements.
No arrows, no buttons, no symbols. Pure atmospheric cinematic background only.
MOOD: The warmest, most vivid, most hopeful slide of a 5-slide series.
"A door opening at the end of a long journey." Clear, solid energy. Decisive.
ASPECT: Exactly 1:1 square, 1080x1080px. Maximum cinematic quality.
"""
MODELS_TO_TRY = [
"gemini-2.5-flash-image",
"gemini-3-pro-image-preview",
"gemini-3.1-flash-image-preview",
]
def generate_gemini_background(token: str, output_path: Path) -> bool:
"""Gemini REST API로 배경 이미지 생성 (SA Bearer 토큰 사용)."""
headers = {
"Authorization": f"Bearer {token}",
"Content-Type": "application/json",
}
payload = {
"contents": [{"parts": [{"text": GEMINI_BG_PROMPT}]}],
"generationConfig": {"responseModalities": ["IMAGE", "TEXT"]},
}
for model_id in MODELS_TO_TRY:
url = f"https://generativelanguage.googleapis.com/v1beta/models/{model_id}:generateContent"
print(f"[배경 생성] 모델 {model_id} 호출 중...")
start = time.time()
try:
resp = requests.post(url, headers=headers, json=payload, timeout=180)
except requests.RequestException as e:
print(f"[WARN] 요청 실패: {e}")
continue
if resp.status_code in (403, 404, 429):
print(f"[WARN] HTTP {resp.status_code} — 다음 모델 시도")
continue
try:
resp.raise_for_status()
except requests.HTTPError as e:
print(f"[ERROR] HTTP {e.response.status_code}: {e.response.text[:300]}")
return False
elapsed = time.time() - start
data = resp.json()
candidates = data.get("candidates", [])
if not candidates:
print(f"[ERROR] candidates 없음: {json.dumps(data)[:300]}")
return False
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]
print(f"[ERROR] 이미지 파트 없음. 텍스트: {text_parts[:2]}")
return False
image_bytes = base64.b64decode(image_part["inlineData"]["data"])
output_path.write_bytes(image_bytes)
print(f"[배경 생성] 완료: {output_path} ({len(image_bytes):,} bytes, {elapsed:.1f}초)")
return True
print("[ERROR] 모든 모델 시도 실패.")
return False
# ──────────────────────────────────────────────────────
# HTML 오버레이 템플릿 생성
# ──────────────────────────────────────────────────────
HTML_TEMPLATE = """
Meta Ad A-5 CTA
5 / 5
서울대보험쌤
전략을 바꿀 준비가 됐다면, 지금 딱 한 번 물어보세요.
서울대보험쌤 팀장이 직접 답합니다. 조건 없는 상담, 당신 상황에 맞는 방향 제시.
무료 상담 신청하기
조건 없음
무료 상담
팀장 직접 응대
"""
def create_html(bg_path: Path, output_html_path: Path) -> None:
"""HTML 오버레이 파일 생성."""
html_content = HTML_TEMPLATE.format(bg_path=str(bg_path))
output_html_path.write_text(html_content, encoding="utf-8")
print(f"[HTML] 오버레이 생성: {output_html_path}")
def capture_screenshot(html_path: Path, output_png_path: Path) -> None:
"""Playwright로 HTML → PNG 캡처."""
print("[캡처] Playwright 실행 중...")
with sync_playwright() as p:
browser = p.chromium.launch(args=["--no-sandbox", "--disable-dev-shm-usage"])
try:
page = browser.new_page(viewport={"width": 1080, "height": 1080})
page.goto(f"file://{html_path.resolve()}", wait_until="networkidle")
page.wait_for_timeout(2500) # 폰트 + 이미지 로딩 대기
output_png_path.parent.mkdir(parents=True, exist_ok=True)
page.screenshot(path=str(output_png_path), type="png")
print(f"[캡처] 완료: {output_png_path}")
finally:
browser.close()
def main():
print("=" * 60)
print("Meta 캐러셀 A-5 CTA 슬라이드 생성")
print("=" * 60)
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
# 1. SA Bearer 토큰 획득
print("[인증] SA 서비스 계정 토큰 획득 중...")
try:
token = gcloud_auth.get_service_account_token()
print(f"[인증] SA 토큰 획득 완료 (길이: {len(token)})")
except Exception as e:
print(f"[ERROR] SA 토큰 획득 실패: {e}")
sys.exit(1)
# 2. 배경 이미지 생성
if BG_PATH.exists():
print(f"[배경] 기존 배경 이미지 재사용: {BG_PATH}")
else:
success = generate_gemini_background(token, BG_PATH)
if not success:
print("[ERROR] 배경 이미지 생성 실패. 중단합니다.")
sys.exit(1)
# 3. HTML 오버레이 생성
create_html(BG_PATH, HTML_PATH)
# 4. Playwright 캡처
capture_screenshot(HTML_PATH, FINAL_PATH)
# 5. 완료 보고
if FINAL_PATH.exists():
size_kb = FINAL_PATH.stat().st_size / 1024
print(f"\n[완료] {FINAL_PATH}")
print(f" 파일 크기: {size_kb:.0f} KB")
print(f" 해상도: 1080x1080px")
else:
print("[ERROR] 최종 이미지 파일이 생성되지 않았습니다.")
sys.exit(1)
if __name__ == "__main__":
main()