#!/usr/bin/env python3
from playwright.sync_api import sync_playwright
import os

BASE = "/home/jay/workspace/output/banners/versions/v-round2/cell-8-snu-leader"

BANNERS = [
    {
        "html": os.path.join(BASE, "meta-feed-1080x1080.html"),
        "png":  os.path.join(BASE, "meta-feed-1080x1080.png"),
        "w": 1080, "h": 1080,
    },
    {
        "html": os.path.join(BASE, "google-resp-1200x628.html"),
        "png":  os.path.join(BASE, "google-resp-1200x628.png"),
        "w": 1200, "h": 628,
    },
]

with sync_playwright() as p:
    browser = p.chromium.launch()
    for b in BANNERS:
        page = browser.new_page(viewport={"width": b["w"], "height": b["h"]})
        page.goto(f"file://{b['html']}", wait_until="networkidle")
        page.wait_for_timeout(2500)
        page.screenshot(
            path=b["png"],
            type="png",
            clip={"x": 0, "y": 0, "width": b["w"], "height": b["h"]}
        )
        page.close()
        size = os.path.getsize(b["png"])
        print(f"[OK] {b['png']}  ({size:,} bytes)")
    browser.close()

print("Rendering complete.")