import sys
import os

def explain():
    print('This converter can be used to transform keras models in the .h5 format into a tflite flatbuffer model .tflite\n')
    print('\tIt must be called like in this exampple: python keras2tflite.py /path/to/input_model.h5 ./output_model.tflite\n')
    print('\tArguments expected: 2\n\tArguments received: ' + str(len(sys.argv))-1)
    versions()

def versions(k=''):
    print(k + 'Python version >= 3.x')
    print(k + 'Keras model version >= 2.2.4')
    print(k + 'TensorFlow Lite version >= 1.11')
    print(k + 'TensorFlow version >= 1.14.x')

if len(sys.argv) != 3:
    explain()

else:
    # Converting a tf.Keras model to a TensorFlow Lite model.
    try:
        os.system('tflite_convert \
                --keras_model_file='+sys.argv[1]+' \
                --output_file='+sys.argv[2])
    except Exception as e:
        print('\nError in the conversion to TFLite:\n' + str(e) + '\n')
    except IOError as e:
        print('\nError when trying to save the model:\n' + str(e) +'\n')
    else:
        print('\nModel saved!\n')