from pathlib import Path

import fire
import matplotlib.pyplot as plt

from wsgi import create_fig


def main(folder: str = "plots"):
    fig, _df, _df_state, timestamp = create_fig()
    timestamp = timestamp.replace(" ", "_")
    timestamp = timestamp.replace(":", "-")
    fig.savefig(Path(folder) / f"digital_plot_{timestamp}.png", dpi=300)


if __name__ == "__main__":
    fire.Fire(main)