app.py

@@ -179,8 +179,7 @@ def main():
                 st.error('Student ID not found. Please try again.')
 
         if st.button("Logout"):
-            for key in list(st.session_state.keys()):
-                del st.session_state[key]
+            del st.session_state['credentials']
             st.rerun()
 
 # Function to handle OAuth callback
@@ -190,23 +189,17 @@ 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!")
-        time.sleep(2)  # Give user time to see the success message
-        st.rerun()  # Rerun the app to clear the URL parameters
+        return credentials
     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']
+        pass
+        # logging.error(f"Error during authentication: {str(e)}")
+        # st.error(f"Authentication failed: {str(e)}")
 
 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()