add app

app.py

@@ -0,0 +1,35 @@
+import gradio as gr
+from selenium import webdriver
+from selenium.common.exceptions import WebDriverException
+from PIL import Image
+from io import BytesIO
+
+def take_screenshot(url):
+    options = webdriver.ChromeOptions()
+    options.add_argument('--headless')
+    options.add_argument('--no-sandbox')
+    options.add_argument('--disable-dev-shm-usage')
+
+    try:
+        wd = webdriver.Chrome(options=options)
+        wd.set_window_size(1920, 1080)  # Adjust the window size here
+        wd.get(url)
+        wd.implicitly_wait(10)
+        screenshot = wd.get_screenshot_as_png()
+    except WebDriverException as e:
+        return Image.new('RGB', (1, 1))
+    finally:
+        if wd:
+            wd.quit()
+
+    return Image.open(BytesIO(screenshot))
+
+iface = gr.Interface(
+    fn=take_screenshot,
+    inputs=gr.inputs.Textbox(label="Website URL", default="https://kargaranamir.github.io"),
+    outputs=gr.Image(type="pil", height=600, width=800),  # Adjust the image size here
+    title="Website Screenshot",
+    description="Take a screenshot of a website.",
+)
+
+iface.launch()

packages.txt

@@ -0,0 +1 @@
+chromium-driver

requirements.txt

@@ -0,0 +1,3 @@
+selenium >=4.0.0, < 5.0.0
+gradio>=3.40.1
+Pillow>=8.3.1,<9.0