# 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"

RUN apt-get update && \
    apt-get install -y ffmpeg

RUN pip install ffmpeg-python
RUN pip install gputil


# App
WORKDIR /app
COPY . /app

# Streamlit settings (optional)
ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false \
    PYTHONUNBUFFERED=1

EXPOSE 8501

# Start the app
CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]