Firoj112 commited on
Commit
2165a5c
·
verified ·
1 Parent(s): dcb3020

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -6,6 +6,7 @@ import helium
6
  from selenium import webdriver
7
  from selenium.webdriver.common.by import By
8
  from selenium.webdriver.common.keys import Keys
 
9
  from io import BytesIO
10
  from PIL import Image
11
  from datetime import datetime
@@ -61,7 +62,7 @@ def close_popups() -> str:
61
  """
62
  webdriver.ActionChains(driver).send_keys(Keys.ESCAPE).perform()
63
 
64
- # Initialize Chrome driver with Selenium directly
65
  try:
66
  chrome_options = webdriver.ChromeOptions()
67
  chrome_options.add_argument("--force-device-scale-factor=1")
@@ -71,10 +72,8 @@ try:
71
  chrome_options.add_argument("--disable-dev-shm-usage")
72
  chrome_options.add_argument("--window-position=0,0")
73
  chrome_options.add_argument("--headless=new")
74
- driver = webdriver.Chrome(
75
- executable_path='/usr/local/bin/chromedriver',
76
- options=chrome_options
77
- )
78
  helium.set_driver(driver)
79
  logger.info("Chrome driver initialized successfully.")
80
  except Exception as e:
 
6
  from selenium import webdriver
7
  from selenium.webdriver.common.by import By
8
  from selenium.webdriver.common.keys import Keys
9
+ from selenium.webdriver.chrome.service import Service
10
  from io import BytesIO
11
  from PIL import Image
12
  from datetime import datetime
 
62
  """
63
  webdriver.ActionChains(driver).send_keys(Keys.ESCAPE).perform()
64
 
65
+ # Initialize Chrome driver with Selenium Service
66
  try:
67
  chrome_options = webdriver.ChromeOptions()
68
  chrome_options.add_argument("--force-device-scale-factor=1")
 
72
  chrome_options.add_argument("--disable-dev-shm-usage")
73
  chrome_options.add_argument("--window-position=0,0")
74
  chrome_options.add_argument("--headless=new")
75
+ service = Service('/usr/local/bin/chromedriver')
76
+ driver = webdriver.Chrome(service=service, options=chrome_options)
 
 
77
  helium.set_driver(driver)
78
  logger.info("Chrome driver initialized successfully.")
79
  except Exception as e: