Update app.py

app.py

@@ -1,191 +1,118 @@
-import streamlit as st
-import json
 import os
 import random
-import hashlib
-from datetime import datetime
-
-# ---------------- FILE SETUP ----------------
-USER_FILE = "users.json"
-TRANS_FILE = "transactions.json"
-
-if not os.path.exists(USER_FILE):
-    with open(USER_FILE, "w") as f:
-        json.dump({}, f)
+import json
+import streamlit as st
+import plotly.express as px
 
-if not os.path.exists(TRANS_FILE):
-    with open(TRANS_FILE, "w") as f:
+# --- File setup ---
+if not os.path.exists("Account_Details.json"):
+    with open("Account_Details.json", "w") as f:
         json.dump({}, f)
 
-# ---------------- PASSWORD HASHING ----------------
-def hash_password(password):
-    return hashlib.sha256(password.encode()).hexdigest()
+# --- App UI ---
+st.set_page_config(page_title="Baryashah Bank App", layout="wide")
+st.markdown("<h1 style='text-align:center;color:darkblue;'>🌟 Baryashah Bank 🌟</h1>", unsafe_allow_html=True)
+st.write("---")
 
-# ---------------- USER FUNCTIONS ----------------
-def load_users():
-    with open(USER_FILE, "r") as f:
+# --- Helper Functions ---
+def load_accounts():
+    with open("Account_Details.json", "r") as f:
         return json.load(f)
 
-def save_users(users):
-    with open(USER_FILE, "w") as f:
-        json.dump(users, f)
-
-def create_account(name, age, password):
-    users = load_users()
-    if name in users:
-        return None, "❌ Account already exists!"
-    acc_no = random.randint(1000, 9999)
-    users[name] = {
-        "age": age,
-        "password": hash_password(password),
-        "account_number": acc_no,
-        "balance": 0
-    }
-    save_users(users)
-    return acc_no, "✅ Account Created!"
-
-def login(name, acc_no, password):
-    users = load_users()
-    if name in users:
-        user = users[name]
-        if user["account_number"] == acc_no and user["password"] == hash_password(password):
-            return user
-    return None
-
-def update_balance(name, new_balance):
-    users = load_users()
-    users[name]["balance"] = new_balance
-    save_users(users)
-
-# ---------------- TRANSACTION FUNCTIONS ----------------
-def record_transaction(name, type_, amount):
-    with open(TRANS_FILE, "r") as f:
-        transactions = json.load(f)
-    if name not in transactions:
-        transactions[name] = []
-    transactions[name].append({
-        "type": type_,
-        "amount": amount,
-        "time": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    })
-    with open(TRANS_FILE, "w") as f:
-        json.dump(transactions, f)
-
-def get_transactions(name):
-    with open(TRANS_FILE, "r") as f:
-        transactions = json.load(f)
-    return transactions.get(name, [])
+def save_accounts(data):
+    with open("Account_Details.json", "w") as f:
+        json.dump(data, f, indent=4)
 
-# ---------------- STREAMLIT UI ----------------
-st.set_page_config(page_title="Baryashah Bank", layout="wide")
-st.title("🏦 Baryashah Bank")
+def send_otp():
+    return str(random.randint(1000,9999))
 
-if "logged_in" not in st.session_state:
-    st.session_state.logged_in = False
-if "user_name" not in st.session_state:
-    st.session_state.user_name = None
+# --- User Options ---
+menu = ["Create Account", "Login / Check Balance", "Reset Password"]
+choice = st.sidebar.selectbox("Menu", menu)
 
-# ---------------- SIDEBAR ----------------
-if not st.session_state.logged_in:
-    menu = st.sidebar.selectbox("Menu", ["Create Account", "Login"])
-else:
-    menu = "Dashboard"
+accounts = load_accounts()
 
