import sys
import typing

import pandas as pd
from PyQt6.QtWidgets import QApplication

from banking_breakdown.ui.generated_wrapper import GeneratedWindowWrapper


def show_main_window(categories: typing.Sequence[str] = None,
                     df: pd.DataFrame = None):
    app = QApplication(sys.argv)
    window = GeneratedWindowWrapper()

    if categories is not None:
        window.add_categories(categories)

    if df is not None:
        window.set_statement_data(df)

    window.show()
    app.exec()


def main():
    from banking_breakdown.statement_parser import get_stripped_statement

    categories = ["Food", "Rent & Utilities", "Hobbies", "Education",
                  "Transportation", "Social Life", "Other"]

    df = get_stripped_statement("../../res/banking_statement_2023.csv")

    show_main_window(categories, df)


if __name__ == "__main__":
    main()