from init_env import init_env
from selenium_parser import load_driver
import streamlit as st


@st.cache_data
def init():
    # print("init")
    init_env()


@st.cache_resource
def get_driver():
    return load_driver()


def run():
    driver = get_driver()
    driver.get("https://www.booking.com/hotel/th/queen-boutique.ru.html#tab-reviews")
    st.write("Page loaded:")
    st.image(driver.get_screenshot_as_png(), caption="screenshot")


init()

st.button("Run", on_click=run)