from __future__ import annotations
from selenium import webdriver
from typing import Iterable

import re
import os
import gradio as Gradio
from gradio.themes.base import Base
from gradio.themes.utils import colors, fonts, sizes

theme = Gradio.themes.Monochrome(
    primary_hue="purple",
    secondary_hue="purple",
    neutral_hue="neutral",
    radius_size=Gradio.themes.sizes.radius_sm,
    font=[Gradio.themes.GoogleFont("Inter"), "ui-sans-serif", "system-ui", "sans-serif"],
)

class PurpleTheme(Base):
    def __init__(
        self,
        *,
        primary_hue: colors.Color | str = colors.purple,
        secondary_hue: colors.Color | str = colors.purple,
        neutral_hue: colors.Color | str = colors.neutral,
        spacing_size: sizes.Size | str = sizes.spacing_md,
        radius_size: sizes.Size | str = sizes.radius_md,
        font: fonts.Font
        | str
        | Iterable[fonts.Font | str] = (
            fonts.GoogleFont("Inter"),
            "ui-sans-serif",
            "sans-serif",
        ),
        font_mono: fonts.Font
        | str
        | Iterable[fonts.Font | str] = (
            fonts.GoogleFont("Space Grotesk"),
            "ui-monospace",
            "monospace",
        ),
    ):
        super().__init__(
            primary_hue=primary_hue,
            secondary_hue=secondary_hue,
            neutral_hue=neutral_hue,
            spacing_size=spacing_size,
            radius_size=radius_size,
            font=font,
            font_mono=font_mono,
        )
        super().set(
            button_primary_background_fill="linear-gradient(90deg, *primary_300, *secondary_400)",
            button_primary_background_fill_hover="linear-gradient(90deg, *primary_200, *secondary_300)",
            button_primary_text_color="white",
            button_primary_background_fill_dark="linear-gradient(90deg, *primary_600, *secondary_800)",
            block_shadow="*shadow_drop_lg",
            button_shadow="*shadow_drop_lg",
            input_background_fill="zinc",
            input_border_color="*secondary_300",
            input_shadow="*shadow_drop",
            input_shadow_focus="*shadow_drop_lg",
        )

custom_theme = PurpleTheme()

title = "Imaginor"
driver_type = 'chromedriver'
description = "<center>Imaginor is a small space for taking screenshots of websites.</center>"

def run_imaginor(text: str):
    regex = r"^(https?://)"
    is_url = re.search(regex, text)

    if is_url:
        driver = webdriver.Chrome()
        driver.get(text)
        screenshot = driver.save_screenshot('./site_screenshot.png')
        driver.quit()

        return ['./site_screenshot.png', 'Website imagined by Imaginor, operation success.']
    else:
        return [None, 'Please enter a valid URL of a website/host.']

Gradio.Interface(
    run_imaginor,
    Gradio.Textbox(placeholder="Enter your URL here", label="Website URL / Endpoint"),
    [Gradio.Image(type="filepath", label="Screenshot"),
    Gradio.Textbox(label="Text result", interactive=False)],
    title=title,
    description=description,
    theme=custom_theme,
    analytics_enabled=False,
    css=".generating {visibility: hidden}"
).launch()