-# ---------------- CREATE ACCOUNT ----------------
-if menu == "Create Account":
-    st.subheader("🆕 Create New Account")
-    col1, col2 = st.columns([2,3])
-    with col1:
-        name = st.text_input("Name")
-        age = st.number_input("Age", min_value=1)
-    with col2:
-        password = st.text_input("Password", type="password")
-        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 ----------------
-elif menu == "Login":
-    st.subheader("🔑 Login")
-    col1, col2 = st.columns([2,3])
-    with col1:
-        name = st.text_input("Name")
-        acc_no = st.number_input("Account Number")
-    with col2:
-        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 = name
-                st.success("✅ Login Successful")
-            else:
-                st.error("❌ Invalid Credentials")
-
-# ---------------- DASHBOARD ----------------
-elif menu == "Dashboard":
-    name = st.session_state.user_name
-    users = load_users()
-    user = users[name]
-
-    st.subheader(f"👋 Welcome, {name}")
-
-    # ---------------- BALANCE CARD ----------------
-    st.markdown(f"""
-        <div style="background-color:#d1f7c4;padding:20px;border-radius:15px">
-        <h3>💰 Balance: {user['balance']}</h3>
-        <p>Account Number: {user['account_number']}</p>
-        </div>
-        """, unsafe_allow_html=True)
-
-    # ---------------- ACTIONS ----------------
-    col1, col2 = st.columns(2)
-    with col1:
-        option = st.selectbox("Choose Action", ["Check Balance", "Deposit", "Withdraw", "Transactions"])
-    with col2:
-        st.markdown("<br>", unsafe_allow_html=True)
-        if st.button("Logout"):
-            st.session_state.logged_in = False
-            st.session_state.user_name = None
-            st.success("👋 Logged out successfully")
-
-    # ---------------- CHECK BALANCE ----------------
-    if option == "Check Balance":
-        st.success(f"💰 Your Balance: {user['balance']}")
-
-    # ---------------- DEPOSIT ----------------
-    elif option == "Deposit":
-        amount = st.number_input("Enter Amount", min_value=1)
-        if st.button("Deposit Now"):
-            user["balance"] += amount
-            update_balance(name, user["balance"])
-            record_transaction(name, "Deposit", amount)
-            st.success(f"✅ Deposited {amount}")
-            st.info(f"💰 New Balance: {user['balance']}")
-
-    # ---------------- WITHDRAW ----------------
-    elif option == "Withdraw":
-        amount = st.number_input("Enter Amount", min_value=1)
-        if st.button("Withdraw Now"):
-            if user["balance"] < amount:
-                st.error("❌ Insufficient Balance")
+# --- Create Account ---
+if choice == "Create Account":
+    st.subheader("Create New 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")
+    if st.button("Create Account", key="create"):
+        if name in accounts:
+            st.error("Account already exists!")
+        else:
+            acc_no = random.randint(1000,9999)
+            accounts[name] = {
+                "age": age,
+                "password": password,
+                "account_number": acc_no,
+                "balance": 0,
+                "transactions": []
+            }
+            save_accounts(accounts)
+            st.success(f"Account created! Your Account Number: {acc_no}")
+            st.balloons()
+
+# --- Password Reset ---
+elif choice == "Reset Password":
+    st.subheader("Reset Your Password")
+    name = st.text_input("Enter Account Name")
+    if name in accounts:
+        if st.button("Send OTP"):
+            otp = send_otp()
+            st.session_state['otp'] = otp
+            st.info(f"Simulated OTP (for demo): {otp}")
+        user_otp = st.text_input("Enter OTP")
+        new_pass = st.text_input("Enter New Password", type="password")
+        if st.button("Reset Password", key="reset"):
+            if 'otp' in st.session_state and user_otp == st.session_state['otp']:
+                accounts[name]['password'] = new_pass
+                save_accounts(accounts)
+                st.success("Password reset successfully!")
             else:
-                user["balance"] -= amount
-                update_balance(name, user["balance"])
-                record_transaction(name, "Withdraw", amount)
-                st.success(f"✅ Withdrawn {amount}")
-                st.info(f"💰 Remaining Balance: {user['balance']}")
-
-    # ---------------- TRANSACTIONS ----------------
-    elif option == "Transactions":
-        txns = get_transactions(name)
-        if txns:
-            st.subheader("📜 Transaction History")
-            for t in reversed(txns):
-                st.info(f"{t['time']} → {t['type']}: {t['amount']}")
+                st.error("Invalid OTP!")
+
+# --- Login / Check Balance ---
+elif choice == "Login / Check Balance":
+    st.subheader("Login")
+    name = st.text_input("Account Name")
+    acc_no = st.number_input("Account Number", min_value=1000, max_value=9999)
+    password = st.text_input("Password", type="password")
+    if st.button("Login"):
+        if name in accounts and accounts[name]['account_number'] == acc_no and accounts[name]['password'] == password:
+            st.success(f"Welcome {name}! ✅")
+            st.write(f"💰 Current Balance: {accounts[name]['balance']}")
+
+            # --- Transaction Section ---
+            col1, col2 = st.columns(2)
+            with col1:
+                withdraw_amt = st.number_input("Withdraw Amount", min_value=0)
+                if st.button("Withdraw", key="withdraw"):
+                    if withdraw_amt <= accounts[name]['balance']:
+                        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)
+                if st.button("Deposit", key="deposit"):
+                    accounts[name]['balance'] += deposit_amt
+                    accounts[name]['transactions'].append({"type":"Deposit", "amount":deposit_amt})
+                    save_accounts(accounts)
+                    st.success(f"💰 Deposit Successful! New Balance: {accounts[name]['balance']}")
+                    st.balloons()
+
+            # --- Transaction Chart ---
+            if accounts[name]['transactions']:
+                df = accounts[name]['transactions']
+                amounts = [t['amount'] if t['type']=='Deposit' else -t['amount'] for t in df]
+                types = [t['type'] for t in df]
+                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.info("No transactions yet")
+            st.error("Login failed! Check credentials.")