Added logout functionality.

app.py

@@ -158,6 +158,12 @@ def signup():
     else:
         st.warning("Passwords do not match.")
 
+def logout():
+    for key in list(st.session_state.keys()):
+        del st.session_state[key]
+    st.session_state.logged_out = True
+    st.rerun()
+
 def marketplace(username):
     # Custom CSS for better aesthetics
     st.markdown("""
@@ -275,23 +281,23 @@ def marketplace(username):
     st.markdown("Created with ❤️ by Harshit S | © 2024 PDF Reader App")
 
 def main():
-  st.title("Choose the suitable category:")
+    if 'logged_in' not in st.session_state:
+        st.session_state.logged_in = False
 
-  if "logged_in" not in st.session_state:
-    st.session_state.logged_in = False
+    if st.session_state.logged_in:
+        marketplace(st.session_state.username)
+    else:
+        st.title("Welcome to AI Chat")
+        choice = st.selectbox("Login/Signup", ["Login", "Sign Up"])
+        
+        if choice == "Login":
+            login()
+        else:
+            signup()
 
-  if st.session_state.logged_in:
-    marketplace(st.session_state.username)
-  else:
-    menu = ["Login", "Sign Up"]
-    choice = st.sidebar.selectbox("Menu", menu)
-
-    if choice == "Login":
-      call = login()
-      if call:
-        main()
-    elif choice == "Sign Up":
-      signup()
+    if st.session_state.get('logged_out', False):
+        st.info("You have been logged out successfully.")
+        st.session_state.logged_out = False
 
 if __name__ == "__main__":
   print(peek())