app.py

@@ -179,7 +179,8 @@ def main():
                 st.error('Student ID not found. Please try again.')
 
         if st.button("Logout"):
-            del st.session_state['credentials']
+            for key in list(st.session_state.keys()):
+                del st.session_state[key]
             st.rerun()
 
 # Function to handle OAuth callback
@@ -189,16 +190,23 @@ def handle_callback():
         flow.fetch_token(code=st.query_params["code"])
         credentials = flow.credentials
         st.session_state['credentials'] = credentials.to_json()
+        st.session_state['authenticated'] = True
         logging.debug("Token fetch successful")
         st.success("Authentication successful!")
-        return credentials
+        time.sleep(2)  # Give user time to see the success message
+        st.rerun()  # Rerun the app to clear the URL parameters
     except Exception as e:
         logging.error(f"Error during authentication: {str(e)}")
         st.error(f"Authentication failed: {str(e)}")
+        st.session_state['authenticated'] = False
+        if 'credentials' in st.session_state:
+            del st.session_state['credentials']
 
 if __name__ == '__main__':
     logging.debug("Starting the application")
     if 'code' in st.query_params:
         logging.debug("Authorization code found in query parameters")
         handle_callback()
+    elif st.session_state.get('authenticated', False):
+        st.success("You are already authenticated.")
     main()