updated docker

Dockerfile

@@ -1,37 +1,40 @@
 FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 
+WORKDIR /app
+
 # Install system dependencies for Python and build tools
 RUN apt-get update && apt-get install -y \
     python3 python3-pip build-essential git
 
 # Create a non-root user
 RUN useradd -m -u 1000 user
-ENV PATH="/home/user/.local/bin:$PATH"
-ENV POETRY_CACHE_DIR=/home/user/.cache/pypoetry
 
-# Set the working directory to a writable location
-WORKDIR /home/user/app
+#ENV POETRY_CACHE_DIR=/home/user/.cache/pypoetry
+
+ENV HOME /home/user
+ENV PATH $HOME/.local/bin:$PATH
 
-RUN nvcc --version && ls /usr/local/cuda
 
 # Install Poetry globally as root
 RUN pip install --no-cache-dir --upgrade pip \
     && pip install --no-cache-dir poetry
 
-# Switch to non-root user for application setup
-USER user
 
-# Copy Poetry configuration files
-COPY --chown=user ./pyproject.toml ./poetry.lock* ./
+USER user
+COPY ./pyproject.toml ./poetry.lock* ./
 
 # Install dependencies with Poetry
 RUN poetry install --only main --no-interaction --no-ansi
 
-# Copy the rest of the application code
-COPY --chown=user . /home/user/app
-
-# Expose the Streamlit default port
-EXPOSE 8501
+WORKDIR $HOME
+RUN mkdir app
+WORKDIR $HOME/app
+COPY . $HOME/app
 
 # Run the Streamlit app
-CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+EXPOSE 8501
+CMD streamlit run app.py \
+    --server.headless true \
+    --server.enableCORS false \
+    --server.enableXsrfProtection false \
+    --server.fileWatcherType none