Create app.py

app.py

@@ -0,0 +1,93 @@
+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
+
+
+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 = "Imaginor is a small space for taking screenshots of websites."
+
+def run_imaginor(text: str):
+    driver = webdriver.Chrome(driver_type)
+    driver.get(text)
+    screenshot = driver.save_screenshot('./site_screenshot.png')
+    driver.quit()
+
+    if not os.path.isfile('./site_screenshot.png'):
+		return ['./site_screenshot.png', 'Website imagined by Imaginor, operation success.']
+	else:
+		return [None, 'Website imagined by Imaginor, operation failed.']
+
+Gradio.Interface(
+    run_imaginor,
+    Gradio.Textbox(placeholder="Enter your URL here", label="Website URL / Endpoint"),
+    [Gradio.Image(type="filepath", "Screenshot"),
+     Gradio.Textbox(label="Text result", interactive=False)],
+    title=title,
+    description=description,
+    theme=custom_theme,
+    analytics_enabled=False,
+    css=".generating {visibility: hidden}"
+).launch(enable_queue=True)