Update app.py

app.py

@@ -87,7 +87,9 @@ def authenticate_gmail(credentials_file):
             if creds and creds.valid:
                 st.session_state.creds = creds
                 st.session_state.authenticated = True
-                st.success("✅ Authentication successful!")
+                if not st.session_state.candidates_message_shown:
+                    st.success("✅ Authentication successful!")
+                    st.session_state.candidates_message_shown = True
                 return creds
         except Exception as e:
             st.error(f"❌ Invalid token.json file: {e}")
@@ -97,7 +99,9 @@ def authenticate_gmail(credentials_file):
             creds.refresh(Request())
             st.session_state.creds = creds
             st.session_state.authenticated = True
-            st.success("✅ Authentication successful!")
+            if not st.session_state.candidates_message_shown:
+                st.success("✅ Authentication successful!")
+                st.session_state.candidates_message_shown = True
             with open('token.json', 'w') as token_file:
                 token_file.write(creds.to_json())
             return creds
@@ -112,13 +116,21 @@ def authenticate_gmail(credentials_file):
 
 def submit_auth_code():
     try:
+        # Attempt to fetch the token using the provided authorization code
         st.session_state.flow.fetch_token(code=st.session_state.auth_code)
         st.session_state.creds = st.session_state.flow.credentials
-        st.session_state.authenticated = True
+        
+        # Attempt to write the credentials to token.json
         with open('token.json', 'w') as token_file:
-            st.session_state.creds.to_json(token_file)
+            token_json = st.session_state.creds.to_json()
+            token_file.write(token_json)
+        
+        # If writing is successful, update the session state
+        st.session_state.authenticated = True
         st.success("✅ Authentication successful!")
     except Exception as e:
+        # If any error occurs, ensure the authenticated flag is not set
+        st.session_state.authenticated = False
         st.error(f"❌ Error during authentication: {e}")
 
 # ===============================