FROM ubuntu:20.04

RUN apt-get update && apt-get install -y curl git libssl-dev make build-essential libbz2-dev \
    libreadline-dev libsqlite3-dev wget curl llvm libncurses5-dev libncursesw5-dev \
    xz-utils tk-dev libffi-dev liblzma-dev python3-openssl git

RUN curl https://pyenv.run | bash

ENV HOME /root
ENV PYENV_ROOT $HOME/.pyenv
ENV PATH $PYENV_ROOT/shims:$PYENV_ROOT/bin:$PATH
ENV PYTHON_VERSION 3.11.3

RUN pyenv install $PYTHON_VERSION
RUN pyenv global $PYTHON_VERSION

RUN python --version

RUN apt-get install -y python3-pip unzip

RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn sacred pymongo mlflow
RUN apt-get install -y git

COPY train.py /app/train.py
COPY predictions.py /app/predictions.py
COPY data.csv /app/data.csv

WORKDIR /app

RUN export SACRED_IGNORE_GIT=TRUE

SHELL ["/bin/bash", "-c"]

RUN source ~/.bashrc && mlflow run . -P epochs=10

CMD ["python3", "predictions.py"]