import gradio as gr
from selenium import webdriver
from selenium.common.exceptions import WebDriverException
from selenium.webdriver.common.by import By
from PIL import Image
from io import BytesIO

resp=""
def take_screenshot(url):
    options = webdriver.ChromeOptions()
    options.add_argument('--headless')
    options.add_argument('--no-sandbox')
    options.add_argument('--disable-dev-shm-usage')
    wd = webdriver.Chrome(options=options)
    wd.get(url)
    wd.implicitly_wait(10)
    resp=driver.find_element(By.TAG_NAME, "a")
    return resp


iface = gr.Interface(
    fn=take_screenshot,
    inputs=gr.Textbox(label="Website URL", value="https://spiegel.de"),
    outputs=gr.Textbox(),
    title="Website Screenshot",
    description="Take a screenshot of a website."
)

iface.launch()