Update app.py

app.py

@@ -1,8 +1,47 @@
-from twitter_scraper_selenium import scrape_profile
-
-twitter_username = "TwitterAPI"
-filename = "twitter_api_data"
-browser = "chrome"
-headless = True
-#print(scrape_profile(twitter_username=twitter_username, filename=filename, browser=browser, headless=headless))
-print(scrape_profile(twitter_username="microsoft",output_format="json",browser="chrome",tweets_count=10))
+#from __future__ import annotations
+from selenium import webdriver
+#from typing import Iterable
+
+import gradio as gr
+
+import re
+from PIL import Image
+from io import BytesIO
+from selenium.common.exceptions import WebDriverException
+
+driver_type = 'chromedriver'
+
+def run_script(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)), 'operation failed.']
+        finally:
+            if driver:
+                driver.quit()
+
+        return [Image.open(BytesIO(screenshot)), 'operation success.']
+    else:
+        return [None, 'Please enter a valid URL of a website/host.']
+
+with gr.Blocks() as app:
+    inp = gr.Textbox()
+    btn= gr.Button()
+    with gr.Row():
+        outim = gr.Image()
+        outp = gr.Textbox()
+    btn.click(run_script,inp,[outim,outp])
+app.launch()