Update Docker configuration to use Chrome instead of Chromium

Dockerfile

@@ -5,14 +5,27 @@ FROM python:3.9
 
 WORKDIR /app
 
-# Install system dependencies
+# Install system dependencies and Chrome
 RUN apt-get update && apt-get install -y \
-    chromium \
-    chromium-driver \
+    wget \
+    gnupg2 \
     libgl1-mesa-glx \
     libglib2.0-0 \
+    && 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.list \
+    && apt-get update \
+    && apt-get install -y google-chrome-stable \
     && rm -rf /var/lib/apt/lists/*
 
+# Install ChromeDriver
+RUN CHROME_VERSION=$(google-chrome --version | awk '{ print $3 }' | cut -d'.' -f1) \
+    && wget -q "https://chromedriver.storage.googleapis.com/LATEST_RELEASE_${CHROME_VERSION}" -O - | xargs -I {} \
+    wget -q "https://chromedriver.storage.googleapis.com/{}/chromedriver_linux64.zip" \
+    && unzip chromedriver_linux64.zip \
+    && mv chromedriver /usr/local/bin/ \
+    && chmod +x /usr/local/bin/chromedriver \
+    && rm chromedriver_linux64.zip
+
 # Create a non-root user
 RUN useradd -m -u 1000 user
 USER user
@@ -28,5 +41,8 @@ COPY --chown=user . .
 ENV PATH="/home/user/.local/bin:${PATH}"
 ENV PYTHONPATH="/home/user/.local/lib/python3.9/site-packages:${PYTHONPATH}"
 
+# Update the chrome binary location in the code
+ENV CHROME_BINARY_LOCATION="/usr/bin/google-chrome"
+
 # Run the application
 CMD ["python", "main.py"]

main.py

@@ -251,7 +251,7 @@ def setup_webdriver():
     chrome_options.add_argument('--headless')
     chrome_options.add_argument('--no-sandbox')
     chrome_options.add_argument('--disable-dev-shm-usage')
-    chrome_options.binary_location = '/usr/bin/chromium'
+    chrome_options.binary_location = os.getenv('CHROME_BINARY_LOCATION', '/usr/bin/google-chrome')
     return webdriver.Chrome(options=chrome_options)
 
 def create_polygon_mask(image_size, points):