Viết chương trình rút tiền ATM bằng python giúp mình với mình đang cần gấp

Tham khảo:

# Số dư khả dụng trong tài khoản ngân hàng
balance = 10000

# Hàm rút tiền
def withdraw(amount):
    global balance  # Sử dụng biến toàn cục balance
    if amount > balance:
        print("Số dư không đủ.")
    else:
        balance -= amount
        print(f"Đã rút {amount} VNĐ từ tài khoản.")
        print(f"Số dư hiện tại là: {balance} VNĐ.")

# Kiểm tra chương trình
withdraw(5000)
withdraw(3000)
withdraw(8000)

`@` Lưu ý: Đây chỉ là chương trình minh họa cách thức hoạt động của việc rút tiền từ ATM.

print("------------Chào mừng đến ngân hàng-------------");
ht=10000000;  #vd
print("Tài khoản:",ht);
while True:
    n=int(input("Bạn muốn rút bao nhiêu tiền trong ngân hàng:"));
    if ht<n:
        print("Không rút được.Trong tài khoản có:",ht);
    else:
        print("Thành công");
        print("Hiện tại trong tài khoản còn:",ht-n);