take_ss.py

from playwright.sync_api import sync_playwright

def take_screenshot(url, path):
    with sync_playwright() as p:
        browser = p.firefox.launch()
        page = browser.new_page()
        page.goto(url)
        page.screenshot(path=path)
        browser.close()