Testing with ChromeDriver

Dockerfile

@@ -9,6 +9,9 @@ RUN apt-get update && apt-get install -y \
     xvfb \
     dbus-x11 \
     gnupg \
+    && wget -q -O - https://dl-ssl.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 \
     && rm -rf /var/lib/apt/lists/*
 
@@ -25,7 +28,8 @@ ENV MPLCONFIGDIR=/tmp/matplotlib \
     FONTCONFIG_FILE=/etc/fonts/fonts.conf \
     FONTCONFIG_CACHE=/tmp/fontconfig \
     GRADIO_TEMP_DIR=/tmp/gradio \
-    CHROMEDRIVER_PATH=/usr/local/bin/chromedriver
+    CHROMEDRIVER_PATH=/usr/local/bin/chromedriver \
+    CHROME_BIN=/usr/bin/google-chrome-stable
 
 # Create directories with correct permissions
 RUN mkdir -p ${XDG_CACHE_HOME} && chmod 777 ${XDG_CACHE_HOME} \
@@ -50,12 +54,12 @@ RUN CHROME_DRIVER_VERSION=`curl -sS https://chromedriver.storage.googleapis.com/
 RUN chromedriver --version && google-chrome-stable --version
 
 # Ensure proper port exposure
-EXPOSE 2828
-EXPOSE 4444
+EXPOSE 7860
+EXPOSE 9222
 
 # Add health check
 HEALTHCHECK --interval=30s --timeout=30s \
-  CMD curl -f http://localhost:4444/status || exit 1
+  CMD curl -f http://localhost:7860 || exit 1
 
 # Copy application
 COPY . .