refacor: clean up the Dockerfile

Dockerfile

@@ -5,45 +5,44 @@ ARG APP_HOME=/home/user
 ARG APP_CODE=/app
 ARG APP_DATA=/data/app
 ARG HF_HOME=/data/huggingface
-ARG HOST_STATIC_FILES=static
 
-WORKDIR $APP_CODE
+ENV APP_CODE $APP_CODE
+ENV APP_DATA $APP_DATA
+ENV HF_HOME $HF_HOME
 
 COPY ./requirements.txt $APP_CODE/requirements.txt
 COPY ./packages.txt $APP_CODE/packages.txt
 
-ENV APP_CODE $APP_CODE
-RUN apt-get update && xargs -r -a $APP_CODE/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
+
+RUN    apt-get update \
+    && xargs -r -a $APP_CODE/packages.txt apt-get install -y \
+    && rm -rf /var/lib/apt/lists/*
 RUN pip3 install --no-cache-dir --upgrade pip
 RUN pip3 install --no-cache-dir -r $APP_CODE/requirements.txt
 
-# Creates an unpriviliedged user
+# Creates an unpriviledged user
 RUN useradd -m -u 1000 user
 
-# Copy our code to $APP_CODE taking car of the permissions
-
+# Copy our code to $APP_CODE and config to the working directory
 COPY --chown=user:user --chmod=u=rX src/ $APP_CODE/
 COPY --chown=user:user --chmod=u=rX .streamlit/ $APP_HOME/.streamlit/
 
 # Persistent disk space
 # Assumes a mount is passed to the docker command, like:
 # $ docker ... --mount type=volume,src=ai-playground-vol,dst=/data ...
-# 
 # see https://huggingface.co/docs/hub/spaces-storage
 VOLUME /data
-RUN mkdir -p /data && chown root:user /data && chmod ug+rwX /data
+RUN mkdir -p /data     && chown user:user /data
+RUN mkdir -p $APP_DATA && chown user:user $APP_DATA
+RUN mkdir -p $HF_HOME  && chown user:user $HF_HOME
 
 # User
 USER user
-ENV APP_DATA $APP_DATA
-RUN mkdir -p $APP_DATA
 ENV HOME $APP_HOME
 ENV PATH $HOME/.local/bin:$PATH
-ENV HF_HOME $HF_HOME
-RUN mkdir -p $HF_HOME
+WORKDIR $APP_HOME
 
 # Expose streamlit application
-WORKDIR $APP_HOME
 EXPOSE 8501
 CMD streamlit run $APP_CODE/app.py  \
     --logger.level=info                 \

scripts/build-run-local-docker.sh

@@ -17,7 +17,7 @@ docker build --label tests -t $CONTAINER_NAME .
 
 # During development ONLY use a `bind mount` to enable
 # editing the code without having to rebuild the container.
-docker run -it                  \
+docker run --rm -it                  \
     -p 8501:8501                \
     --mount type=volume,src=$VOLUME_NAME,dst=/data            \
     --mount type=bind,source=${ROOT_DIRECTORY}/src/,target=/app/,readonly       \