FROM continuumio/miniconda:latest

WORKDIR /root
ENV MODEL_DIR=/root/maestro

# packages needed
RUN apt-get update && \
    apt-get install -y unzip curl wget vim libsndfile-dev build-essential libasound2-dev libjack-dev portaudio19-dev

# checkpoint
RUN wget https://storage.googleapis.com/magentadata/models/onsets_frames_transcription/maestro_checkpoint.zip && \
    unzip maestro_checkpoint.zip && \
    mv train maestro

# installation
RUN curl https://raw.githubusercontent.com/tensorflow/magenta/master/magenta/tools/magenta-install.sh > /tmp/magenta-install.sh && \
    sed 's/sudo //' -i /tmp/magenta-install.sh && \
    bash --login /tmp/magenta-install.sh

# install fluidsynth for MIDI to WAV conversion
RUN apt-get install fluidsynth fluid-soundfont-gm --no-install-recommends

RUN echo "conda activate magenta" >> /root/.bashrc