Update app/lens_images_core.py

app/lens_images_core.py

@@ -17,9 +17,7 @@ if not LOGGER.handlers:
         format="%(asctime)s | %(levelname)s | %(name)s | %(message)s",
     )
 
-COOKIE_JSON_URL = os.getenv("COOKIE_JSON_URL")
-if not COOKIE_JSON_URL:
-    raise RuntimeError("Missing COOKIE_JSON_URL secret. Set it in Space Settings > Secrets.")
+COOKIE_JSON_URL = os.getenv("COOKIE_JSON_URL", "") 
 UA = "Mozilla/5.0 (Lens OCR Images)"
 
 _COMMON_CHROME_PATHS = [
@@ -52,11 +50,12 @@ def _find_chrome_binary() -> str | None:
     return None
 
 def _build_chrome() -> webdriver.Chrome:
-    bin_loc = _find_chrome_binary() or "/usr/bin/chromium"
-    drv_path = os.getenv("CHROMEDRIVER", "/usr/bin/chromedriver")
     opts = ChromeOptions()
-    opts.binary_location = bin_loc
-    
+
+    bin_loc = _find_chrome_binary()
+    if bin_loc:
+        opts.binary_location = bin_loc
+
     extra = os.getenv(
         "CHROME_EXTRA_ARGS",
         "--disable-gpu --no-sandbox --disable-dev-shm-usage --window-size=1920,1080 --headless=new",
@@ -64,8 +63,20 @@ def _build_chrome() -> webdriver.Chrome:
     for a in extra:
         if a:
             opts.add_argument(a)
-    service = ChromeService(executable_path=drv_path)
-    return webdriver.Chrome(service=service, options=opts)
+
+    drv_path = os.getenv("CHROMEDRIVER")
+    if drv_path and os.path.exists(drv_path):
+        return webdriver.Chrome(service=ChromeService(executable_path=drv_path), options=opts)
+
+    try:
+        installed_path = chromedriver_autoinstaller.install()
+        if installed_path and os.path.exists(installed_path):
+            return webdriver.Chrome(service=ChromeService(executable_path=installed_path), options=opts)
+    except Exception as e:
+        LOGGER.warning("chromedriver_autoinstaller failed: %s; will try Selenium Manager", e)
+
+    return webdriver.Chrome(options=opts)
+
 
 _cached_cookie_obj: Dict[str, Any] | None = None
 _cached_cookie_fetched_at: float = 0.0