Update google_function.py

google_function.py

@@ -140,11 +140,12 @@ def local_storage_set(key, value):
     return st_javascript(f"localStorage.setItem('{key}', '{value}');")
 
 def connetti_google():
+    redirect = 'https://matteoscript-streamlitchat-new.hf.space'
     load_dotenv()
     json_variable_str = os.getenv("JSON_GOOGLE")
     with open("./credentials.json", "w") as f:
         f.write(json_variable_str)
-    flow = InstalledAppFlow.from_client_secrets_file("./credentials.json", SCOPES, redirect_uri='https://matteoscript-streamlitchat-new.hf.space')
+    flow = InstalledAppFlow.from_client_secrets_file("./credentials.json", SCOPES, redirect_uri=redirect)
     token = local_storage_get("token")
     print('------------1------------------')
     print(token)
@@ -161,7 +162,8 @@ def connetti_google():
         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";')
+            nav_script = """<meta http-equiv="refresh" content="0; url='%s'">""" % (redirect)
+            st.write(nav_script, unsafe_allow_html=True)
         else:
             print('------------4------------------') 
             st.write("")