chore: publish local test to test env

app.py

@@ -8,7 +8,6 @@ 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",
@@ -67,24 +66,30 @@ custom_theme = PurpleTheme()
 
 title = "Imaginor"
 driver_type = 'chromedriver'
-description = "Imaginor is a small space for taking screenshots of websites."
+description = "<center>Imaginor is a small space for taking screenshots of websites.</center>"
 
 def run_imaginor(text: str):
-    driver = webdriver.Chrome(driver_type)
-    driver.get(text)
-    screenshot = driver.save_screenshot('./site_screenshot.png')
-    driver.quit()
+    regex = r"^(https?://)"
+    is_url = re.search(regex, text)
+
+    if is_url:
+        options = webdriver.ChromeOptions()
+        options.add_argument("--window-size=1920,1080")
+
+        driver = webdriver.Chrome(options=options)
+        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.']
+        return ['site_screenshot.png', 'Website imagined by Imaginor, operation success.']
     else:
-        return [None, 'Website imagined by Imaginor, operation failed.']
+        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)],
+    Gradio.Textbox(label="Text result", interactive=False)],
     title=title,
     description=description,
     theme=custom_theme,