Update google_function.py

google_function.py

@@ -23,8 +23,6 @@ SCOPES = ["https://www.googleapis.com/auth/gmail.compose",
           "https://www.googleapis.com/auth/forms.body",
           "https://www.googleapis.com/auth/spreadsheets.readonly"]
 
-creds = None 
-
 #---------------------------------------------------------- LETTURA EMAIL ---------------------------------------------------------
 def converti_email_txt(body):
     try:
@@ -37,7 +35,7 @@ def converti_email_txt(body):
 
 def leggi_gmail(max_results=10):
     links = []
-    service = build("gmail", "v1", credentials=creds)
+    service = build("gmail", "v1", credentials=st.session_state.creds)
     results = service.users().messages().list(userId="me", labelIds=["INBOX"], q="is:unread",  maxResults=max_results).execute()
     messages = results.get("messages", [])
     testo_email = ''
@@ -80,7 +78,7 @@ def leggi_gmail(max_results=10):
 def scrivi_bozza_gmail(testo):
     draft_url = ''
     try:
-        service = build("gmail", "v1", credentials=creds)
+        service = build("gmail", "v1", credentials=st.session_state.creds)
         message = EmailMessage()
         message.set_content(testo)
         message["To"] = "gduser1@workspacesamples.dev"
@@ -100,7 +98,7 @@ def scrivi_bozza_gmail(testo):
 #---------------------------------------------------------- LEGGI GOOGLE CALENDAR  ---------------------------------------------------------            
 def leggi_calendario_google(max_results=10):
     try:
-        service = build("calendar", "v3", credentials=creds)
+        service = build("calendar", "v3", credentials=st.session_state.creds)
         calendar_list_result = service.calendarList().list().execute()
         calendars = calendar_list_result.get('items', [])
         descrizione_eventi = ''
@@ -154,9 +152,8 @@ def connetti_google():
         print('------------2------------------')
         #local_storage_set("token", '')    
         flow.fetch_token(code=token)
-        creds = flow.credentials
-        print(creds)
-        st.write("Login Done")
+        st.session_state.creds = flow.credentials
+        print(st.session_state.creds)
         st.session_state.login_effettuato = True
     else:
         print('------------3------------------')