Update Dockerfile

Dockerfile

@@ -1,59 +1,51 @@
-# ===== Base image =====
-# Use GPU build only if you enabled a GPU in Space settings.
-#FROM tensorflow/tensorflow:2.9.1-gpu
-# For CPU-only Spaces, use:
-FROM tensorflow/tensorflow:2.9.1
-
-ENV PYTHONDONTWRITEBYTECODE=1 \
-    PYTHONUNBUFFERED=1 \
-    PIP_NO_CACHE_DIR=1 \
-    PYTHONPATH=/app:${PYTHONPATH} \
-    MPLCONFIGDIR=/app/.matplotlib
-
-# ----- System deps -----
-ARG DEBIAN_FRONTEND=noninteractive
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    libopenexr-dev ffmpeg sudo build-essential git pkg-config && \
-    rm -rf /var/lib/apt/lists/*
-
-# ----- Python tooling -----
-RUN python -m pip install --upgrade pip "setuptools<70" wheel
-
-# ----- Pins compatible with TF 2.9.x -----
-RUN pip install \
-    "protobuf==3.19.6" \
-    "tensorflow-addons==0.17.1" \
-    "numpy==1.22.4" "scipy==1.8.1" "pandas==1.3.5"
-
-# ----- Your other deps -----
-RUN pip install \
-    tqdm h5py scikit-image==0.19.3 jupyter typing_extensions \
-    matplotlib opencv-python gdown==2.3.1 pydicom scikit-learn==1.1.3 \
-    tabulate pingouin streamlit==1.22.0 openpyxl ffmpeg-python gputil
-
-# If you actually need it, uncomment and pin to a version that exists on PyPI:
-# RUN pip install "tensorflow-mri<0.23"   # e.g., ==0.22.0
-
-# ----- App code -----
-WORKDIR /app
-COPY . /app
-
-# Writable config/cache dirs
-RUN mkdir -p /app/.matplotlib /app/.streamlit && chmod -R 777 /app/.matplotlib /app/.streamlit
-
-# Streamlit config (no heredoc; use printf to avoid quoting problems)
-RUN printf "%s\n" \
-  "[server]" \
-  "headless = true" \
-  "enableCORS = false" \
-  "enableXsrfProtection = false" \
-  'address = "0.0.0.0"' \
-  "[client]" \
-  'toolbarMode = "minimal"' \
-  "[browser]" \
-  "gatherUsageStats = false" \
-  > /app/.streamlit/config.toml
-
-# ----- Entrypoint -----
-EXPOSE 7860
-CMD ["bash","-lc","python -m streamlit run app.py --server.port ${PORT:-7860}"]
+# See here for image contents: https://github.com/microsoft/vscode-dev-containers/tree/v0.245.0/containers/python-3/.devcontainer/base.Dockerfile
+
+# [Choice] Python version (use -bullseye variants on local arm64/Apple Silicon): 3, 3.10, 3.9, 3.8, 3.7, 3.6, 3-bullseye, 3.10-bullseye, 3.9-bullseye, 3.8-bullseye, 3.7-bullseye, 3.6-bullseye, 3-buster, 3.10-buster, 3.9-buster, 3.8-buster, 3.7-buster, 3.6-buster
+FROM tensorflow/tensorflow:2.9.1-gpu
+RUN apt-get install libopenexr-dev -y
+RUN pip install tensorflow-mri
+RUN pip install tqdm
+RUN pip install h5py
+RUN pip install tensorflow-addons
+# RUN pip install sklearn
+RUN pip install scikit-image
+RUN pip install jupyter
+RUN pip install typing_extensions --upgrade
+#RUN pip install neptune
+#RUN pip install neptune-tensorflow-keras
+RUN pip install utils
+RUN pip install matplotlib
+RUN pip install --upgrade pip
+#RUN pip install -U torch-tb-profiler
+RUN pip install opencv-python
+#RUN pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html
+RUN pip install gdown==2.3.1
+RUN pip install pydicom
+RUN pip install scikit-learn
+RUN pip install tabulate
+RUN pip install pingouin
+RUN pip install streamlit
+RUN pip install openpyxl
+RUN pip install "protobuf<=3.20.3"
+
+
+# Create non-root user.
+ARG USERNAME=vscode
+ARG USER_UID=1003
+ARG USER_GID=$USER_UID
+
+RUN groupadd --gid $USER_GID $USERNAME && \
+    useradd --uid $USER_UID --gid $USER_GID -m $USERNAME && \
+    # Add user to sudoers.
+    apt-get update && \
+    apt-get install -y sudo && \
+    echo $USERNAME ALL=\(root\) NOPASSWD:ALL > /etc/sudoers.d/$USERNAME && \
+    chmod 0440 /etc/sudoers.d/$USERNAME && \
+    # Change default shell to bash.
+    usermod --shell /bin/bash $USERNAME
+
+RUN apt-get update && \
+    apt-get install -y ffmpeg
+
+RUN pip install ffmpeg-python
+RUN pip install gputil