kargaranamir commited on
Commit
c3b7bd1
1 Parent(s): c774858
Files changed (3) hide show
  1. app.py +35 -0
  2. packages.txt +1 -0
  3. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from selenium import webdriver
3
+ from selenium.common.exceptions import WebDriverException
4
+ from PIL import Image
5
+ from io import BytesIO
6
+
7
+ def take_screenshot(url):
8
+ options = webdriver.ChromeOptions()
9
+ options.add_argument('--headless')
10
+ options.add_argument('--no-sandbox')
11
+ options.add_argument('--disable-dev-shm-usage')
12
+
13
+ try:
14
+ wd = webdriver.Chrome(options=options)
15
+ wd.set_window_size(1920, 1080) # Adjust the window size here
16
+ wd.get(url)
17
+ wd.implicitly_wait(10)
18
+ screenshot = wd.get_screenshot_as_png()
19
+ except WebDriverException as e:
20
+ return Image.new('RGB', (1, 1))
21
+ finally:
22
+ if wd:
23
+ wd.quit()
24
+
25
+ return Image.open(BytesIO(screenshot))
26
+
27
+ iface = gr.Interface(
28
+ fn=take_screenshot,
29
+ inputs=gr.inputs.Textbox(label="Website URL", default="https://kargaranamir.github.io"),
30
+ outputs=gr.Image(type="pil", height=600, width=800), # Adjust the image size here
31
+ title="Website Screenshot",
32
+ description="Take a screenshot of a website.",
33
+ )
34
+
35
+ iface.launch()
packages.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ chromium-driver
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ selenium >=4.0.0, < 5.0.0
2
+ gradio>=3.40.1
3
+ Pillow>=8.3.1,<9.0