Update app.py

app.py

@@ -1,71 +1,89 @@
 import streamlit as st
-import sqlite3
+import json
+import os
 import random
+import hashlib
+from datetime import datetime
 
-# ---------------- DATABASE SETUP ----------------
-conn = sqlite3.connect('bank.db', check_same_thread=False)
-cursor = conn.cursor()
+# ---------------- FILE SETUP ----------------
+USER_FILE = "users.json"
+TRANS_FILE = "transactions.json"
 
-cursor.execute('''
-CREATE TABLE IF NOT EXISTS accounts (
-    name TEXT,
-    age INTEGER,
-    password TEXT,
-    account_number INTEGER PRIMARY KEY,
-    balance INTEGER
-)
-''')
-conn.commit()
+if not os.path.exists(USER_FILE):
+    with open(USER_FILE, "w") as f:
+        json.dump({}, f)
 
-# ---------------- FUNCTIONS ----------------
+if not os.path.exists(TRANS_FILE):
+    with open(TRANS_FILE, "w") as f:
+        json.dump({}, f)
 
-def create_account(name, age, password):
-    acc_no = random.randint(1000, 9999)
+# ---------------- PASSWORD HASHING ----------------
+def hash_password(password):
+    return hashlib.sha256(password.encode()).hexdigest()
 
-    cursor.execute("SELECT * FROM accounts WHERE name=?", (name,))
-    if cursor.fetchone():
-        return None, "❌ Account already exists!"
+# ---------------- USER FUNCTIONS ----------------
+def load_users():
+    with open(USER_FILE, "r") as f:
+        return json.load(f)
 
-    cursor.execute("INSERT INTO accounts VALUES (?, ?, ?, ?, ?)",
-                   (name, age, password, acc_no, 0))
-    conn.commit()
+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):
-    cursor.execute("SELECT * FROM accounts WHERE name=? AND account_number=? AND password=?",
-                   (name, acc_no, password))
-    return cursor.fetchone()
-
-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
-    cursor.execute("UPDATE accounts SET balance=? WHERE account_number=?",
-                   (new_balance, user[3]))
-    conn.commit()
-    return new_balance
-
-def withdraw(user, amount):
-    if user[4] < amount:
-        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, "✅ Withdrawal Successful"
-
-# ---------------- SESSION INIT ----------------
+    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, [])
+
+# ---------------- STREAMLIT UI ----------------
+st.set_page_config(page_title="Baryashah Bank", layout="wide")
+st.title("🏦 Baryashah Bank")
+
 if "logged_in" not in st.session_state:
     st.session_state.logged_in = False
-
-# ---------------- UI ----------------
-
-st.title("🏦 Baryashah Bank")
+if "user_name" not in st.session_state:
+    st.session_state.user_name = None
 
 # ---------------- SIDEBAR ----------------
 if not st.session_state.logged_in:
@@ -75,74 +93,99 @@ else:
 
 # ---------------- 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 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)
+    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")
-
-    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")
-        else:
-            st.error("❌ Invalid Credentials")
+    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, 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
+    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.info(f"💰 Your Balance: {user[4]}")
+        st.success(f"💰 Your Balance: {user['balance']}")
 
-    # DEPOSIT
+    # ---------------- 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
+        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"):
-            new_balance, msg = withdraw(user, amount)
-
-            if new_balance is None:
-                st.error(msg)
+        if st.button("Withdraw Now"):
+            if user["balance"] < amount:
+                st.error("❌ Insufficient Balance")
             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")
+                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']}")
+        else:
+            st.info("No transactions yet")