Update Dockerfile

Dockerfile

@@ -4,7 +4,7 @@ FROM python:3.12-slim
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
     WDM_LOCAL=1 \
-    WDM_PATH=/tmp/.wdm \
+    WDM_PATH=/home/user/.wdm \
     GRADIO_SERVER_NAME="0.0.0.0" \
     GRADIO_SERVER_PORT=7860
 
@@ -26,10 +26,6 @@ RUN wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key
     && apt-get install -y google-chrome-stable \
     && rm -rf /var/lib/apt/lists/*
 
-# Create and set permissions for ChromeDriver directory (as root)
-RUN mkdir -p /tmp/.wdm && \
-    chmod -R 777 /tmp/.wdm
-
 # Create a user with UID 1000 (as root)
 RUN useradd -m -u 1000 user
 
@@ -42,6 +38,10 @@ COPY --chown=user . $HOME/app
 # Switch to non-root user
 USER user
 
+# Create ChromeDriver directory in user's home (as user)
+RUN mkdir -p $HOME/.wdm && \
+    chmod -R 700 $HOME/.wdm
+
 # Install dependencies (as user)
 RUN pip install --no-cache-dir -r requirements.txt