vikarshana commited on
Commit
3bfc14d
·
verified ·
1 Parent(s): b012041

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -11
app.py CHANGED
@@ -40,17 +40,21 @@ async def capture_direct_download_link(url, opts=None):
40
  chrome_options.add_argument("--disable-features=IsolateOrigins,site-per-process")
41
  chrome_options.add_argument("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36")
42
 
43
- # ✅ FIX: Set custom writable cache directory for webdriver-manager
44
- import tempfile
45
- cache_dir = tempfile.mkdtemp(prefix="wdm_")
46
-
47
- service = Service(
48
- ChromeDriverManager(
49
- chrome_type=ChromeType.GOOGLE,
50
- cache_dir=cache_dir
51
- ).install()
52
- )
53
- driver = webdriver.Chrome(service=service, options=chrome_options)
 
 
 
 
54
 
55
  driver.set_window_size(1280, 900)
56
 
 
40
  chrome_options.add_argument("--disable-features=IsolateOrigins,site-per-process")
41
  chrome_options.add_argument("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36")
42
 
43
+ import tempfile
44
+
45
+ # Set cache directory via environment variable (before importing webdriver-manager)
46
+ cache_dir = tempfile.mkdtemp(prefix="wdm_")
47
+ os.environ['WDM_LOCAL'] = cache_dir # 👈 This is the correct way
48
+
49
+ # Now import and use normally
50
+ from webdriver_manager.chrome import ChromeDriverManager
51
+ from webdriver_manager.core.os_manager import ChromeType
52
+ from selenium.webdriver.chrome.service import Service
53
+
54
+ service = Service(
55
+ ChromeDriverManager(chrome_type=ChromeType.GOOGLE).install()
56
+ )
57
+ driver = webdriver.Chrome(service=service, options=chrome_options)
58
 
59
  driver.set_window_size(1280, 900)
60