main.py

from balance import show_balance
from deposit import deposit_money
from withdraw import withdraw_money
from statement import show_statement

while True:
    print("\n--- ATM Menu ---")
    print("1. Check Balance")
    print("2. Deposit")
    print("3. Withdraw")
    print("4. Statement")
    print("5. Exit")

    choice = input("Enter your choice: ")

    if choice == "1":
        show_balance()
    elif choice == "2":
        deposit_money()
    elif choice == "3":
        withdraw_money()
    elif choice == "4":
        show_statement()
    elif choice == "5":
        print("Thank you for using ATM")
        break
    else:
        print("Invalid choice")