#! /bin/bash

set -e
set -o pipefail

if ! grep tabtab .gitignore; then
   echo >> .gitignore
   echo tabtab >> .gitignore
fi

image_ui=bel/tabtab:ui-$(date +%Y%m%d)
if ! docker images | grep ${image_ui##*:}; then
   if ! [ -d ./tabtab ]; then
      git clone https://github.com/ainize-team/tabtab.git
   fi
   pushd tabtab
   cp ../tabtab.requirements.txt ./requirements.txt
   docker build -t $image_ui .
   popd
fi

if ! [ -d ./models ]; then
   mkdir ./models
   python3 -c 'from tqdm import tqdm' &> /dev/null || pip3 install tqdm
   python3 ./download.py 124M
fi

image_server=${image_ui//:ui/:server}
if ! docker images | grep ${image_server##*:}; then
   if ! [ -d ./gpt2-large ]; then
      git clone https://github.com/Henriquepheak/gpt2-large.git
   fi
   pushd ./gpt2-large
   docker build -t $image_server .
   popd
fi

cleanup() {
   docker rm -f $(docker ps -aq)
}
trap cleanup EXIT

localhost=$(ifconfig  | grep -o '192.168[^ ]*' | head -n 1)
docker run -p 12313:80 --rm -d --name tabtab-server                                            $image_server
docker run -p 12314:80 --rm -d --name tabtab-ui     -e GPT2_SERVER_URL=http://$localhost:12313 $image_ui

docker logs -f $image_ui