Update google_function.py

google_function.py

@@ -11,6 +11,7 @@ from email.message import EmailMessage
 from bs4 import BeautifulSoup
 import webbrowser
 import datetime
+from streamlit_javascript import st_javascript
 import streamlit as st
 
 SCOPES = ["https://www.googleapis.com/auth/gmail.compose", 
@@ -21,6 +22,8 @@ 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:
@@ -32,7 +35,6 @@ def converti_email_txt(body):
     return body
 
 def leggi_gmail(max_results=10):
-    creds=connetti_google()
     links = []
     service = build("gmail", "v1", credentials=creds)
     results = service.users().messages().list(userId="me", labelIds=["INBOX"], q="is:unread",  maxResults=max_results).execute()
@@ -76,7 +78,6 @@ def leggi_gmail(max_results=10):
 #---------------------------------------------------------- SCRITTURA BOZZA EMAIL ---------------------------------------------------------            
 def scrivi_bozza_gmail(testo):
     draft_url = ''
-    creds=connetti_google()
     try:
         service = build("gmail", "v1", credentials=creds)
         message = EmailMessage()
@@ -97,7 +98,6 @@ def scrivi_bozza_gmail(testo):
 
 #---------------------------------------------------------- LEGGI GOOGLE CALENDAR  ---------------------------------------------------------            
 def leggi_calendario_google(max_results=10):
-    creds=connetti_google()
     try:
         service = build("calendar", "v3", credentials=creds)
         calendar_list_result = service.calendarList().list().execute()
@@ -133,29 +133,41 @@ def leggi_calendario_google(max_results=10):
     return descrizione_eventi, links
 
 #---------------------------------------------------------- CONNESSIONE ACCOUNT GOOGLE ---------------------------------------------------------
+def local_storage_get(key):
+    return st_javascript(f"localStorage.getItem('{key}');")
+
+def local_storage_set(key, value):
+    return st_javascript(f"localStorage.setItem('{key}', '{value}');")
+
 def connetti_google():
     load_dotenv()
     json_variable_str = os.getenv("JSON_GOOGLE")
-    print(json_variable_str)
     with open("./credentials.json", "w") as f:
         f.write(json_variable_str)
-    if os.path.exists("./credentials.json"):
-        print('ESISTE')
-    else:
-        print('NON ESISTE')
-    creds = None
-    st.write("Welcome to My App!")
-    auth_code = st.query_params.get("code")
     flow = InstalledAppFlow.from_client_secrets_file("./credentials.json", SCOPES, redirect_uri='https://matteoscript-streamlitchat-new.hf.space')
-    if auth_code:
-        flow.fetch_token(code=auth_code)
+    token = local_storage_get("token")
+    print('------------1------------------')
+    print(token)
+    if token and token != '':
+        print('------------2------------------')
+        local_storage_set("token", '')    
+        flow.fetch_token(code=token)
         creds = flow.credentials
+        print(creds)
         st.write("Login Done")
+        st.session_state.login_effettuato = True
     else:
-        authorization_url, state = flow.authorization_url(include_granted_scopes="true",)
-        print(authorization_url)
-        st.markdown(f'<a href="{authorization_url}">Click here to sign in with Google</a>', unsafe_allow_html=True)
-    return creds
+        print('------------3------------------')
+        auth_code = st.query_params.get("code")
+        if auth_code:
+            local_storage_set("token", auth_code)
+            st_javascript('window.location.href = "https://matteoscript-streamlitchat-new.hf.space";')
+        else:
+            print('------------4------------------') 
+            st.write("")
+            authorization_url, state = flow.authorization_url(include_granted_scopes="true",)
+            st.write('## Login Bonsi A.I.')
+            st.markdown(f'<a href="{authorization_url}" target="_self">Effettua Login Google</a>', unsafe_allow_html=True)
 
 def connetti_google_2(): 
     load_dotenv()