Update Dockerfile to use Python virtual environment for package installation

Dockerfile

@@ -10,22 +10,23 @@ RUN curl https://sh.rustup.rs -sSf | sh -s -- -y
 
 # Set up environment variables
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
-    PATH=/home/user/.cargo/bin:$PATH
+    PATH=/home/user/.local/bin:/home/user/.cargo/bin:$PATH
 
 # Set the working directory in the container
-# WORKDIR /usr/src/app
 WORKDIR $HOME/app
 
-# Copy the current directory contents into the container at /usr/src/app
-# COPY . /usr/src/app
+# Copy the current directory contents into the container at /home/user/app
 COPY --chown=user . $HOME/app
 
 # Set user to root
-# USER root
+USER root
 
 # Install any needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+RUN python -m venv /venv && \
+    /venv/bin/pip install --no-cache-dir -r requirements.txt
+
+# Set user back to non-root
+USER user
 
 # Make port 80 available to the world outside this container
 EXPOSE 80
@@ -34,4 +35,5 @@ EXPOSE 80
 # ENV NAME World
 
 # Run train_llm.py when the container launches
-CMD ["python", "./train_llm.py"]
+CMD ["/bin/bash", "-c", "source /venv/bin/activate && python train_llm.py"]
+# CMD ["/venv/bin/python", "train_llm.py"]