Update app.py

app.py

@@ -1,183 +1,148 @@
+import streamlit as st
+import random
 import os
+import smtplib
+from email.message import EmailMessage
 import json
-import random
-import streamlit as st
-import plotly.express as px
-
-# ----------------- Setup -----------------
-if not os.path.exists("accounts.json"):
-    with open("accounts.json", "w") as f:
-        json.dump({}, f)
-
-st.set_page_config(page_title="Baryashah Bank Pro", layout="wide")
-st.markdown("""
-<style>
-body {
-    background-color: #f5f7fa;
-}
-h1 {
-    color: #1f77b4;
-    text-align: center;
-    font-family: 'Arial', sans-serif;
-}
-.stButton>button {
-    background-color: #1f77b4;
-    color: white;
-    height: 3em;
-    width: 100%;
-    border-radius: 10px;
-    border: none;
-    font-size: 16px;
-    margin-top: 10px;
-}
-.stTextInput>div>input {
-    height: 2.5em;
-    border-radius: 10px;
-    border: 1px solid #ccc;
-    padding-left: 10px;
-}
-</style>
-""", unsafe_allow_html=True)
-
-st.markdown("<h1>🌟 Baryashah Bank Pro 🌟</h1>", unsafe_allow_html=True)
-st.write("---")
-
-# ----------------- Helper Functions -----------------
-def load_accounts():
-    with open("accounts.json","r") as f:
-        return json.load(f)
-
-def save_accounts(data):
-    with open("accounts.json","w") as f:
+
+# ---------- UTILITY FUNCTIONS ----------
+
+def load_data():
+    try:
+        with open('accounts.json', 'r') as f:
+            return json.load(f)
+    except:
+        return {}
+
+def save_data(data):
+    with open('accounts.json', 'w') as f:
         json.dump(data, f, indent=4)
 
-def send_otp_simulation(email):
-    otp = str(random.randint(1000,9999))
+def send_otp(email):
+    otp = str(random.randint(1000, 9999))
     st.session_state['otp'] = otp
-    st.info(f"✅ OTP sent to {email} (simulated): {otp}")
+
+    sender_email = os.environ['EMAIL_USER']
+    sender_password = os.environ['EMAIL_PASS']
+
+    msg = EmailMessage()
+    msg['Subject'] = "Your Baryashah Bank OTP"
+    msg['From'] = sender_email
+    msg['To'] = email
+    msg.set_content(f"Your OTP for Baryashah Bank is: {otp}")
+
+    with smtplib.SMTP_SSL('smtp.gmail.com', 465) as smtp:
+        smtp.login(sender_email, sender_password)
+        smtp.send_message(msg)
+
+    st.info(f"✅ OTP sent to {email}")
     return otp
 
-accounts = load_accounts()
+def verify_otp(user_input):
+    return user_input == st.session_state.get('otp')
+
+# ---------- APP UI ----------
+
+st.title("🌟 Baryashah Bank Pro Dashboard 🌟")
+st.write("Professional banking simulation. Login, withdraw, deposit & reset password securely.")
 
-# ----------------- Sidebar Menu -----------------
-menu = ["Create Account","Login","Reset Password","Recover Account Name"]
+data = load_data()
+
+menu = ["Create Account", "Login", "Reset Password"]
 choice = st.sidebar.selectbox("Menu", menu)
 
-# ----------------- Create Account -----------------
-if choice=="Create Account":
-    st.subheader("Create Account")
+# ---------- CREATE ACCOUNT ----------
+if choice == "Create Account":
+    st.header("📝 Create New Account")
     name = st.text_input("Full Name")
     email = st.text_input("Email")
     age = st.number_input("Age", min_value=1, max_value=120)
-    password = st.text_input("Password", type="password")
-    pin = st.text_input("4-digit PIN for withdrawals", type="password", max_chars=4)
-    
+    password = st.text_input("Create Password", type="password")
+
     if st.button("Create Account"):
         if age < 18:
-            st.error("❌ You must be 18+ to open an account!")
-        elif email in accounts:
-            st.error("❌ Account already exists with this email!")
-        elif len(pin)!=4 or not pin.isdigit():
-            st.error("❌ PIN must be 4 digits")
+            st.error("❌ Must be 18 or older to create an account.")
+        elif email in data:
+            st.error("❌ Account with this email already exists.")
         else:
