from pathlib import Path import shutil from struct import unpack from uuid import uuid4 from selenium import webdriver from selenium.webdriver.chrome.options import Options from selenium.webdriver.chrome.service import Service from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.support.ui import WebDriverWait viewport = (1280, 720) output_directory = Path("artifacts") output_directory.mkdir(parents=True, exist_ok=True) output_file = output_directory / f"example-home-{uuid4().hex}.png"