#!/usr/bin/env python3
"""
Take a screenshot of the local bunker-demo index.html to show the widget is embedded.
"""
import asyncio
from pathlib import Path
from playwright.async_api import async_playwright

HTML_PATH = "/mnt/e/genesis-system/deploy/bunker-demo/index.html"
SCREENSHOT_PATH = "/mnt/e/genesis-system/hive/progress/lovable_widget_injected.png"


async def main():
    async with async_playwright() as p:
        browser = await p.chromium.launch(headless=True, args=["--no-sandbox"])
        page = await browser.new_page(viewport={"width": 1440, "height": 900})

        file_url = f"file://{HTML_PATH}"
        print(f"Loading: {file_url}")
        await page.goto(file_url, wait_until="domcontentloaded", timeout=15000)
        await page.wait_for_timeout(3000)

        # Scroll to bottom to show widget area
        await page.evaluate("window.scrollTo(0, document.body.scrollHeight)")
        await page.wait_for_timeout(1000)

        await page.screenshot(path=SCREENSHOT_PATH, full_page=False)
        print(f"Screenshot saved: {SCREENSHOT_PATH}")

        # Also verify widget is in DOM
        widget = await page.query_selector("telnyx-ai-agent")
        if widget:
            agent_id = await widget.get_attribute("agent-id")
            print(f"Widget found in DOM. agent-id: {agent_id}")
        else:
            print("WARNING: Widget element not found in DOM")

        await browser.close()


if __name__ == "__main__":
    asyncio.run(main())