import pandas as pd
from banking_breakdown import document_builder
import subprocess
import os
import shutil

import pandas as pd

from banking_breakdown import types
import numpy as np


def main():
    categories = ["A", "B", "C", "D", "E", "F", "G"]
    values = np.array([10, 12, 53, 12, 90, 23, 32])
    values = values / values.sum() * 100

    total_value = np.random.normal(size=10) + 4
    net_income = np.diff(total_value)

    category_overview_df = pd.DataFrame(
        {"category": categories, "value": values.astype('int32')})
    t = np.linspace(0, total_value.size, total_value.size)
    total_value_df = pd.DataFrame({"t": t, "value": total_value})
    t = np.linspace(0, net_income.size, net_income.size)
    net_income_df = pd.DataFrame({"t": t, "value": net_income})

    report_data = types.ReportData(category_overview=category_overview_df,
                                   net_income=net_income_df,
                                   total_value=total_value_df)

    document_builder.build_document(report_data)


if __name__ == "__main__":
    main()