add button

DevXT-LLC · Jun 11, 2024 · b8d6b9b · b8d6b9b
1 parent e3c737f
commit b8d6b9b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/GoogleSSO.py b/GoogleSSO.py
@@ -8,8 +8,8 @@
 class GoogleSSO:
     def __init__(
         self,
-        access_token,  # Will come from db stored for user
-        refresh_token,  # Will come from db stored for user
+        access_token,  # Comes from Google OAuth and stored in db
+        refresh_token,  # Comes from Google OAuth and stored in db
     ):
         self.refresh_token = refresh_token
         self.access_token = access_token

diff --git a/components/Auth.py b/components/Auth.py
@@ -139,6 +139,8 @@ def get_user():
                 email = st.text_input("Email")
                 otp = st.text_input("MFA Token")
                 login_button = st.form_submit_button("Login")
+                st.divider()
+                google_sso_button(redirect_uri=auth_uri)
                 if login_button:
                     auth_response = requests.post(
                         f"{auth_uri}/v1/login",