Update app.py

app.py

@@ -57,7 +57,7 @@ def count_tokens(text):
     return len(text.split())
 
 # ===============================
-# 3. Gmail Authentication Functions
+# 3. Gmail Authentication Functions (Updated)
 # ===============================
 def reset_session_state():
     st.session_state.authenticated = False
@@ -78,6 +78,7 @@ def reset_session_state():
             os.remove(filename)
 
 def authenticate_gmail(credentials_file):
+    SCOPES = ['https://www.googleapis.com/auth/gmail.readonly']
     creds = None
     if os.path.exists('token.json'):
         try:
@@ -86,9 +87,7 @@ def authenticate_gmail(credentials_file):
             if creds and creds.valid:
                 st.session_state.creds = creds
                 st.session_state.authenticated = True
-                if not st.session_state.candidates_message_shown:
-                    st.success("✅ Authentication successful!")
-                    st.session_state.candidates_message_shown = True
+                st.success("✅ Authentication successful!")
                 return creds
         except Exception as e:
             st.error(f"❌ Invalid token.json file: {e}")
@@ -98,9 +97,7 @@ def authenticate_gmail(credentials_file):
             creds.refresh(Request())
             st.session_state.creds = creds
             st.session_state.authenticated = True
-            if not st.session_state.candidates_message_shown:
-                st.success("✅ Authentication successful!")
-                st.session_state.candidates_message_shown = True
+            st.success("✅ Authentication successful!")
             with open('token.json', 'w') as token_file:
                 token_file.write(creds.to_json())
             return creds
@@ -118,11 +115,9 @@ def submit_auth_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
-        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(st.session_state.creds.to_json())
+            st.session_state.creds.to_json(token_file)
+        st.success("✅ Authentication successful!")
     except Exception as e:
         st.error(f"❌ Error during authentication: {e}")