Update app.py

app.py

@@ -1,10 +1,12 @@
 import asyncio
 import time
 import logging
+import os
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.chrome.service import Service
 from webdriver_manager.chrome import ChromeDriverManager
+from webdriver_manager.core.utils import ChromeType
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
@@ -38,8 +40,16 @@ async def capture_direct_download_link(url, opts=None):
         chrome_options.add_argument("--disable-features=IsolateOrigins,site-per-process")
         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")
 
-        # ✅ FIX: Use webdriver-manager to auto-install correct ChromeDriver
-        service = Service(ChromeDriverManager().install())
+        # ✅ FIX: Set custom writable cache directory for webdriver-manager
+        cache_dir = "/app/.wdm"
+        os.makedirs(cache_dir, exist_ok=True)
+
+        service = Service(
+            ChromeDriverManager(
+                chrome_type=ChromeType.GOOGLE,
+                cache_dir=cache_dir
+            ).install()
+        )
         driver = webdriver.Chrome(service=service, options=chrome_options)
 
         driver.set_window_size(1280, 900)
@@ -188,5 +198,5 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=True
+        share=False  # ← MUST be False on Hugging Face Spaces
     )