Update Dockerfile

Dockerfile

@@ -1,25 +1,51 @@
+# 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"
+
 
-# --- your apt/pip installs here ---
 RUN apt-get update && \
-    apt-get install -y --no-install-recommends libopenexr-dev ffmpeg && \
-    rm -rf /var/lib/apt/lists/*
+    apt-get install -y ffmpeg
 
-RUN pip install --upgrade pip
-RUN pip install tensorflow-mri tqdm h5py tensorflow-addons scikit-image jupyter \
-    typing_extensions utils matplotlib opencv-python gdown==2.3.1 pydicom \
-    scikit-learn tabulate pingouin streamlit openpyxl "protobuf<=3.20.3" \
-    ffmpeg-python gputil
 
-# --- make sure Streamlit doesn’t block uploads ---
-ENV STREAMLIT_SERVER_ENABLE_XSRF_PROTECTION=false \
-    STREAMLIT_SERVER_ENABLE_CORS=false \
-    STREAMLIT_BROWSER_GATHER_USAGE_STATS=false \
-    PORT=7860
+RUN pip install ffmpeg-python
+RUN pip install gputil
+
 
+# App
 WORKDIR /app
 COPY . /app
 
-EXPOSE 7860
+# Streamlit settings (optional)
+ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false \
+    PYTHONUNBUFFERED=1
+
+EXPOSE 8501
 
-CMD ["bash","-lc","streamlit run app.py --server.address=0.0.0.0 --server.port=$PORT"]
+# Start the app
+CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]