#!/usr/bin/env python3 """GA 배너 M2-3 영업지원 세트 생성 스크립트. Gemini 배경 + HTML/CSS 오버레이 하이브리드 방식. - 1200x628 (가로형) - 1080x1080 (정사각형) """ from __future__ import annotations import base64 import sys import time from pathlib import Path import requests from playwright.sync_api import sync_playwright sys.path.insert(0, str(Path(__file__).parent)) import gcloud_auth # ───────────────────────────────────────────────────────────────────────────── # 경로 설정 # ───────────────────────────────────────────────────────────────────────────── OUTPUT_DIR = Path("/home/jay/workspace/output/google-ads/banners") FONT_DIR = Path.home() / ".local/share/fonts/Pretendard" BG_PATH = OUTPUT_DIR / "_ga_support_bg.jpg" HTML_1200 = OUTPUT_DIR / "_ga_support_1200x628.html" HTML_1080 = OUTPUT_DIR / "_ga_support_1080x1080.html" OUT_1200 = OUTPUT_DIR / "ga-support-1200x628.png" OUT_1080 = OUTPUT_DIR / "ga-support-1080x1080.png" # Gemini 배경 프롬프트 GEMINI_PROMPT = ( "A bright and warm small educational space scene. " "2 to 3 people sit around a round table looking at laptops and having a natural conversation. " "Natural light streaming through windows, indoor green plants, clean white modern interior. " "Collaborative and communicative comfortable work environment. " "Mid-range to distant shot of people, not too close. " "Photo realistic quality, warm tones, bright and airy atmosphere. " "High resolution photograph scene." ) GEMINI_API_BASE = "https://generativelanguage.googleapis.com/v1beta" GEMINI_SCOPE = "https://www.googleapis.com/auth/generative-language" # ───────────────────────────────────────────────────────────────────────────── # 배경 생성 # ───────────────────────────────────────────────────────────────────────────── def generate_background() -> Path: """Gemini API로 배경 이미지 생성.""" print("[배경] Gemini API 토큰 획득 중...") token = gcloud_auth.get_service_account_token(GEMINI_SCOPE) print(f"[배경] 토큰 획득 완료 ({len(token)} chars)") models_to_try = [ "gemini-2.0-flash-preview-image-generation", "gemini-3-pro-image-preview", "imagen-3.0-generate-001", ] for model_id in models_to_try: print(f"[배경] 모델 시도: {model_id}") url = f"{GEMINI_API_BASE}/models/{model_id}:generateContent" headers = { "Authorization": f"Bearer {token}", "Content-Type": "application/json", } payload = { "contents": [{"parts": [{"text": GEMINI_PROMPT}]}], "generationConfig": {"responseModalities": ["IMAGE", "TEXT"]}, } try: resp = requests.post(url, headers=headers, json=payload, timeout=120) if resp.status_code in (403, 404): print(f" → {resp.status_code} 접근 불가, 다음 모델 시도...") continue resp.raise_for_status() except requests.HTTPError as e: print(f" → HTTP 오류: {e}, 다음 모델 시도...") continue 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 mime = image_part["inlineData"].get("mimeType", "image/jpeg") img_bytes = base64.b64decode(image_part["inlineData"]["data"]) ext = ".jpg" if "jpeg" in mime else ".png" bg_path = BG_PATH.with_suffix(ext) bg_path.write_bytes(img_bytes) print(f"[배경] 저장 완료: {bg_path} ({len(img_bytes):,} bytes)") return bg_path raise RuntimeError("모든 Gemini 모델 시도 실패") # ───────────────────────────────────────────────────────────────────────────── # HTML 템플릿 빌더 # ───────────────────────────────────────────────────────────────────────────── def font_faces() -> str: """Pretendard 폰트 @font-face 정의.""" weights = [ ("Bold", 700), ("SemiBold", 600), ("Medium", 500), ("Regular", 400), ("ExtraBold", 800), ("Black", 900), ] lines = [] for name, w in weights: fpath = FONT_DIR / f"Pretendard-{name}.otf" lines.append(f""" @font-face {{ font-family: 'Pretendard'; src: url('file://{fpath}') format('opentype'); font-weight: {w}; }}""") return "\n".join(lines) def build_html_1200(bg_path: str) -> str: """1200x628 가로형 배너 HTML.""" ff = font_faces() return f"""
영업지원 GA
영업이 힘들다면,
시스템 있는 GA
DB 제공|교육|함께 성장
영업지원 GA 알아보기 →
""" def build_html_1080(bg_path: str) -> str: """1080x1080 정사각형 배너 HTML.""" ff = font_faces() return f"""
영업지원 GA
영업이 힘들다면,
시스템 있는 GA
✓
DB 제공 — 영업할 고객, 미리 준비
✓
교육 — 입사부터 성장까지 지원
✓
함께 성장 — 팀과 함께하는 커리어
영업지원 GA 알아보기 →
""" # ───────────────────────────────────────────────────────────────────────────── # 캡처 함수 # ───────────────────────────────────────────────────────────────────────────── def capture_banner(html_path: Path, output_path: Path, width: int, height: int) -> None: """Playwright로 HTML 배너 캡처.""" with sync_playwright() as p: browser = p.chromium.launch() try: page = browser.new_page(viewport={"width": width, "height": height}) page.goto(f"file://{html_path.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") finally: browser.close() # ───────────────────────────────────────────────────────────────────────────── # 메인 # ───────────────────────────────────────────────────────────────────────────── def main() -> None: OUTPUT_DIR.mkdir(parents=True, exist_ok=True) # 1. 배경 이미지 생성 print("=" * 60) print("GA 영업지원 배너 생성 시작") print("=" * 60) t0 = time.time() bg_path = generate_background() # 2. HTML 빌드 및 저장 print("\n[HTML] 템플릿 빌드 중...") html_1200 = build_html_1200(str(bg_path.resolve())) HTML_1200.write_text(html_1200, encoding="utf-8") print(f" → {HTML_1200}") html_1080 = build_html_1080(str(bg_path.resolve())) HTML_1080.write_text(html_1080, encoding="utf-8") print(f" → {HTML_1080}") # 3. 캡처 print("\n[캡처] 1200x628 배너...") capture_banner(HTML_1200, OUT_1200, 1200, 628) size_1200 = OUT_1200.stat().st_size print(f" → {OUT_1200} ({size_1200:,} bytes)") print("\n[캡처] 1080x1080 배너...") capture_banner(HTML_1080, OUT_1080, 1080, 1080) size_1080 = OUT_1080.stat().st_size print(f" → {OUT_1080} ({size_1080:,} bytes)") elapsed = time.time() - t0 print("\n" + "=" * 60) print(f"완료! 총 소요 시간: {elapsed:.1f}초") print(f"산출물 1: {OUT_1200}") print(f"산출물 2: {OUT_1080}") print("=" * 60) if __name__ == "__main__": main()