Python script to take screenshot of webpage and annotate with URL

requirements.txt

appdirs==1.4.4
asyncio==3.4.3
Pillow==7.1.2
pyee==7.0.1
pyppeteer==0.2.2
tqdm==4.46.0
urllib3==1.25.9
websockets==8.1

screenshotr.py

#!/usr/bin/env python3

import os
import sys

import asyncio
import pyppeteer
from PIL import Image, ImageDraw, ImageFont


font_folder = "/Library/Fonts/"
system_fonts_foler = "/System/Library/Fonts/"

font_name = "HelveticaNeue.ttc"

font_size = 19
margin = 15

draw = None

def add_text(img, text):
    pass


def add_text_container(draw, img_size, txt_size):
    ow, oh = img_size
    rh = txt_size[1] + (2*margin)
    rw = txt_size[0] + (2*margin)
    draw.rectangle((0,(oh - rh - 10),(rw+10),oh), fill="#0b0c0c")
    draw.rectangle((5,(oh - rh - 5),(rw+5),(oh-5)), fill="white")
    return (0,(oh - rh - 10),(rw+10),oh)


def add_url_to_img(url, screenshot):
    img = Image.open(screenshot)
    ow, oh = img.size
    draw = ImageDraw.Draw(img)

    txt_str = f"Screenshot from:\n{url}"

    chosen_font = ImageFont.truetype(os.path.join(system_fonts_foler, font_name), 19)
    size = draw.textsize(txt_str, chosen_font)
    container = add_text_container(draw, img.size, size)
    
    draw.text((container[0]+20,container[1]+20), txt_str, fill='#0b0c0c', font=chosen_font)
    #draw.text((15,15), txt_str, fill='#0b0c0c', font=chosen_font)
    img.save(screenshot)


async def screenshotr(url, output, fullPage=True):
    browser = None
    page = None
    if browser is None:
        browser = await pyppeteer.launch({
            'args': ['--no-sandbox']
        })
    if page is None:
        page = await browser.newPage()

    screenshot_path = f"screenshots/{output}.png"

    if not os.path.isdir("screenshots"):
        os.mkdir("screenshots")

    await page.goto(url)
    # default setting
    # sets the page size for non fullpage screenshots
    await page.setViewport({"width":1200,"height":1600})
    await page.emulateMedia('screen')

    opts = {
        'path': screenshot_path,
        'type': 'png',
        'fullPage': fullPage
    }
    # if not fullPage:
    #     opts['clip'] = {
    #         'x': 0,
    #         'y': 0,
    #         'width': 1000,
    #         'height': 1600
    #     }

    await page.screenshot(opts)

    await browser.close()
    add_url_to_img(url, screenshot_path)


if __name__ == "__main__":
    args_provided = sys.argv[1:]
    print(args_provided)
    if len(args_provided) >= 2:
        fullPage = True
        if len(args_provided) > 2:
            if args_provided[2] and args_provided[2] in ['F', 'f', 'False', 'false']:
                fullPage = False
        asyncio.get_event_loop().run_until_complete(screenshotr(url=args_provided[0], output=args_provided[1], fullPage=fullPage))
    else:
        print("No URL provided")