Bank Account Management System with Transaction Analysis.py

# Bank Account Management System with Transaction Analytics

import matplotlib.pyplot as plt
import numpy as np


class BankAccount: 

    # Create the account object
    def __init__(self, name, pin, balance=0):
        self.name = name
        self.balance = balance
        self.pin = pin 
        self.history = []

    # Deposit funds
    def deposit(self, amount):
        self.balance += amount
        self.history.append(f"Deposited ${amount}")
        print(f"Amount deposited: ${amount}. New balance: ${self.balance}")

    # Withdraw funds
    def withdrawal(self, amount):
        if amount <= self.balance:
            self.balance -= amount
            self.history.append(f'Withdrew ${amount}')
            print(f"Amount withdrawn: ${amount}. New balance: ${self.balance}")
        else:
            print("Insufficient funds!")

    # Show balance
    def show_balance(self):
        print(f"Account '{self.name}' balance: ${self.balance}")

    # Change PIN
    def change_pin(self):
        new_pin = input("Enter new 4-digit PIN: ")
        if new_pin == self.pin:
            print("You cannot use the same PIN.")
        elif not new_pin.isdigit() or len(new_pin) != 4:
            print("PIN must be exactly 4 digits.")
        else:
            self.pin = new_pin
            print("PIN successfully changed.")

    # Change account name
    def new_account_name(self):
        new_name = input("Enter new account name: ")
        if new_name == self.name:
            print("You cannot use the same account name.")
        else:
            self.name = new_name
            print("Account name successfully changed.")

    # Show transaction history
    def transaction_history(self):
        if not self.history:
            print("No transactions have been made!")
        else:
            print("\n*** Transaction History ***")
            for x in self.history:
                print(x)

    # Show transaction trend (Option 7)
    def transaction_trend(self):
        deposits = []
        withdrawals = []
        for entry in self.history:
            if "Deposited" in entry:
                deposits.append(float(entry.split("$")[1]))
            elif "Withdrew" in entry:
                withdrawals.append(float(entry.split("$")[1]))

        plt.figure(figsize=(8,5))
        plt.plot(range(1, len(deposits)+1), deposits, marker='o', label='Deposits')
        plt.plot(range(1, len(withdrawals)+1), withdrawals, marker='o', label='Withdrawals')
        plt.title(f"Transaction Trend for {self.name}")
        plt.xlabel("Transaction Number")
        plt.ylabel("Amount ($)")
        plt.legend()
        plt.grid(True)

 # Set x-axis ticks to increment by 1
        max_x = max(len(deposits), len(withdrawals))
        plt.xticks(np.arange(1, max_x + 1, 1))
  
# Set y-axis ticks to show all transaction amounts
        all_amounts = deposits + withdrawals
        if all_amounts:  # only if there are transactions
            plt.yticks(sorted(all_amounts))

        plt.show()




# Main Program
name = input("Enter your account name: ")
pin = input("Enter 4-digit PIN: ")
account = BankAccount(name, pin)

# Authentication
entered_pin = input("Enter your PIN to access account: ")
if entered_pin != account.pin:
    print("Incorrect PIN! Access denied.")
    exit()

# Main menu loop
while True:
    print("\n1. Deposit  2. Withdraw  3. Show balance  4. Show Transaction History")
    print("5. Change PIN  6. Change Account Name  7. Show Transaction Trend  8. Exit")
    choice = input("Choose an option: ")

    if choice == "1":
        amt = float(input("Enter amount to deposit: "))
        account.deposit(amt)
    elif choice == "2":
        amt = float(input("Enter amount to withdraw: "))
        account.withdrawal(amt)
    elif choice == "3":
        account.show_balance()
    elif choice == "4":
        account.transaction_history()
    elif choice == "5":
        account.change_pin()
    elif choice == "6":
        account.new_account_name()
    elif choice == "7":
        account.transaction_trend()
    elif choice == "8":
        print("Exiting... Goodbye!")
        break
    else:
        print("Invalid choice. Try again.")