"""shoot.py <url> <out.png> — headless full-page screenshot of a running UI route."""
import asyncio
import sys

from browser_use import BrowserSession
from browser_config import make_profile


async def main():
    url, out = sys.argv[1], sys.argv[2]
    wait = float(sys.argv[3]) if len(sys.argv) > 3 else 4.0
    session = BrowserSession(browser_profile=make_profile(headless=True, persist=False))
    await session.start()
    await session.navigate_to(url)
    await asyncio.sleep(wait)  # let the Next.js client render
    await session.take_screenshot(path=out, full_page=True)
    await session.kill()
    print("saved", out)


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