Update app.py

app.py

@@ -21,24 +21,26 @@ conn.commit()
 
 def create_account(name, age, password):
     acc_no = random.randint(1000, 9999)
-    
+
     cursor.execute("SELECT * FROM accounts WHERE name=?", (name,))
     if cursor.fetchone():
-        return "❌ Account already exists!"
+        return None, "❌ Account already exists!"
 
     cursor.execute("INSERT INTO accounts VALUES (?, ?, ?, ?, ?)",
                    (name, age, password, acc_no, 0))
     conn.commit()
-    
-    return f"✅ Account Created! Your Account Number: {acc_no}"
+
+    return acc_no, "✅ Account Created!"
 
 def login(name, acc_no, password):
     cursor.execute("SELECT * FROM accounts WHERE name=? AND account_number=? AND password=?",
                    (name, acc_no, password))
     return cursor.fetchone()
 
-def get_balance(user):
-    return user[4]
+def get_user(name, acc_no, password):
+    cursor.execute("SELECT * FROM accounts WHERE name=? AND account_number=? AND password=?",
+                   (name, acc_no, password))
+    return cursor.fetchone()
 
 def deposit(user, amount):
     new_balance = user[4] + amount
@@ -49,60 +51,98 @@ def deposit(user, amount):
 
 def withdraw(user, amount):
     if user[4] < amount:
-        return "❌ Insufficient Balance"
-    
+        return None, "❌ Insufficient Balance"
+
     new_balance = user[4] - amount
     cursor.execute("UPDATE accounts SET balance=? WHERE account_number=?",
                    (new_balance, user[3]))
     conn.commit()
-    return new_balance
+    return new_balance, "✅ Withdrawal Successful"
 
-# ---------------- STREAMLIT UI ----------------
+# ---------------- SESSION INIT ----------------
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
+
+# ---------------- UI ----------------
 
 st.title("🏦 Baryashah Bank")
 
-menu = st.sidebar.selectbox("Menu", ["Create Account", "Login"])
+# ---------------- SIDEBAR ----------------
+if not st.session_state.logged_in:
+    menu = st.sidebar.selectbox("Menu", ["Create Account", "Login"])
+else:
+    menu = "Dashboard"
 
-# CREATE ACCOUNT
+# ---------------- CREATE ACCOUNT ----------------
 if menu == "Create Account":
     st.subheader("Create New Account")
+
     name = st.text_input("Name")
     age = st.number_input("Age", min_value=1)
     password = st.text_input("Password", type="password")
 
-    if st.button("Create"):
-        result = create_account(name, age, password)
-        st.success(result)
+    if st.button("Create Account"):
+        acc_no, msg = create_account(name, age, password)
+        if acc_no:
+            st.success(f"{msg} 🎉")
+            st.info(f"Your Account Number: {acc_no}")
+        else:
+            st.error(msg)
 
-# LOGIN
+# ---------------- LOGIN ----------------
 elif menu == "Login":
     st.subheader("Login")
+
     name = st.text_input("Name")
     acc_no = st.number_input("Account Number")
     password = st.text_input("Password", type="password")
 
     if st.button("Login"):
         user = login(name, acc_no, password)
-        
+
         if user:
+            st.session_state.logged_in = True
+            st.session_state.user = (name, acc_no, password)
             st.success("✅ Login Successful")
-
-            option = st.selectbox("Choose", ["Check Balance", "Deposit", "Withdraw"])
-
-            if option == "Check Balance":
-                st.info(f"💰 Balance: {get_balance(user)}")
-
-            elif option == "Deposit":
-                amount = st.number_input("Amount", min_value=1)
-                if st.button("Deposit"):
-                    new_balance = deposit(user, amount)
-                    st.success(f"New Balance: {new_balance}")
-
-            elif option == "Withdraw":
-                amount = st.number_input("Amount", min_value=1)
-                if st.button("Withdraw"):
-                    result = withdraw(user, amount)
-                    st.success(result)
-
         else:
-            st.error("❌ Invalid Credentials")
+            st.error("❌ Invalid Credentials")
+
+# ---------------- DASHBOARD ----------------
+elif menu == "Dashboard":
+
+    name, acc_no, password = st.session_state.user
+    user = get_user(name, acc_no, password)  # 🔥 always fresh data
+
+    st.subheader(f"Welcome, {name} 👋")
+
+    option = st.selectbox("Choose Action", ["Check Balance", "Deposit", "Withdraw"])
+
+    # CHECK BALANCE
+    if option == "Check Balance":
+        st.info(f"💰 Your Balance: {user[4]}")
+
+    # DEPOSIT
+    elif option == "Deposit":
+        amount = st.number_input("Enter Amount", min_value=1)
+        if st.button("Deposit"):
+            new_balance = deposit(user, amount)
+            st.success(f"✅ Deposited Successfully")
+            st.info(f"💰 New Balance: {new_balance}")
+
+    # WITHDRAW
+    elif option == "Withdraw":
+        amount = st.number_input("Enter Amount", min_value=1)
+        if st.button("Withdraw"):
+            new_balance, msg = withdraw(user, amount)
+
+            if new_balance is None:
+                st.error(msg)
+            else:
+                st.success(msg)
+                st.info(f"💰 Remaining Balance: {new_balance}")
+
+    # LOGOUT
+    if st.button("Logout"):
+        st.session_state.logged_in = False
+        st.session_state.user = None
+        st.success("👋 Logged out successfully")