Update app.py

app.py

@@ -4,7 +4,7 @@ import streamlit as st
 
 # Set page config
 st.set_page_config(
-    page_title="TransactPro: The Ultimate Financial Platform",
+    page_title="Quantum Bank: Invented by James Burvel O'Callaghan III",
     page_icon="💳",
     layout="wide",
 )
@@ -130,31 +130,66 @@ if get_env_var('PLAID_CLIENT_ID') and get_env_var('PLAID_SECRET'):
 else:
     plaid_client = None
 
+# Initialize user database (In-memory for demonstration purposes)
+if 'user_database' not in st.session_state:
+    st.session_state['user_database'] = {}
+
 # JWT Authentication
 def authenticate_user():
     st.sidebar.title("User Authentication")
-    username = st.sidebar.text_input("Username")
-    password = st.sidebar.text_input("Password", type="password")
-    if st.sidebar.button("Login"):
-        # Hash the password (Use a secure method in production)
-        hashed_password = hashlib.sha256(password.encode()).hexdigest()
-        # Check credentials (Replace with real authentication)
-        if username == 'user' and hashed_password == hashlib.sha256("password".encode()).hexdigest():
-            token = jwt.encode({'user': username}, 'secret', algorithm='HS256')
-            st.session_state['auth_token'] = token
-            st.success("Logged in successfully!")
-        else:
-            st.error("Invalid username or password")
+    auth_option = st.sidebar.radio("Select an option", ["Login", "Create Account"])
+
+    if auth_option == "Login":
+        username = st.sidebar.text_input("Username")
+        password = st.sidebar.text_input("Password", type="password")
+        if st.sidebar.button("Login"):
+            # Hash the password
+            hashed_password = hashlib.sha256(password.encode()).hexdigest()
+            # Check credentials
+            user_db = st.session_state['user_database']
+            if username in user_db and user_db[username]['password'] == hashed_password:
+                token = jwt.encode({'user': username}, 'secret', algorithm='HS256')
+                st.session_state['auth_token'] = token
+                st.success("Logged in successfully!")
+            else:
+                st.error("Invalid username or password")
+
+    elif auth_option == "Create Account":
+        username = st.sidebar.text_input("Choose a Username")
+        password = st.sidebar.text_input("Choose a Password", type="password")
+        confirm_password = st.sidebar.text_input("Confirm Password", type="password")
+        if st.sidebar.button("Create Account"):
+            if password != confirm_password:
+                st.error("Passwords do not match")
+            elif username in st.session_state['user_database']:
+                st.error("Username already exists")
+            else:
+                # Hash the password
+                hashed_password = hashlib.sha256(password.encode()).hexdigest()
+                # Store user in the database
+                st.session_state['user_database'][username] = {'password': hashed_password}
+                st.success("Account created successfully! Please login.")
 
 if 'auth_token' not in st.session_state:
     authenticate_user()
 else:
     # Proceed with the application
 
+    # Decode JWT to get username
+    token = st.session_state['auth_token']
+    decoded_token = jwt.decode(token, 'secret', algorithms=['HS256'])
+    username = decoded_token['user']
+
     # Title of the app
-    st.title("TransactPro: The Ultimate Financial Platform")
+    st.title(f"Quantum Bank: Welcome, {username}")
     st.sidebar.header("Navigation")
 
+    # Logout button
+    if st.sidebar.button("Logout"):
+        del st.session_state['auth_token']
+        st.success("Logged out successfully")
+        st.experimental_rerun()
+
     # Main application navigation
     service_choice = st.sidebar.selectbox("Choose a Service", [
         "Dashboard",