selenium SUCKSSSS

Dockerfile

@@ -1,5 +1,18 @@
 FROM python:3.10-slim
 
+# Install system dependencies for Chrome/Chromium
+RUN apt-get update && apt-get install -y \
+    wget \
+    gnupg \
+    unzip \
+    curl \
+    && wget -q -O - https://dl.google.com/linux/linux_signing_key.pub | apt-key add - \
+    && echo "deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main" >> /etc/apt/sources.list.d/google-chrome.list \
+    && apt-get update \
+    && apt-get install -y google-chrome-stable \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
 WORKDIR /app
 
 COPY requirements.txt requirements.txt
@@ -7,4 +20,8 @@ RUN pip install -r requirements.txt
 
 COPY . .
 
+# Create necessary directories for Selenium and webdriver-manager
+RUN mkdir -p /tmp/selenium /tmp/webdriver-manager
+
 CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]
+

app.py

@@ -16,6 +16,8 @@ import pytz
 from webdriver_manager.chrome import ChromeDriverManager
 
 os.environ["SE_CACHE_PATH"] = "/tmp/selenium"
+os.environ["WDM_LOCAL"] = "1"
+os.environ["WDM_CACHE_PATH"] = "/tmp/webdriver-manager"
 
 app = Flask(__name__)
 CORS(app)
@@ -267,7 +269,7 @@ def seek():
     options.add_argument("--remote-debugging-port=9222")
     
     # Use webdriver-manager to automatically handle ChromeDriver
-    service = Service(ChromeDriverManager().install())
+    service = Service(ChromeDriverManager(cache_path="/tmp/webdriver-manager").install())
     driver = webdriver.Chrome(service=service, options=options)
     driver.get(url)