| from __future__ import annotations | |
| from selenium import webdriver | |
| from typing import Iterable | |
| import gradio as Gradio | |
| from gradio.themes.base import Base | |
| from gradio.themes.utils import colors, fonts, sizes | |
| import re | |
| from PIL import Image | |
| from io import BytesIO | |
| from selenium.common.exceptions import WebDriverException | |
| 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. Handy for open-source projects that involves <br/>getting the site screenshots for status checks and view previews from a bird's eye view.</center>" | |
| def run_imaginor(text: str): | |
| regex = r"^(https?://)" | |
| is_url = re.search(regex, text) | |
| if is_url: | |
| options = webdriver.ChromeOptions() | |
| options.add_argument('--headless') | |
| options.add_argument('--no-sandbox') | |
| options.add_argument('--disable-dev-shm-usage') | |
| try: | |
| driver = webdriver.Chrome(options=options) | |
| driver.get(text) | |
| driver.implicitly_wait(30) | |
| driver.set_window_size(1920, 1080) | |
| screenshot = driver.get_screenshot_as_png() | |
| except WebDriverException as e: | |
| return [Image.new('RGB', (1, 1)), 'Website imagined by Imaginor, operation failed.'] | |
| finally: | |
| if driver: | |
| driver.quit() | |
| return [Image.open(BytesIO(screenshot)), '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="pil", label="Screenshot"), | |
| Gradio.Textbox(label="Text result", interactive=False)], | |
| title=title, | |
| description=description, | |
| theme=custom_theme, | |
| analytics_enabled=False, | |
| css=".generating {visibility: hidden}" | |
| ).launch(debug=True) |