Update app.py

app.py

@@ -116,15 +116,22 @@ def login_page():
             else:
                 st.error("Please fill in all fields.")
 
+def logout_user():
+    """Logs out the user safely by clearing session state and navigating to the login page."""
+    st.session_state.authenticated = False
+    st.session_state.username = None
+    st.session_state.page = "login"
+    st.session_state.current_project = None  # Clear project data
+    st.session_state.project_uploaded = False  # Reset project uploaded flag
+
 
 def project_staging_page():
     # Sidebar with logout button
     st.sidebar.image("SimplifAI Logo Long.jpeg", use_container_width=True)
     st.sidebar.title(f"Hello, {st.session_state.username}!")
     if st.sidebar.button("Log Out"):
-        st.session_state.authenticated = False
-        st.session_state.username = None
-        st.session_state.page = "login"
+        logout_user()
+        st.rerun()
 
     # User's folder for projects
     user_folder = os.path.join("user_projects", st.session_state.username)
@@ -437,9 +444,7 @@ def generate_documentation_page():
         st.rerun()
 
     if st.sidebar.button("Log Out"):
-        st.session_state.authenticated = False
-        st.session_state.username = None
-        st.session_state.page = "login"
+        logout_user()
         st.rerun()
 
     st.subheader(f"Generate Documentation for {st.session_state.current_project}")
@@ -594,9 +599,7 @@ def saved_documentation_page():
         st.session_state.page = "project_view"
         st.rerun()
     if st.sidebar.button("Log Out"):
-        st.session_state.authenticated = False
-        st.session_state.username = None
-        st.session_state.page = "login"
+        logout_user()
         st.rerun()
 
     st.subheader(f"Saved Documentation for {st.session_state.current_project}")
@@ -641,9 +644,7 @@ def project_view_page():
         st.session_state.page = "project_staging"
         st.rerun()
     if st.sidebar.button("Log Out"):
-        st.session_state.authenticated = False
-        st.session_state.username = None
-        st.session_state.page = "login"
+        logout_user()
         st.rerun()
 
     # Main content for project page