screenshot.py

from selenium.common.exceptions import WebDriverException
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
#from selenium_webdriver import WebElement
import uuid

driver_type = 'chromedriver'
driver=False

def create_ss(graph):
    uid=uuid.uuid4()
    code_out=""
    for ea in graph.split("\n"):
        code=ea.strip().strip("\n")
        code_out+=code
    url=f"https://omnibus-text-to-chart.hf.space/file=merm.html?mermaid={code_out}"
    is_url=True
    wait_time=10000
    if is_url:
        options = webdriver.ChromeOptions()
        options.add_argument('--headless')
        options.add_argument('--no-sandbox')
        options.add_argument('--disable-dev-shm-usage')
        mes='<center>operation success'
        try:
            driver = webdriver.Chrome(options=options)
            driver.get(url)
            html=driver.page_source
            print(html)            
            driver.implicitly_wait(int(wait_time*30))
            driver.set_window_size(int(width), int(height))
            obj = driver.find_element(By.ID, "chart_box")            
            screenshot = obj.screenshot(f'{uid}-tmp.png')
    return(f'{uid}-tmp.png')