Update app.py

app.py

@@ -27,7 +27,7 @@ def send_otp():
     return str(random.randint(1000,9999))
 
 # --- User Options ---
-menu = ["Create Account", "Login / Check Balance", "Reset Password"]
+menu = ["Create Account", "Login / Check Balance", "Reset Password", "Recover Account Number"]
 choice = st.sidebar.selectbox("Menu", menu)
 
 accounts = load_accounts()
@@ -38,14 +38,18 @@ if choice == "Create Account":
     name = st.text_input("Full Name")
     age = st.number_input("Age", min_value=1, max_value=120)
     password = st.text_input("Create Password", type="password")
+    pin = st.text_input("Create 4-digit PIN (for withdrawals)", type="password", max_chars=4)
     if st.button("Create Account", key="create"):
         if name in accounts:
             st.error("Account already exists!")
+        elif len(pin)!=4 or not pin.isdigit():
+            st.error("PIN must be 4 digits")
         else:
             acc_no = random.randint(1000,9999)
             accounts[name] = {
                 "age": age,
                 "password": password,
+                "pin": pin,
                 "account_number": acc_no,
                 "balance": 0,
                 "transactions": []
@@ -73,6 +77,17 @@ elif choice == "Reset Password":
             else:
                 st.error("Invalid OTP!")
 
+# --- Account Number Recovery ---
+elif choice == "Recover Account Number":
+    st.subheader("Recover Account Number")
+    name = st.text_input("Enter Account Name")
+    password = st.text_input("Enter Password", type="password")
+    if st.button("Recover Account Number"):
+        if name in accounts and accounts[name]['password'] == password:
+            st.success(f"Your Account Number is: {accounts[name]['account_number']}")
+        else:
+            st.error("Invalid credentials!")
+
 # --- Login / Check Balance ---
 elif choice == "Login / Check Balance":
     st.subheader("Login")
@@ -88,15 +103,18 @@ elif choice == "Login / Check Balance":
             col1, col2 = st.columns(2)
             with col1:
                 withdraw_amt = st.number_input("Withdraw Amount", min_value=0)
+                user_pin = st.text_input("Enter PIN for Withdrawal", type="password")
                 if st.button("Withdraw", key="withdraw"):
-                    if withdraw_amt <= accounts[name]['balance']:
+                    if user_pin != accounts[name]['pin']:
+                        st.error("Incorrect PIN! Withdrawal denied.")
+                    elif withdraw_amt > accounts[name]['balance']:
+                        st.error("Insufficient Balance!")
+                    else:
                         accounts[name]['balance'] -= withdraw_amt
                         accounts[name]['transactions'].append({"type":"Withdraw", "amount":withdraw_amt})
                         save_accounts(accounts)
                         st.success(f"💸 Withdraw Successful! New Balance: {accounts[name]['balance']}")
                         st.balloons()
-                    else:
-                        st.error("Insufficient Balance!")
 
             with col2:
                 deposit_amt = st.number_input("Deposit Amount", min_value=0)
@@ -115,4 +133,4 @@ elif choice == "Login / Check Balance":
                 fig = px.bar(x=list(range(1,len(amounts)+1)), y=amounts, color=types, labels={'x':'Transaction','y':'Amount'}, title="Transaction History")
                 st.plotly_chart(fig)
         else:
-            st.error("Login failed! Check credentials.")
+            st.error("Login failed! Check credentials.")