Create app.py

app.py

@@ -0,0 +1,108 @@
+import streamlit as st
+import sqlite3
+import random
+
+# ---------------- DATABASE SETUP ----------------
+conn = sqlite3.connect('bank.db', check_same_thread=False)
+cursor = conn.cursor()
+
+cursor.execute('''
+CREATE TABLE IF NOT EXISTS accounts (
+    name TEXT,
+    age INTEGER,
+    password TEXT,
+    account_number INTEGER PRIMARY KEY,
+    balance INTEGER
+)
+''')
+conn.commit()
+
+# ---------------- FUNCTIONS ----------------
+
+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!"
+
+    cursor.execute("INSERT INTO accounts VALUES (?, ?, ?, ?, ?)",
+                   (name, age, password, acc_no, 0))
+    conn.commit()
+    
+    return f"✅ Account Created! Your Account Number: {acc_no}"
+
+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 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 "❌ 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
+
+# ---------------- STREAMLIT UI ----------------
+
+st.title("🏦 Baryashah Bank")
+
+menu = st.sidebar.selectbox("Menu", ["Create Account", "Login"])
+
+# 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)
+
+# 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.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")