added login and logout

app.py

@@ -1,18 +1,52 @@
 import streamlit as st
+import time
 
 #importing the pages
 from resume_screener import ResumeScreener
 from user_documentation import UserDocumentation
 from home import Home
+from login import login
 
 UserDocPage = st.Page(UserDocumentation, title="User Documentation", icon=":material/help:")
 ResumeScreenerPage = st.Page(ResumeScreener, title="Resume Screener", icon=":material/recent_actors:")
 HomePage = st.Page(Home, title="Home", icon=":material/home:")
+LoginPage = st.Page(login, title="Login", icon=":material/login:")
 
+if 'USER_DOC_PAGE' not in st.session_state:
+    st.session_state.USER_DOC_PAGE = UserDocPage
+
+if 'RESUME_SCREENER_PAGE' not in st.session_state:
+    st.session_state.RESUME_SCREENER_PAGE = ResumeScreenerPage
+
+if 'HOME_PAGE' not in st.session_state:
+    st.session_state.HOME_PAGE = HomePage
+
+if 'CURRENT_PAGE' not in st.session_state:
+    st.session_state.CURRENT_PAGE = None
 
 if "LOGGED_IN" not in st.session_state:
     st.session_state.LOGGED_IN = False
 
+def logout():
+    @st.dialog("Are you sure you want to log out?")
+
+    def chose():
+        col1, col2 = st.columns(spec=[1,1])
+
+        with col1:
+            if st.button("No", use_container_width=True):
+                st.switch_page(st.session_state.CURRENT_PAGE)
+
+        with col2:
+            if st.button("Yes", use_container_width=True):
+                st.session_state.clear()
+                st.rerun()
+
+    chose()
+
+
+logout_page = st.Page(logout, title="Log out", icon=":material/logout:")
+
 if st.session_state.LOGGED_IN:
     pages = {
         "Dashboard": [
@@ -21,6 +55,9 @@ if st.session_state.LOGGED_IN:
         ],
         "Resources": [
             UserDocPage,
+        ],
+        "Account": [
+            logout_page
         ]
     }
 
@@ -28,45 +65,4 @@ if st.session_state.LOGGED_IN:
     pg.run()
 
 else:
-    col1, col2, col3 = st.columns(spec=[1, 3, 1])
-
-    with col2:
-        st.title("Login")
-        username = st.text_input("Email Address", value="johndoe@hr.company.com")
-        password = st.text_input("Password", type="password")
-
-        if st.button("Login"):
-
-            class UnregisteredUser(Exception):
-                """
-                Represents an exception when the user is not registered.
-                """
-
-                def __init__(self, message="User not registered"):
-                    self.message = message
-                    super().__init__(self.message)
-
-            class IncompleteDetails(Exception):
-                """
-                Represents an exception when the user's details are incomplete.
-                """
-
-                def __init__(self, message="Incomplete details"):
-                    self.message = message
-                    super().__init__(self.message)
-
-            #imaginary db calls for existing user, if user is in imaginary db, user is logged in
-            try:
-                if username == "johndoe@hr.company.com" and password == "password":
-                    st.session_state.LOGGED_IN = True
-                    st.rerun()
-
-                if username == "" or password == "":
-                    raise IncompleteDetails
-
-                raise UnregisteredUser
-
-            except UnregisteredUser as e:
-                st.error(e)
-            except IncompleteDetails as e:
-                st.error(e)
+    login()