main.py

import pandas as pd
import argparse
import os
import matplotlib.pyplot as plt
from data_loader import load_data
from data_analyzer import validate_data
from data_analyzer import compute_metrics
from data_plot import plot
from data_report import save_outputs
from arg_parser import parse_arguments


def main():
    args = parse_arguments()

    df, individual_dfs = load_data(args.input)

    validate_data(df)

    df, avg_delP, max_delP, category_averages = compute_metrics(df)

    save_outputs(df, category_averages, avg_delP, max_delP, args.output)

    # Generate plots per dataset
    for case_name, case_df in individual_dfs.items():
      plot(case_df, case_name, args.output)

if __name__ == "__main__":
    main()