Refactor Dockerfile to create missing directories and update ownership

Dockerfile

@@ -20,10 +20,15 @@ ENV HOME=/home/user \
 WORKDIR $HOME
 
 # Create directories for the app code to be copied into
-RUN mkdir -p $HOME/app
-RUN mkdir -p $HOME/src 
-RUN mkdir -p $HOME/data 
-RUN mkdir -p $HOME/config
+RUN mkdir $HOME/app
+RUN mkdir $HOME/src 
+RUN mkdir $HOME/data 
+RUN mkdir $HOME/config
+
+# Check if /data directory exists, if not create a local db directory. Hugging face spaces has it by default, but not local docker.
+RUN if [ ! -d "/data" ]; then \
+		mkdir $HOME/db; \
+	fi
 
 # Install Poetry
 RUN pip3 install poetry==1.7.1
@@ -46,7 +51,7 @@ RUN poetry install --no-root
 
 # Copy the rest of the repo into home
 RUN cp -R /clonedir/src /clonedir/data /clonedir/config /clonedir/app $HOME
-RUN chown -R user:user $HOME/src $HOME/data $HOME/config
+RUN chown -R user:user $HOME/src $HOME/data $HOME/config $HOME/db
 
 # Expose the port Streamlit runs on
 EXPOSE 8501
@@ -66,4 +71,4 @@ ENTRYPOINT ["streamlit", "run", "Home.py", "--server.port=8501", "--server.addre
 
 # To run remotely from hugging face spaces
 # docker run -it -p 7860:7860 --platform=linux/amd64 \
-# 	registry.hf.space/ai-aerospace-aerospace-chatbots:latest 
+# 	registry.hf.space/ai-aerospace-aerospace-chatbots:latest