Update app.py

app.py

@@ -419,56 +419,56 @@ def main():
             st.warning(
                 'Plesase Process RFP Details to access this feature', icon="⚠️")
 
-if __name__ == "__main__":
-    main() 
-
 # if __name__ == "__main__":
-#     print(localStorage.getItem('email'))
-#     if localStorage.getItem('email'):
-#         main()     
-#     else:
-#         st.title("SIGN UP TO ACCESS THE APP!")
-#         AUTHORIZE_ENDPOINT = "https://accounts.google.com/o/oauth2/v2/auth"
-#         TOKEN_ENDPOINT = "https://oauth2.googleapis.com/token"
-#         REVOKE_ENDPOINT = "https://oauth2.googleapis.com/revoke"
-
-#         # create a button to start the OAuth2 flow
-#         oauth2 = OAuth2Component(
-#             CLIENT_ID,
-#             CLIENT_SECRET,
-#             AUTHORIZE_ENDPOINT,
-#             TOKEN_ENDPOINT,
-#             TOKEN_ENDPOINT,
-#             REVOKE_ENDPOINT,
-#         )
-#         result = oauth2.authorize_button(
-#             name="Continue with Google",
-#             icon="https://www.google.com.tw/favicon.ico",
-#             redirect_uri="https://huggingface.co/spaces/coffeebeans-ai/rfp_to_story",
-#             scope="openid email profile",
-#             key="google",
-#             height=600,
-#             width=500,
-#             extras_params={"prompt": "consent", "access_type": "offline"},
-#             use_container_width=False,
-#             pkce="S256",
-#         )
-
-#         if result:
-#             # decode the id_token jwt and get the user's email address
-#             id_token = result["token"]["id_token"]
-#             # verify the signature is an optional step for security
-#             payload = id_token.split(".")[1]
-#             # add padding to the payload if needed
-#             payload += "=" * (-len(payload) % 4)
-#             payload = json.loads(base64.b64decode(payload))
-#             email = payload["email"]
-#             st.session_state["auth"] = email
-#             st.session_state["token"] = result["token"]
-#             is_valid = validate_email(email)
-#             if is_valid:
-#                 localStorage.setItem("email",email)
-#                 st.rerun()
-#             else:
-#                 st.warning("Unauthorized. you are not authorized to access this app.")
+#     main() 
+
+if __name__ == "__main__":
+    print(localStorage.getItem('email'))
+    if localStorage.getItem('email'):
+        main()     
+    else:
+        st.title("SIGN UP TO ACCESS THE APP!")
+        AUTHORIZE_ENDPOINT = "https://accounts.google.com/o/oauth2/v2/auth"
+        TOKEN_ENDPOINT = "https://oauth2.googleapis.com/token"
+        REVOKE_ENDPOINT = "https://oauth2.googleapis.com/revoke"
+
+        # create a button to start the OAuth2 flow
+        oauth2 = OAuth2Component(
+            CLIENT_ID,
+            CLIENT_SECRET,
+            AUTHORIZE_ENDPOINT,
+            TOKEN_ENDPOINT,
+            TOKEN_ENDPOINT,
+            REVOKE_ENDPOINT,
+        )
+        result = oauth2.authorize_button(
+            name="Continue with Google",
+            icon="https://www.google.com.tw/favicon.ico",
+            redirect_uri="https://huggingface.co/spaces/coffeebeans-ai/rfp_to_story",
+            scope="openid email profile",
+            key="google",
+            height=600,
+            width=500,
+            extras_params={"prompt": "consent", "access_type": "offline"},
+            use_container_width=False,
+            pkce="S256",
+        )
+
+        if result:
+            # decode the id_token jwt and get the user's email address
+            id_token = result["token"]["id_token"]
+            # verify the signature is an optional step for security
+            payload = id_token.split(".")[1]
+            # add padding to the payload if needed
+            payload += "=" * (-len(payload) % 4)
+            payload = json.loads(base64.b64decode(payload))
+            email = payload["email"]
+            st.session_state["auth"] = email
+            st.session_state["token"] = result["token"]
+            is_valid = validate_email(email)
+            if is_valid:
+                localStorage.setItem("email",email)
+                st.rerun()
+            else:
+                st.warning("Unauthorized. you are not authorized to access this app.")