Update app/lens_images_core.py

app/lens_images_core.py

@@ -1,11 +1,12 @@
 
-import os, json, time, hashlib, httpx, base64, re, asyncio, threading, shutil, logging
+import os, json, time, hashlib, httpx, base64, re, asyncio, threading, shutil, logging, atexit, tempfile, uuid
 from typing import Dict, Any
 from urllib.parse import urlparse
 
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options as ChromeOptions
 from selenium.webdriver.chrome.service import Service as ChromeService
+from selenium.common.exceptions import SessionNotCreatedException
 
 LOGGER = logging.getLogger("lens_images_core")
 if not LOGGER.handlers:
@@ -34,7 +35,6 @@ _PROFILE_DIRS = []
 
 def _mk_profile_dir() -> str:
     base = os.getenv("CHROME_PROFILE_BASE", tempfile.gettempdir())
-    # ผูกกับ PID + thread + UUID เพื่อไม่ชนกัน
     p = os.path.join(base, f"chrome-profile-{os.getpid()}-{threading.get_ident()}-{uuid.uuid4().hex}")
     os.makedirs(p, exist_ok=True)
     _PROFILE_DIRS.append(p)
@@ -49,22 +49,6 @@ def _cleanup_profiles():
 
 atexit.register(_cleanup_profiles)
 
-def _find_chrome_binary() -> str | None:
-    env = os.getenv("CHROME_BINARY")
-    if env and shutil.which(env):
-        return env
-    for p in _COMMON_CHROME_PATHS:
-        if os.path.isfile(p) and os.access(p, os.X_OK):
-            return p
-    try:
-        import subprocess, shlex
-        out = subprocess.check_output(shlex.split("which google-chrome"), stderr=subprocess.DEVNULL).decode().strip()
-        if out:
-            return out
-    except Exception:
-        pass
-    return None
-
 def _build_chrome() -> webdriver.Chrome:
     opts = ChromeOptions()
 
@@ -86,7 +70,7 @@ def _build_chrome() -> webdriver.Chrome:
             return webdriver.Chrome(service=ChromeService(executable_path=drv_path), options=opts)
         return webdriver.Chrome(options=opts)
     except SessionNotCreatedException as e:
-        LOG.warning("SessionNotCreated: %s; retry with a fresh profile dir", e)
+        LOGGER.warning("SessionNotCreated: %s; retry with a fresh profile dir", e)
         profile_dir2 = _mk_profile_dir()
         opts.arguments = [a for a in opts.arguments if not a.startswith("--user-data-dir=")]
         opts.add_argument(f"--user-data-dir={profile_dir2}")