Spaces:
Build error
Build error
File size: 3,756 Bytes
bf23d89 db00b6c bf23d89 d496e67 bf23d89 9416100 bf23d89 9416100 bf23d89 40b2567 bf23d89 40b2567 d496e67 bd062d7 d496e67 bf23d89 d496e67 9416100 40b2567 bf23d89 d496e67 40b2567 bf23d89 0123135 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 |
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.</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(10)
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() |