Update Dockerfile

Dockerfile

@@ -16,9 +16,16 @@ RUN apt-get update && apt-get install -y \
     libgbm-dev \
     libgtk-3-0 \
     chromium \
-    chromium-driver \
     && rm -rf /var/lib/apt/lists/*
 
+# Download and install ChromeDriver
+RUN CHROME_VERSION=$(chromium --version | grep -oP '[0-9]+(\.[0-9]+)+') && \
+    DRIVER_VERSION=$(curl -sS "https://chromedriver.storage.googleapis.com/LATEST_RELEASE_$CHROME_VERSION") && \
+    wget -O /tmp/chromedriver.zip "https://chromedriver.storage.googleapis.com/$DRIVER_VERSION/chromedriver_linux64.zip" && \
+    unzip /tmp/chromedriver.zip -d /usr/local/bin/ && \
+    rm /tmp/chromedriver.zip && \
+    chmod +x /usr/local/bin/chromedriver
+
 WORKDIR /code
 RUN mkdir -p /code/uploads && chmod 755 /code/uploads
 
@@ -34,7 +41,7 @@ USER myuser
 
 # Set environment variables for Chrome
 ENV CHROME_BIN=/usr/bin/chromium
-ENV CHROME_DRIVER=/usr/bin/chromium-driver
+ENV CHROME_DRIVER=/usr/local/bin/chromedriver
 
 # Default command to run the application
 CMD ["python", "app.py"]