-            accounts[email] = {
+            data[email] = {
                 "name": name,
                 "age": age,
                 "password": password,
-                "pin": pin,
                 "balance": 0,
                 "transactions": []
             }
-            save_accounts(accounts)
-            st.success(f"🎉 Account created successfully for {name}!")
-            st.balloons()
-
-# ----------------- Reset Password -----------------
-elif choice=="Reset Password":
-    st.subheader("Reset Password")
-    email = st.text_input("Enter Email")
-    if email in accounts:
-        if st.button("Send OTP"):
-            send_otp_simulation(email)
-        otp_input = st.text_input("Enter OTP")
-        new_pass = st.text_input("New Password", type="password")
-        if st.button("Reset Password Now"):
-            if 'otp' in st.session_state and otp_input == st.session_state['otp']:
-                accounts[email]['password'] = new_pass
-                save_accounts(accounts)
-                st.success("✅ Password reset successfully!")
-            else:
-                st.error("❌ Invalid OTP!")
-
-# ----------------- Recover Account Name -----------------
-elif choice=="Recover Account Name":
-    st.subheader("Recover Account Name")
-    email = st.text_input("Enter Email")
-    password = st.text_input("Enter Password", type="password")
-    if st.button("Recover"):
-        if email in accounts and accounts[email]['password']==password:
-            st.success(f"Your account name is: {accounts[email]['name']}")
-        else:
-            st.error("❌ Invalid credentials!")
+            save_data(data)
+            st.success(f"✅ Account created for {name}!")
 
-# ----------------- Login -----------------
-elif choice=="Login":
-    st.subheader("Login")
+# ---------- LOGIN ----------
+elif choice == "Login":
+    st.header("🔑 Login")
     email = st.text_input("Email")
     password = st.text_input("Password", type="password")
-    
-    if st.button("Login"):
-        if email in accounts and accounts[email]['password']==password:
-            st.info("OTP sent to your email for 2FA (simulated).")
-            otp = send_otp_simulation(email)
-            otp_input = st.text_input("Enter OTP for 2FA")
-            if st.button("Verify OTP"):
-                if otp_input == otp:
-                    st.success(f"Welcome {accounts[email]['name']}!")
-                    st.markdown(f"### 💰 Balance: {accounts[email]['balance']}")
-
-                    # ---------------- Transaction Section ----------------
-                    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 user_pin != accounts[email]['pin']:
-                                st.error("❌ Incorrect PIN!")
-                            elif withdraw_amt > accounts[email]['balance']:
-                                st.error("❌ Insufficient Balance!")
+
+    if st.button("Send OTP"):
+        if email not in data:
+            st.error("❌ Email not registered!")
+        elif data[email]["password"] != password:
+            st.error("❌ Wrong password!")
+        else:
+            send_otp(email)
+
+    otp_input = st.text_input("Enter OTP")
+    if st.button("Verify OTP"):
+        if verify_otp(otp_input):
+            st.success(f"✅ Welcome {data[email]['name']}!")
+            # Dashboard
+            st.subheader("💰 Account Dashboard")
+            st.metric("Balance", f"${data[email]['balance']}")
+            col1, col2 = st.columns(2)
+            with col1:
+                deposit_amount = st.number_input("Deposit Amount", min_value=1)
+                if st.button("Deposit"):
+                    data[email]['balance'] += deposit_amount
+                    data[email]['transactions'].append(f"Deposit: +${deposit_amount}")
+                    save_data(data)
+                    st.success(f"✅ Deposited ${deposit_amount} successfully!")
+            with col2:
+                withdraw_amount = st.number_input("Withdraw Amount", min_value=1)
+                if st.button("Withdraw"):
+                    # Send OTP before withdrawal
+                    send_otp(email)
+                    otp_withdraw = st.text_input("Enter OTP to withdraw")
+                    if st.button("Confirm Withdrawal"):
+                        if verify_otp(otp_withdraw):
+                            if withdraw_amount <= data[email]['balance']:
+                                data[email]['balance'] -= withdraw_amount
+                                data[email]['transactions'].append(f"Withdraw: -${withdraw_amount}")
+                                save_data(data)
+                                st.success(f"✅ Withdrawn ${withdraw_amount} successfully!")
                             else:
-                                otp_w = send_otp_simulation(email)
-                                otp_input_w = st.text_input("Enter OTP for Withdrawal")
-                                if st.button("Confirm Withdraw"):
-                                    if otp_input_w == otp_w:
-                                        accounts[email]['balance'] -= withdraw_amt
-                                        accounts[email]['transactions'].append({"type":"Withdraw","amount":withdraw_amt})
-                                        save_accounts(accounts)
-                                        st.success(f"✅ Withdrawal Successful! New Balance: {accounts[email]['balance']}")
-                                        st.balloons()
-                                    else:
-                                        st.error("❌ Invalid OTP for withdrawal!")
-
-                    with col2:
-                        deposit_amt = st.number_input("Deposit Amount", min_value=0)
-                        if st.button("Deposit", key="deposit"):
-                            accounts[email]['balance'] += deposit_amt
-                            accounts[email]['transactions'].append({"type":"Deposit","amount":deposit_amt})
-                            save_accounts(accounts)
-                            st.success(f"✅ Deposit Successful! New Balance: {accounts[email]['balance']}")
-                            st.balloons()
-
-                    # ---------------- Transaction Chart ----------------
-                    if accounts[email]['transactions']:
-                        df = accounts[email]['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.error("❌ 2FA OTP Invalid!")
+                                st.error("❌ Insufficient Balance!")
+                        else:
+                            st.error("❌ Wrong OTP!")
+            # Show transactions
+            st.subheader("📜 Transaction History")
+            for t in data[email]['transactions']:
+                st.write(t)
+        else:
+            st.error("❌ OTP incorrect!")
+
+# ---------- PASSWORD RESET ----------
+elif choice == "Reset Password":
+    st.header("🔄 Reset Password")
+    email = st.text_input("Registered Email")
+    if st.button("Send OTP for Reset"):
+        if email not in data:
+            st.error("❌ Email not registered!")
+        else:
+            send_otp(email)
+    otp_input = st.text_input("Enter OTP received")
+    new_pass = st.text_input("Enter New Password", type="password")
+    if st.button("Reset Password"):
+        if verify_otp(otp_input):
+            data[email]['password'] = new_pass
+            save_data(data)
+            st.success("✅ Password reset successfully!")
         else:
-            st.error("❌ Login failed!")
+            st.error("❌ OTP incorrect!")