Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
|
|
|
|
|
|
| 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)
|