darsoarafa commited on
Commit
c5b019d
·
verified ·
1 Parent(s): 98fe73e

Update extract.py

Browse files
Files changed (1) hide show
  1. extract.py +2 -1
extract.py CHANGED
@@ -2,7 +2,7 @@ from selenium import webdriver
2
  from selenium.common.exceptions import WebDriverException
3
  from PIL import Image
4
  from io import BytesIO
5
-
6
  def take_webdata(url):
7
  options = webdriver.ChromeOptions()
8
  options.add_argument('--headless')
@@ -14,6 +14,7 @@ def take_webdata(url):
14
  wd.set_window_size(1080, 720) # Adjust the window size here
15
  wd.get(url)
16
  wd.implicitly_wait(5)
 
17
  # Get the page title
18
  page_title = wd.title
19
  screenshot = wd.get_screenshot_as_png()
 
2
  from selenium.common.exceptions import WebDriverException
3
  from PIL import Image
4
  from io import BytesIO
5
+ import time
6
  def take_webdata(url):
7
  options = webdriver.ChromeOptions()
8
  options.add_argument('--headless')
 
14
  wd.set_window_size(1080, 720) # Adjust the window size here
15
  wd.get(url)
16
  wd.implicitly_wait(5)
17
+ time.sleep(5)
18
  # Get the page title
19
  page_title = wd.title
20
  screenshot = wd.get_screenshot_as_png()