Update dockerfile

dockerfile

@@ -1,6 +1,5 @@
 FROM python:3.10-slim
 
-# Set environment variables
 ENV PYTHONDONTWRITEBYTECODE 1
 ENV PYTHONUNBUFFERED 1
 
@@ -14,33 +13,30 @@ RUN apt-get update && apt-get install -y \
     libx11-xcb1 libxcomposite1 libxdamage1 libxrandr2 xdg-utils \
     --no-install-recommends && rm -rf /var/lib/apt/lists/*
 
-# Install Google Chrome (stable)
+# Install Google Chrome stable
 RUN wget -q https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb && \
     dpkg -i google-chrome-stable_current_amd64.deb || apt-get -fy install && \
     rm google-chrome-stable_current_amd64.deb
 
-RUN CHROME_VERSION=$(google-chrome --version | grep -oP '\d+\.\d+\.\d+\.\d+') && \
-    echo "Using Chrome version: $CHROME_VERSION" && \
-    wget -q -O /tmp/chromedriver.zip "https://edgedl.me.gvt1.com/edgedl/chrome/chrome-for-testing/$CHROME_VERSION/linux64/chromedriver-linux64.zip" && \
-    unzip /tmp/chromedriver.zip -d /tmp/ && \
-    mv /tmp/chromedriver-linux64/chromedriver /usr/local/bin/chromedriver && \
+# Get Chrome major version
+RUN CHROME_VERSION=$(google-chrome --version | grep -oP '\d+' | head -1) && \
+    echo "Detected Chrome major version: $CHROME_VERSION" && \
+    CHROMEDRIVER_VERSION=$(curl -s "https://chromedriver.storage.googleapis.com/LATEST_RELEASE_$CHROME_VERSION") && \
+    echo "Matching ChromeDriver version: $CHROMEDRIVER_VERSION" && \
+    wget -q -O /tmp/chromedriver.zip "https://chromedriver.storage.googleapis.com/$CHROMEDRIVER_VERSION/chromedriver_linux64.zip" && \
+    unzip /tmp/chromedriver.zip -d /usr/local/bin/ && \
     chmod +x /usr/local/bin/chromedriver && \
-    rm -rf /tmp/chromedriver*
-
+    rm /tmp/chromedriver.zip
 
 # Install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy app
+# Copy app code
 COPY . /app
 
-# Expose FastAPI port
 EXPOSE 7860
 
-# Set correct path
-ENV PATH="/usr/bin:/usr/local/bin:${PATH}"
-
+ENV PATH="/usr/local/bin:$PATH"
 
-# Run app
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]