updated docker

Dockerfile

@@ -1,43 +1,43 @@
 FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
+# System-level environment variables
 ENV HOST 0.0.0.0
-
-#WORKDIR /app
+ENV HOME /home/user
+ENV PATH $HOME/.local/bin:$PATH
 
 # Install system dependencies for Python and build tools
 RUN apt-get update && apt-get install -y \
-    python3 python3-pip build-essential git
+    python3 python3-pip build-essential git \
+    && rm -rf /var/lib/apt/lists/*
 
 # Create a non-root user
 RUN useradd -m -u 1000 user
 
-#ENV POETRY_CACHE_DIR=/home/user/.cache/pypoetry
-
-ENV HOME /home/user
-ENV PATH $HOME/.local/bin:$PATH
-
-
 # Install Poetry globally as root
 RUN pip install --no-cache-dir --upgrade pip \
-    && poetry config virtualenvs.create false \
     && pip install --no-cache-dir poetry
 
-
+# Switch to non-root user
 USER user
-COPY ./pyproject.toml ./poetry.lock* ./
+
+# Ensure Poetry works for non-root user
+ENV POETRY_CACHE_DIR=/home/user/.cache/pypoetry
+
+# Set the working directory
+WORKDIR /home/user/app
+
+# Copy Poetry configuration files
+COPY --chown=user ./pyproject.toml ./poetry.lock* ./
 
 # Install dependencies with Poetry
-RUN poetry install --only main --no-interaction --no-ansi
+RUN poetry config virtualenvs.create false \
+    && poetry install --only main --no-interaction --no-ansi
 
-WORKDIR $HOME
-RUN mkdir app
-WORKDIR $HOME/app
-COPY --chown=user . $HOME/app
+# Copy application files
+COPY --chown=user . /home/user/app
 
-# Run the Streamlit app
+# Expose the Streamlit default port
 EXPOSE 8501
-CMD streamlit run app.py \
-    --server.headless true \
-    --server.enableCORS false \
-    --server.enableXsrfProtection false \
-    --server.fileWatcherType none
+
+# Run the Streamlit app
+CMD ["streamlit", "run", "app.py", "--server.headless=true", "--server.enableCORS=false", "--server.enableXsrfProtection=false", "--server.fileWatcherType=none"]