#!/usr/bin/env python3 """M3-2 서울대보험쌤 × 관리자 비전 Google 광고 배너 생성. Gemini AI 배경 + Playwright HTML 오버레이 하이브리드 방식. 출력: snu-leader-1200x628.png, snu-leader-1080x1080.png """ from __future__ import annotations import base64 import sys import time from pathlib import Path import requests # gcloud_auth 임포트를 위해 경로 추가 sys.path.insert(0, "/home/jay/workspace/tools/ai-image-gen") import gcloud_auth from playwright.sync_api import sync_playwright OUTPUT_DIR = Path("/home/jay/workspace/output/google-ads/banners") OUTPUT_DIR.mkdir(parents=True, exist_ok=True) FONT_DIR = Path.home() / ".local/share/fonts/Pretendard" # Gemini 모델 MODEL_ID = "gemini-2.0-flash-preview-image-generation" FALLBACK_MODEL_ID = "gemini-2.0-flash-exp" GEMINI_API_BASE = "https://generativelanguage.googleapis.com/v1beta" # 배경 이미지 프롬프트 (공유 오피스 장면) BG_PROMPT = ( "Photo of a modern shared office space in Seoul. Wide shot showing individual " "workstations and a meeting area coexisting. On one side, a private office is visible " "through glass partitions. Bright natural sunlight streaming through large windows. " "Plant decorations throughout, warm wooden tone furniture. The look of a growing " "organization in its early stages. Photorealistic quality, warm yet organized atmosphere. " "No people. Warm tones, beige and wood colors dominating. Soft shadows, professional " "interior photography style. High resolution, sharp focus." ) def generate_gemini_background(output_path: Path) -> bool: """Gemini API로 배경 이미지 생성 (SA 토큰 우선).""" print(f"[배경] Gemini 이미지 생성 중...") # SA 토큰 우선 시도 sa_token = None try: sa_token = gcloud_auth.get_service_account_token() print(f"[배경] SA Bearer 토큰 인증 사용 (길이: {len(sa_token)})") except Exception as e: print(f"[배경] SA 토큰 실패: {e}, API Key fallback 시도...") if sa_token: headers = { "Authorization": f"Bearer {sa_token}", "Content-Type": "application/json", } models_to_try = ["gemini-3-pro-image-preview", "gemini-3.1-flash-image-preview"] base_url_fmt = f"{GEMINI_API_BASE}/models/{{model}}:generateContent" else: api_key = gcloud_auth.get_api_key("GEMINI_API_KEY") if not api_key: print("[오류] 인증 수단 없음") return False headers = {"Content-Type": "application/json"} models_to_try = ["gemini-3-pro-image-preview", "gemini-3.1-flash-image-preview", "gemini-2.5-flash-image"] base_url_fmt = f"{GEMINI_API_BASE}/models/{{model}}:generateContent?key={api_key}" payload = { "contents": [{"parts": [{"text": BG_PROMPT}]}], "generationConfig": {"responseModalities": ["IMAGE", "TEXT"]}, } for model in models_to_try: url = base_url_fmt.format(model=model) print(f"[배경] 모델 시도: {model}") try: start = time.time() resp = requests.post(url, headers=headers, json=payload, timeout=120) if resp.status_code not in (200,): print(f"[배경] {model} 실패 ({resp.status_code}): {resp.text[:150]}") continue data = resp.json() parts = data.get("candidates", [{}])[0].get("content", {}).get("parts", []) image_part = next((p for p in parts if "inlineData" in p), None) if not image_part: print(f"[배경] {model}: 이미지 없음. {[p.get('text','') for p in parts][:1]}") continue image_bytes = base64.b64decode(image_part["inlineData"]["data"]) output_path.write_bytes(image_bytes) elapsed = time.time() - start print(f"[배경] 저장 완료: {output_path} ({len(image_bytes):,} bytes, {elapsed:.1f}초)") return True except Exception as e: print(f"[오류] {model} 호출 실패: {type(e).__name__}: {e}") continue print("[오류] 모든 모델 실패") return False # ──────────────────────────────────────────────────────────────── # HTML 템플릿: 1200x628 (가로형) # ──────────────────────────────────────────────────────────────── def build_html_1200x628(bg_image_path: str) -> str: font_dir = str(FONT_DIR) return f"""
서울대보험쌤
서울대 출신 지점장과
함께 시작
AI자동화 | 멘토링 | 100일 로드맵
조직 설계 상담 받기 →
""" # ──────────────────────────────────────────────────────────────── # HTML 템플릿: 1080x1080 (정사각형) # ──────────────────────────────────────────────────────────────── def build_html_1080x1080(bg_image_path: str) -> str: font_dir = str(FONT_DIR) return f"""
서울대보험쌤
서울대 출신 지점장과
함께 시작
AI자동화 | 멘토링 | 100일 로드맵
조직 설계 상담 받기 →
incar-top.tistory.com
""" def capture_banner(html_content: str, output_path: Path, width: int, height: int) -> bool: """Playwright로 HTML을 캡처하여 PNG 저장.""" html_file = output_path.with_suffix(".html") html_file.write_text(html_content, encoding="utf-8") print(f"[HTML] 템플릿 저장: {html_file}") try: with sync_playwright() as p: browser = p.chromium.launch() try: page = browser.new_page(viewport={"width": width, "height": height}) page.goto(f"file://{html_file.resolve()}", wait_until="networkidle") page.wait_for_timeout(2500) # 폰트 렌더링 대기 page.screenshot(path=str(output_path), type="png") finally: browser.close() size_kb = output_path.stat().st_size / 1024 print(f"[캡처] 저장 완료: {output_path} ({size_kb:.0f} KB)") return True except Exception as e: print(f"[오류] 캡처 실패: {type(e).__name__}: {e}") return False def main() -> None: print("=" * 60) print("M3-2 서울대보험쌤 × 관리자 비전 배너 생성") print("=" * 60) bg_path = OUTPUT_DIR / "snu_leader_bg.jpg" # 1. Gemini 배경 생성 if bg_path.exists(): print(f"[배경] 기존 이미지 재사용: {bg_path}") bg_ok = True else: bg_ok = generate_gemini_background(bg_path) if not bg_ok: print("[경고] 배경 생성 실패. 크림 단색 배경으로 대체합니다.") # fallback: 빈 크림 배경 (배경 없이도 진행) bg_image_path = "" else: bg_image_path = str(bg_path.resolve()) results = {} # 2. 1200x628 가로형 배너 print("\n[배너 1] 1200x628 가로형 생성 중...") output_1200 = OUTPUT_DIR / "snu-leader-1200x628.png" html_1200 = build_html_1200x628(bg_image_path) ok1 = capture_banner(html_1200, output_1200, 1200, 628) results["1200x628"] = {"path": str(output_1200), "success": ok1} # 3. 1080x1080 정사각형 배너 print("\n[배너 2] 1080x1080 정사각형 생성 중...") output_1080 = OUTPUT_DIR / "snu-leader-1080x1080.png" html_1080 = build_html_1080x1080(bg_image_path) ok2 = capture_banner(html_1080, output_1080, 1080, 1080) results["1080x1080"] = {"path": str(output_1080), "success": ok2} # 4. 결과 요약 print("\n" + "=" * 60) print("생성 결과 요약") print("=" * 60) for size, r in results.items(): status = "성공" if r["success"] else "실패" print(f" [{size}] {status}: {r['path']}") if __name__ == "__main__": main()