# -*- coding: utf-8 -*-
"""Script to plot results of ANN model testing.

@author: Laura C. Kühle
"""
import sys
import seaborn as sns

from tcd.Plotting import plot_evaluation_results

sns.set()


def main() -> None:
    """Plot results of ANN model tests."""
    with open(str(snakemake.log[0]), 'w', encoding='utf-8') as logfile:
        sys.stdout = logfile
        sys.stderr = logfile

        # Plot evaluation results
        plot_evaluation_results(evaluation_file=snakemake.input['json_file'],
                                directory=snakemake.params['directory'],
                                colors=snakemake.params['colors'])


if __name__ == '__main__':
    if "snakemake" in locals():
        main()
    else:
        print('Not Defined.')