Testing with ChromeDriver

Dockerfile

@@ -5,10 +5,11 @@ FROM python:3.10-slim
 RUN apt-get update && apt-get install -y \
     wget \
     curl \
-    firefox-esr \
     fonts-dejavu \
     xvfb \
     dbus-x11 \
+    gnupg \
+    && apt-get install -y google-chrome-stable \
     && rm -rf /var/lib/apt/lists/*
 
 # Set up Xvfb and working directory
@@ -24,7 +25,7 @@ ENV MPLCONFIGDIR=/tmp/matplotlib \
     FONTCONFIG_FILE=/etc/fonts/fonts.conf \
     FONTCONFIG_CACHE=/tmp/fontconfig \
     GRADIO_TEMP_DIR=/tmp/gradio \
-    GECKODRIVER_PATH=/usr/local/bin/geckodriver
+    CHROMEDRIVER_PATH=/usr/local/bin/chromedriver
 
 # Create directories with correct permissions
 RUN mkdir -p ${XDG_CACHE_HOME} && chmod 777 ${XDG_CACHE_HOME} \
@@ -36,16 +37,17 @@ RUN mkdir -p ${XDG_CACHE_HOME} && chmod 777 ${XDG_CACHE_HOME} \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Install geckodriver properly
-RUN wget -q https://github.com/mozilla/geckodriver/releases/download/v0.32.2/geckodriver-v0.32.2-linux64.tar.gz \
-    && tar -xzf geckodriver-*.tar.gz \
-    && chmod 755 geckodriver \
-    && mv geckodriver /usr/local/bin/ \
-    && rm geckodriver-*.tar.gz \
-    && geckodriver --version
+# Install ChromeDriver
+# Check latest version from https://chromedriver.chromium.org/downloads
+RUN CHROME_DRIVER_VERSION=`curl -sS https://chromedriver.storage.googleapis.com/LATEST_RELEASE` \
+    && wget -q https://chromedriver.storage.googleapis.com/$CHROME_DRIVER_VERSION/chromedriver_linux64.zip \
+    && unzip chromedriver_linux64.zip \
+    && rm chromedriver_linux64.zip \
+    && mv chromedriver /usr/local/bin/ \
+    && chmod +x /usr/local/bin/chromedriver
 
 # Verify installations
-RUN geckodriver --version && firefox-esr --version
+RUN chromedriver --version && google-chrome-stable --version
 
 # Ensure proper port exposure
 EXPOSE 2828