from playwright.sync_api import sync_playwright

with sync_playwright() as p:
    browser = p.chromium.launch(args=['--no-sandbox', '--disable-setuid-sandbox'])
    context = browser.new_context(
        viewport={"width": 1080, "height": 1080},
        device_scale_factor=1
    )
    page = context.new_page()
    page.goto('http://localhost:8787/melai-fb-ad-v7.html', wait_until='networkidle')
    page.wait_for_timeout(3000)
    page.screenshot(
        path='/home/melai/Documents/melai-fb-ad-v7.png',
        clip={"x": 0, "y": 0, "width": 1080, "height": 1080},
        type='png'
    )
    browser.close()
    print('Done')
