data_handler.py

import pandas as pd
import os
import streamlit as st

COLUMNS = ["Date", "Type", "Description", "Amount", "Category", "From", "To"]

#get user file
def get_file_name():
    username = st.session_state.get("username", "default")

    os.makedirs("data", exist_ok=True)
    return f"data/{username}.csv"

#create empty
def create_empty_data():
    return pd.DataFrame(columns=COLUMNS)


#Load Data
def load_data():
    FILE_NAME = get_file_name()

    if not os.path.exists(FILE_NAME):
        df = create_empty_data()
        df.to_csv(FILE_NAME, index=False)
        return df

    try:
        df = pd.read_csv(FILE_NAME)
        df.columns = df.columns.str.strip()

        # ensure columns exist
        for col in COLUMNS:
            if col not in df.columns:
                df[col] = None

        return df[COLUMNS]

    except Exception as e:
        print("Error loading data:", e)
        return create_empty_data()


#Save data
def save_data(df):
    FILE_NAME = get_file_name()

    try:
        df.to_csv(FILE_NAME, index=False)
    except Exception as e:
        print("Error saving data:", e)