AkashKumarave commited on
Commit
8f93b56
·
verified ·
1 Parent(s): 7a90b38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -8,6 +8,7 @@ from bs4 import BeautifulSoup
8
  from selenium import webdriver
9
  from selenium.webdriver.chrome.options import Options
10
  from selenium.webdriver.chrome.service import Service
 
11
  import time
12
  import re
13
  import base64
@@ -122,11 +123,13 @@ def convert_website():
122
 
123
  logger.info(f"Converting website: {url} with viewport width: {viewport_width}")
124
 
125
- # Set up Chrome driver
126
  options = get_chrome_options()
127
  options.add_argument(f"--window-size={viewport_width},{viewport_height}")
128
 
129
- driver = webdriver.Chrome(options=options)
 
 
130
  driver.set_window_size(viewport_width, viewport_height)
131
 
132
  try:
@@ -442,4 +445,4 @@ def parse_css_text(css_text):
442
 
443
  if __name__ == "__main__":
444
  port = int(os.environ.get("PORT", 7860))
445
- app.run(host="0.0.0.0", port=port)
 
8
  from selenium import webdriver
9
  from selenium.webdriver.chrome.options import Options
10
  from selenium.webdriver.chrome.service import Service
11
+ from webdriver_manager.chrome import ChromeDriverManager
12
  import time
13
  import re
14
  import base64
 
123
 
124
  logger.info(f"Converting website: {url} with viewport width: {viewport_width}")
125
 
126
+ # Set up Chrome driver with webdriver_manager
127
  options = get_chrome_options()
128
  options.add_argument(f"--window-size={viewport_width},{viewport_height}")
129
 
130
+ # Use ChromeDriverManager to handle driver installation
131
+ service = Service(ChromeDriverManager().install())
132
+ driver = webdriver.Chrome(service=service, options=options)
133
  driver.set_window_size(viewport_width, viewport_height)
134
 
135
  try:
 
445
 
446
  if __name__ == "__main__":
447
  port = int(os.environ.get("PORT", 7860))
448
+ app.run(host="0.0.0.0", port=port)