Update app.py

app.py

@@ -1,110 +1,85 @@
 import streamlit as st
 
-# ------------------------
-# Bank Account OOP Section
-# ------------------------
-
+# BankAccount class using OOP
 class BankAccount:
-    def __init__(self, account_number, owner, balance=0.0):
-        self.account_number = account_number
-        self.owner = owner
-        self.balance = balance
+    def __init__(self, account_id, name):
+        self.account_id = account_id
+        self.name = name
+        self.balance = 0
 
     def deposit(self, amount):
         if amount > 0:
             self.balance += amount
-            return f"Deposited {amount} to {self.owner}'s account."
-        return "Invalid deposit amount."
+            return True
+        return False
 
     def withdraw(self, amount):
         if 0 < amount <= self.balance:
             self.balance -= amount
-            return f"Withdrew {amount} from {self.owner}'s account."
-        return "Insufficient balance or invalid amount."
-
-    def to_dict(self):
-        return {
-            "Account Number": self.account_number,
-            "Owner": self.owner,
-            "Balance": f"${self.balance:.2f}"
-        }
-
-class Bank:
-    def __init__(self):
-        self.accounts = {}
-
-    def create_account(self, account_number, owner):
-        if account_number in self.accounts:
-            return "Account number already exists."
-        self.accounts[account_number] = BankAccount(account_number, owner)
-        return f"Account created for {owner}."
-
-    def get_account(self, account_number):
-        return self.accounts.get(account_number)
-
-    def get_all_accounts(self):
-        return [acc.to_dict() for acc in self.accounts.values()]
-
-
-# ------------------------
-# Streamlit UI Section
-# ------------------------
-
-# Initialize session state for bank object
-if "bank" not in st.session_state:
-    st.session_state.bank = Bank()
-
-st.title("🏦 Simple Bank Account Manager")
-
-menu = st.sidebar.radio("Select Action", ["Create Account", "Deposit", "Withdraw"])
-
-if menu == "Create Account":
-    st.subheader("🧾 Create New Bank Account")
-    acc_num = st.text_input("Enter Account Number", key="create_acc_num")
-    owner = st.text_input("Enter Owner's Name", key="create_owner")
-    if st.button("Create Account"):
-        if acc_num and owner:
-            message = st.session_state.bank.create_account(acc_num, owner)
-            st.success(message)
-        else:
-            st.warning("Please fill all fields.")
-
-elif menu == "Deposit":
-    st.subheader("💰 Deposit to Account")
-    acc_num = st.text_input("Enter Account Number", key="deposit_acc_num")
-    amount = st.number_input("Enter Amount to Deposit", min_value=0.0, key="deposit_amt")
-    if st.button("Deposit"):
-        bank = st.session_state.bank
-        account = bank.get_account(acc_num)
-        if account:
-            message = account.deposit(amount)
-            st.success(message)
+            return True
+        return False
+
+# Store accounts in session state
+if 'accounts' not in st.session_state:
+    st.session_state.accounts = {}
+
+st.set_page_config(page_title="🏦 Bank App", layout="centered")
+st.title("🏦 Bank Account Management System")
+
+# Section 1: Create New Account
+st.header("➕ Create New Account")
+with st.form("create_account"):
+    new_id = st.text_input("Enter Account ID")
+    new_name = st.text_input("Enter Account Holder Name")
+    create_btn = st.form_submit_button("Create Account")
+    if create_btn:
+        if new_id in st.session_state.accounts:
+            st.warning("⚠️ Account ID already exists.")
+        elif not new_id or not new_name:
+            st.warning("⚠️ Please enter both ID and Name.")
         else:
-            st.error("Account not found.")
-
-elif menu == "Withdraw":
-    st.subheader("💸 Withdraw from Account")
-    acc_num = st.text_input("Enter Account Number", key="withdraw_acc_num")
-    amount = st.number_input("Enter Amount to Withdraw", min_value=0.0, key="withdraw_amt")
-    if st.button("Withdraw"):
-        bank = st.session_state.bank
-        account = bank.get_account(acc_num)
+            st.session_state.accounts[new_id] = BankAccount(new_id, new_name)
+            st.success(f"✅ Account for {new_name} created with ID {new_id}.")
+
+# Section 2: Deposit / Withdraw
+st.header("💸 Deposit / Withdraw Money")
+with st.form("transaction"):
+    trans_id = st.text_input("Enter Account ID")
+    trans_type = st.radio("Choose Transaction Type", ["Deposit", "Withdraw"], horizontal=True)
+    trans_amount = st.number_input("Enter Amount (Rs.)", min_value=0, step=100)
+    trans_btn = st.form_submit_button("Submit Transaction")
+
+    if trans_btn:
+        account = st.session_state.accounts.get(trans_id)
         if account:
-            message = account.withdraw(amount)
-            st.success(message)
+            if trans_type == "Deposit":
+                if account.deposit(trans_amount):
+                    st.success(f"✅ Deposited Rs. {trans_amount} to {account.name}'s account.")
+                else:
+                    st.error("❌ Invalid deposit amount.")
+            elif trans_type == "Withdraw":
+                if account.withdraw(trans_amount):
+                    st.success(f"✅ Withdrew Rs. {trans_amount} from {account.name}'s account.")
+                else:
+                    st.error("❌ Insufficient balance or invalid amount.")
         else:
-            st.error("Account not found.")
-
-# ------------------------
-# Display Account List
-# ------------------------
-
-st.divider()
-st.subheader("📋 All Bank Accounts")
-accounts = st.session_state.bank.get_all_accounts()
-
-if accounts:
-    st.table(accounts)
+            st.warning("⚠️ Account ID not found.")
+
+# Section 3: Show All Accounts
+st.header("📄 All Accounts")
+if st.session_state.accounts:
+    account_data = {
+        "Account ID": [],
+        "Name": [],
+        "Balance (Rs.)": []
+    }
+    for acc in st.session_state.accounts.values():
+        account_data["Account ID"].append(acc.account_id)
+        account_data["Name"].append(acc.name)
+        account_data["Balance (Rs.)"].append(acc.balance)
+    
+    st.table(account_data)
 else:
-    st.info("No accounts created yet.")
+    st.info("ℹ️ No accounts created yet.")
+