budgettracker.py

import tkinter as tk
from tkinter import messagebox, ttk
import json
from collections import defaultdict

class User:
    def __init__(self, name, age, salary):
        self.name = name
        self.age = age
        self.salary = salary 

    def __str__(self):
        return f"Name: {self.name}, Age: {self.age}, Salary: {self.salary}"

class Expenses:
    def __init__(self):
        self.expenses = []  # Stores expense records as (price, category) 

    def log_expense(self, price, category):
        self.expenses.append((price, category))

    def get_total_expenses(self):
        return sum(price for price, _ in self.expenses)

class BudgetTracker:
    def __init__(self, window):
        self.window = window
        self.window.title("Your Budget Bestie")
        self.window.geometry("400x550")
        self.window.configure(bg="pink")

        self.user = None
        self.expenses = Expenses()

        self.create_user_frame()
        self.create_expenses_frame()
        self.create_summary_frame()
        self.load_data()  # Load saved data at startup

   
    def create_user_frame(self):
        user_frame = tk.Frame(self.window, bg="pink")
        user_frame.pack(pady=10)

        # Name
        tk.Label(user_frame, text="Name:", font=("Arial", 10, "bold"), bg="black", fg="blue").grid(row=0, column=0, sticky="w", padx=5, pady=(5,0))
        self.name_entry = tk.Entry(user_frame, width=25, bg="white", fg="black")
        self.name_entry.grid(row=1, column=0, padx=5, pady=(0,5))

        # Age
        tk.Label(user_frame, text="Age:", font=("Arial", 10, "bold"), bg="white", fg="blue").grid(row=2, column=0, sticky="w", padx=5, pady=(5,0))
        self.age_entry = tk.Entry(user_frame, width=25, bg="white", fg="black")
        self.age_entry.grid(row=3, column=0, padx=5, pady=(0,5))

        # Salary
        tk.Label(user_frame, text="Salary:", font=("Arial", 10, "bold"), bg="white", fg="blue").grid(row=4, column=0, sticky="w", padx=5, pady=(5,0))
        self.salary_entry = tk.Entry(user_frame, width=25, bg="white", fg="black")
        self.salary_entry.grid(row=5, column=0, padx=5, pady=(0,5))

        tk.Button(user_frame, text="Set Budget", command=self.set_budget, bg="white", fg="blue").grid(row=6, column=0, pady=10)


    def create_expenses_frame(self):
        expenses_frame = tk.Frame(self.window, bg="white")
        expenses_frame.pack(padx=10, pady=10)

        tk.Label(expenses_frame, text="Expense Price:", font=("Arial", 10, "bold"), bg="white", fg="blue").grid(row=0, column=0, sticky="w", padx=5, pady=(5,0))
        self.price_entry = tk.Entry(expenses_frame, width=25, bg="white", fg="black")
        self.price_entry.grid(row=1, column=0, padx=5, pady=(0,5))

        # Category Dropdown using ttk.Combobox
        tk.Label(expenses_frame, text="Category:", font=("Arial", 10, "bold"), bg="white", fg="blue").grid(row=2, column=0, sticky="w", padx=5, pady=(5,0))
        self.category_var = tk.StringVar(value="Food")
        categories = ["Food", "Transport", "Rent", "Entertainment", "Other"]
        self.category_menu = ttk.Combobox(expenses_frame, textvariable=self.category_var, values=categories, state="readonly", width=23)
        self.category_menu.grid(row=3, column=0, padx=5, pady=(0,5))

        # Buttons
        tk.Button(expenses_frame, text="Add Expense", command=self.add_expense, bg="blue", fg="white").grid(row=4, column=0, padx=5, pady=(5,5))
        tk.Button(expenses_frame, text="View Expenses", command=self.view_expenses, bg="blue", fg="white").grid(row=5, column=0, padx=5, pady=(5,5))
        tk.Button(expenses_frame, text="View Summary", command=self.show_summary, bg="blue", fg="white").grid(row=6, column=0, padx=5, pady=(5,5))

    def create_summary_frame(self):
        self.summary_frame = tk.Frame(self.window, bg="black")
        self.summary_frame.pack(pady=10)

        self.savings_label = tk.Label(self.summary_frame, text="Your Savings: $0", font=("Arial", 12, "bold"), bg="black", fg="blue")
        self.savings_label.pack()

        # Progress bar
        style = ttk.Style()
        style.theme_use('clam')
        style.configure("blue.Horizontal.TProgressbar", troughcolor='black', background='blue', thickness=20)
        self.progress = ttk.Progressbar(self.summary_frame, length=250, mode='determinate', style="blue.Horizontal.TProgressbar")
        self.progress.pack(pady=5)


    def set_budget(self):
        try:
            name = self.name_entry.get()
            age = int(self.age_entry.get())
            salary = float(self.salary_entry.get())

            self.user = User(name, age, salary)
            messagebox.showinfo("Budget Set", f"Budget set for {self.user.name} with a salary of ${self.user.salary:.2f}")
            self.update_savings()
            self.save_data()

        except ValueError:
            messagebox.showerror("Input Error", "Please enter valid numbers for age and salary.")

    def add_expense(self):
        if not self.user:
            messagebox.showwarning("No Budget Set", "Please enter your budget details first.")
            return

        try:
            price = float(self.price_entry.get())
            category = self.category_var.get()

            if price <= 0:
                raise ValueError

            self.expenses.log_expense(price, category)
            messagebox.showinfo("Expense Logged", f"Added ${price:.2f} under {category}.")
            self.update_savings()
            self.save_data()

        except ValueError:
            messagebox.showerror("Input Error", "Please enter a valid price.")

    def view_expenses(self):
        if not self.expenses.expenses:
            messagebox.showinfo("No Expenses", "You have not logged any expenses yet.")
            return

        expense_list = "\n".join([f"${price:.2f} - {category}" for price, category in self.expenses.expenses])
        messagebox.showinfo("Expenses", f"Your Expenses:\n{expense_list}")

    def show_summary(self):
        if not self.expenses.expenses:
            messagebox.showinfo("Summary", "No expenses logged yet.")
            return

        breakdown = defaultdict(float)
        for price, category in self.expenses.expenses:
            breakdown[category] += price

        summary_text = "\n".join(f"{cat}: ${amt:.2f}" for cat, amt in breakdown.items())
        messagebox.showinfo("Expense Breakdown", summary_text)

    def update_savings(self):
        if self.user:
            total_expenses = self.expenses.get_total_expenses()
            savings = self.user.salary - total_expenses
            percent = (total_expenses / self.user.salary) * 100 if self.user.salary else 0
            self.savings_label.config(text=f"Your Savings: ${savings:.2f} ({percent:.1f}% spent)")
            self.progress['value'] = min(percent, 100)

    
    def save_data(self):
        data = {
            "user": vars(self.user) if self.user else None,
            "expenses": self.expenses.expenses
        }
        with open("budget_data.json", "w") as f:
            json.dump(data, f)

    def load_data(self):
        try:
            with open("budget_data.json", "r") as f:
                data = json.load(f)
                if data["user"]:
                    u = data["user"]
                    self.user = User(u["name"], u["age"], u["salary"])
                    self.name_entry.insert(0, u["name"])
                    self.age_entry.insert(0, str(u["age"]))
                    self.salary_entry.insert(0, str(u["salary"]))
                self.expenses.expenses = data["expenses"]
                self.update_savings()
        except FileNotFoundError:
            pass

if __name__ == "__main__":
    root = tk.Tk()
    app = BudgetTracker(root)
    root.mainloop()