Update Dockerfile

Dockerfile

@@ -1,5 +1,6 @@
 FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
 
+# Install necessary packages
 RUN apt update && \
     apt install -y --no-install-recommends \
         curl \
@@ -25,11 +26,14 @@ RUN apt update && \
         imagemagick \
     && git lfs install \
     && rm -rf /var/lib/apt/lists/*
-    
+
+# Fix ImageMagick policy
 RUN sed -i '/<policy domain="path" rights="none" pattern="@\*"/d' /etc/ImageMagick-6/policy.xml
 
+# Set working directory
 WORKDIR /home/
 
+# Environment variables
 ENV USERNAME=user \
     USER_UID=1000 \
     USER_GID=1000 \
@@ -41,10 +45,10 @@ ENV USERNAME=user \
     VISUAL=code \
     GIT_EDITOR="code --wait" \
     OPENVSCODE_SERVER_ROOT=/home/.vscode \
-    OPENVSCODE=/home/.vscode/bin/openvscode-server
+    OPENVSCODE=/home/.vscode/bin/openvscode-server \
+    PATH="/home/user/.local/bin:$PATH"
 
-# Downloading the latest VSC Server release and extracting the release archive
-# Rename `openvscode-server` cli tool to `code` for convenience
+# Download and install OpenVSCode Server
 RUN RELEASE_TAG=$(curl -sX GET "https://api.github.com/repos/gitpod-io/openvscode-server/releases/latest" | awk '/tag_name/{print $4;exit}' FS='[""]') && \
     arch=$(uname -m) && \
     if [ "${arch}" = "x86_64" ]; then \
@@ -60,8 +64,6 @@ RUN RELEASE_TAG=$(curl -sX GET "https://api.github.com/repos/gitpod-io/openvscod
     cp ${OPENVSCODE_SERVER_ROOT}/bin/remote-cli/openvscode-server ${OPENVSCODE_SERVER_ROOT}/bin/remote-cli/code && \
     rm -f ${RELEASE_TAG}-linux-${arch}.tar.gz
 
-
-    
 # Install Firefox within VS-code
 RUN install -d -m 0755 /etc/apt/keyrings && \
     wget -q https://packages.mozilla.org/apt/repo-signing-key.gpg -O- | tee /etc/apt/keyrings/packages.mozilla.org.asc > /dev/null && \
@@ -70,7 +72,7 @@ RUN install -d -m 0755 /etc/apt/keyrings && \
     echo 'Package: *\nPin: origin packages.mozilla.org\nPin-Priority: 1000\n' | tee /etc/apt/preferences.d/mozilla && \
     apt-get update && \
     apt-get install -y firefox
-    
+
 # Download and install GeckoDriver for x86_64 architecture
 RUN wget https://github.com/mozilla/geckodriver/releases/download/v0.34.0/geckodriver-v0.34.0-linux64.tar.gz && \
     tar -xvzf geckodriver-v0.34.0-linux64.tar.gz && \
@@ -78,20 +80,24 @@ RUN wget https://github.com/mozilla/geckodriver/releases/download/v0.34.0/geckod
     sudo chmod +x /usr/local/bin/geckodriver && \
     geckodriver --version
 
-RUN RUN curl -sSL https://pdm-project.org/install-pdm.py | python3 - 
+# Install PDM
+RUN curl -sSL https://pdm-project.org/install-pdm.py | python3 -
 
+# Set working directory for user
 WORKDIR /home/user/
 
-# Creating the user and usergroup
+# Create user and usergroup
 RUN groupadd --gid ${USER_GID} ${USERNAME} \
     && useradd --uid ${USER_UID} --gid ${USERNAME} -m -s /bin/bash ${USERNAME} \
     && echo ${USERNAME} ALL=\(root\) NOPASSWD:ALL > /etc/sudoers.d/${USERNAME} \
     && chmod 0440 /etc/sudoers.d/${USERNAME}
 
+# Set permissions
 RUN chmod g+rw /home && \
     chown -R ${USERNAME}:${USERNAME} ${OPENVSCODE_SERVER_ROOT} && \
     chown -R ${USERNAME}:${USERNAME} /home/${USERNAME}
 
+# Switch to user
 USER $USERNAME
 
 # Install oh-my-zsh & Init
@@ -106,4 +112,5 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt && \
     rm -rf requirements.txt
 
+# Entry point
 ENTRYPOINT ["/bin/sh", "-c", "exec $OPENVSCODE --host 0.0.0.0 --port 7860 --without-connection-token"]