#!/bin/sh -e

echo $1
echo $2
if [ "$1" = "" ]; then
			echo "Argument not provided. Provide S3 bucket name ..."
			exit 1
		else
			export bucket=$1
fi

if [ "$2" = "" ]; then
                        echo "Argument not provided. Provide model artifact name ..."
                        exit 1
                else
                        export model=$2
fi

if [ "$3" = "" ]; then
                        echo "Argument not provided. Provide model archive output name ..."
                        exit 1
                else
                        export model_archive=$3
fi

echo "Arguments are good"

aws s3 cp s3://${bucket}/${model} .

echo "copied model artifact from s3"

torch-model-archiver -f --model-name ${model_archive} --version 1.0 --model-file ./model.py --serialized-file ./${model} --handler  ./model_handler.py

mkdir -p model_layout/config 
mkdir -p model_layout/model-store
cp config.properties model_layout/config/.
cp cifar.mar model_layout/model-store/

aws s3 cp ./model_layout "s3://${bucket}" --recursive

echo "copied back model layout to s3"