#!/usr/bin/env python3
"""컨셉 #27 Kia Style — 대담한 에너지 이미지 생성"""
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, FONT_DIR, HEAD_SUB_RATIO, SUBHEAD_MIN_PX
sys.path.insert(0, str(Path(__file__).parent))
import gcloud_auth
_SIZE_22PX = 22
_SIZE_44PX = 44
_SUBHEAD_PX = SUBHEAD_MIN_PX
_SIZE_104PX = 104
_LH_RATIO = HEAD_SUB_RATIO
_LH_1_1 = 1.1
# ── 경로 설정 ──────────────────────────────────────────────────────────────
OUTPUT_DIR = WORKSPACE_ROOT / "output/meta-ads/concept-catalog/27-kia-style"
BG_PATH = OUTPUT_DIR / "bg.jpg"
SAMPLE_PATH = OUTPUT_DIR / "sample.png"
GEMINI_API_BASE = "https://generativelanguage.googleapis.com/v1beta"
BG_PROMPT = (
"Dynamic diagonal split composition background, "
"bottom-left area dark charcoal #1A1A1A, top-right area vivid red #BB162B, "
"sharp diagonal edge at 15-degree angle cutting across the frame from bottom-left to top-right, "
"subtle motion blur texture suggesting speed and energy, "
"high energy bold graphic design style, "
"no text, no people, no logos, 1080x1080 square"
)
HTML_TEMPLATE = f"""
열심히는 하는데, 결과가 없다면
월급 제자리걸음?
이제 판을 바꿔!
CHANGE NOW
"""
def generate_background() -> bool:
"""Gemini SA 토큰으로 배경 이미지를 생성합니다."""
sa_token = gcloud_auth.get_service_account_token("https://www.googleapis.com/auth/generative-language")
if not sa_token:
print(" [오류] SA 토큰 없음")
return False
headers = {"Authorization": f"Bearer {sa_token}", "Content-Type": "application/json"}
url = f"{GEMINI_API_BASE}/models/gemini-3.1-flash-image-preview:generateContent"
payload = {
"contents": [{"parts": [{"text": BG_PROMPT}]}],
"generationConfig": {"responseModalities": ["TEXT", "IMAGE"]},
}
print(" 배경 생성 요청 중...")
resp = requests.post(url, headers=headers, json=payload, timeout=120)
if resp.status_code != 200:
print(f" [오류] {resp.status_code}: {resp.text[:300]}")
return False
data = resp.json()
for part in data.get("candidates", [{}])[0].get("content", {}).get("parts", []):
if "inlineData" in part:
img_bytes = base64.b64decode(part["inlineData"]["data"])
BG_PATH.write_bytes(img_bytes)
print(f" 배경 저장 완료: {BG_PATH}")
return True
print(" [오류] 이미지 파트 없음")
return False
def render_html(html_content: str) -> bool:
"""Playwright로 HTML을 PNG로 캡처합니다."""
html_file = OUTPUT_DIR / "overlay.html"
html_file.write_text(html_content, encoding="utf-8")
with sync_playwright() as p:
browser = p.chromium.launch()
page = browser.new_page(viewport={"width": 1080, "height": 1080})
page.goto(f"file://{html_file}")
page.wait_for_timeout(1500)
page.screenshot(path=str(SAMPLE_PATH), type="png")
browser.close()
print(f" PNG 저장 완료: {SAMPLE_PATH}")
return True
def main():
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
print("[1/2] 배경 이미지 생성...")
ok = generate_background()
if not ok:
sys.exit(1)
print("[2/2] HTML 오버레이 렌더링...")
bg_url = f"file://{BG_PATH}"
html = HTML_TEMPLATE.replace("__BG_PATH__", bg_url)
render_html(html)
print(f"\n완료: {SAMPLE_PATH}")
if __name__ == "__main__":
main()