Spaces:

LinhVuu
/

Banking-Advanced-Authentication-System

Sleeping

Banking-Advanced-Authentication-System / app.py

LinhVuu

updated

8ef54e5 about 1 month ago

No virus

5.15 kB

	# Import libraries
	import os
	import streamlit as st
	from PIL import Image
	from streamlit_extras.switch_page_button import switch_page
	from baam_functions import *
	from datetime import datetime
	from bokeh.models.widgets import Button
	from bokeh.models import CustomJS
	from streamlit_bokeh_events import streamlit_bokeh_events

	# Set direction as current folder
	sourceFileDir = os.path.dirname(os.path.abspath(__file__))
	os.chdir(sourceFileDir)

	logo = Image.open('img/logo.png')
	st.set_page_config(page_title = "BAAM", page_icon = logo)

	# Initialise variables to pass between pages
	st.session_state['user_dict'] = {}
	st.session_state['verification'] = False
	st.session_state['location'] = ""
	st.session_state['raw_location'] = ""
	st.session_state['username'] = ""
	# st.session_state['typing_speed'] = 0

	def main():
	"""Banking Advanced Authentication System"""

	# # Note that API key's running out of budget
	# contact_url = "https://www.linkedin.com/in/linhvuu"
	# st.write("I am running out of energy. Please contact [my assistant](%s) to wake me up." % contact_url)

	menu = ["Home", "SignUp", "Login"]
	choice = st.sidebar.selectbox("Menu", menu)

	if choice == "Home":
	st.subheader("Banking Advanced Authentication Module")
	st.image("img/home.png")
	st.write("Please select one action from the Menu on the side bar. Thank you.")
	st.image("img/menu.png")

	elif choice == "SignUp":
	st.subheader("Create New Account")
	new_user = st.text_input("Username")
	new_password = st.text_input("Password",type='password')

	if st.button("Signup"):
	create_user_table()
	add_user_data(new_user,make_hashes(new_password))
	st.success("You have successfully created a valid Account")
	st.info("Go to Login Menu to login")

	elif choice == "Login":

	# Header of the page
	col1, col2, col3 = st.columns(3)
	with col1:
	st.write(' ')

	# Logo
	with col2:
	st.image("img/Standard_Chartered.png", width=175)
	with col3:
	st.write(' ')

	# Login form
	st.subheader("Login")
	username = st.text_input("Username")
	password = st.text_input("Password",type='password')

	is_tester = st.checkbox('Login to test')
	login_button = Button(label="Login")

	# Collect location when login button is clicked
	login_button.js_on_event("button_click", CustomJS(code="""
	navigator.geolocation.getCurrentPosition(
	(loc) => {
	document.dispatchEvent(new CustomEvent("GET_LOCATION", {detail: {lat: loc.coords.latitude, lon: loc.coords.longitude}}))
	}
	)
	"""))

	# Get location
	location = streamlit_bokeh_events(
	login_button,
	events="GET_LOCATION",
	key="get_location",
	refresh_on_update=False,
	override_height=75,
	debounce_time=0)

	# If can get location
	if location:

	# Hash password & verify password
	hashed_pswd = make_hashes(password)
	result = login_user(username,check_hashes(password, hashed_pswd))

	# When the username and password are correct
	if result:

	# time_start = st.session_state['time_start']

	# Stop counting time after the user finished entering username, password and clicked login button
	login_time = datetime.now()

	# # *** Calculate the login duration in seconds (the time it takes user to enter username, password and click login button)
	# login_duration = login_time - time_start
	# login_duration_seconds = login_duration.total_seconds()

	# # Calculate typing speed (characters per minute)
	# typing_speed = (len(username) + len(password)) * 60 / login_duration_seconds

	# Launch app for end user
	if not(is_tester):

	# Collect user's information when logging in
	#user_dict, st.session_state['location'] = collect_data(username, location, login_time, typing_speed)
	user_dict, st.session_state['location'] = collect_data(username, location, login_time)

	# Store user information to pass to other pages
	st.session_state['user_dict'] = user_dict
	verification = verify_user(user_dict)

	# THAO LE CODE:
	# weight edit in logic_tl
	weight = {'device_uuid': 40, 'mac_address': 40, 'device_name': 30.0, 'device_model': 20.0, \
	'device_vendor': 4.0, 'ip_v4': 30, 'isp_name': 15.0, 'ip_country': 3.0, 'suburd': 30, 'district': 22.5, \
	'city': 15.0, 'country': 3.0}
	# ThaoLe note:
	# st.write(f"print the user_dict {user_dict}")
	# st.write(f'type of user_dict {type(user_dict)}')
	# end not

	st.session_state['verification'] = verification

	# Add to the login database after successful verification
	if verification:
	add_login_data(user_dict)

	# Open Transfer Money Page
	switch_page("Transfer Money 💸")

	# Launch Testing Tool
	else:
	st.session_state['raw_location'] = location
	st.session_state['username'] = username
	# st.session_state['typing_speed'] = typing_speed
	switch_page("SelectUseCase")

	else:
	st.warning("Incorrect Username/Password")

	# Start counting time
	st.session_state['time_start'] = datetime.now()

	# #To export database to csv file (can be commented out if not needed)
	# export_csv()

	if __name__ == '__main__':
	main()