Upload 6 files

.env_dummy

@@ -0,0 +1,12 @@
+CLIENT_ID = "537582539411-30v3dcg5hila3d6gaihn9ld3ue4tq4d4.apps.googleusercontent.com"
+CLIENT_SECRET = "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"
+REDIRECT_URI = "http://localhost:8501"
+host = "host"
+port = port_number
+database = "database_name"
+user = "root"
+password = "your_pwd"
+azure_event_key = "xxxxxxx"
+azure_event_endpoint = "xxxxxxxxx"
+azure_servicebus_conection_str = "XXXXX"
+azure_servicebus_queue_name = "XXXXXX"

Makefile

@@ -0,0 +1,16 @@
+VENV = .venv
+PYTHON = $(VENV)/bin/python3
+PIP = $(VENV)/bin/pip
+STREAMLIT = $(VENV)/bin/streamlit
+run: $(VENV)/bin/activate
+	$(STREAMLIT) run app.py 
+
+
+$(VENV)/bin/activate: requirements.txt
+	python3 -m venv $(VENV)
+	$(PIP) install -r requirements.txt
+
+
+clean:
+	rm -rf __pycache__
+	rm -rf $(VENV)

__init__.py

@@ -0,0 +1,112 @@
+import streamlit as st
+import asyncio
+from httpx_oauth.clients.google import GoogleOAuth2
+
+__version__ = "0.1"
+
+
+async def write_authorization_url(client, redirect_uri):
+    authorization_url = await client.get_authorization_url(
+        redirect_uri,
+        scope=["profile", "email"],
+        extras_params={"access_type": "offline"},
+    )
+    return authorization_url
+
+
+async def write_access_token(client, redirect_uri, code):
+    token = await client.get_access_token(code, redirect_uri)
+    return token
+
+
+async def get_user_info(client, token):
+    user_id, user_email = await client.get_id_email(token)
+    return user_id, user_email
+
+
+async def revoke_token(client, token):
+    return await client.revoke_token(token)
+
+
+def login_button(authorization_url, app_name, app_desc):
+    st.markdown('''<link href="https://cdn.jsdelivr.net/npm/bootstrap@5.0.2/dist/css/bootstrap.min.css" rel="stylesheet"
+    integrity="sha384-EVSTQN3/azprG1Anm3QDgpJLIm9Nao0Yz1ztcQTwFspd3yD65VohhpuuCOmLASjC" crossorigin="anonymous">''',
+    unsafe_allow_html=True)
+
+    container = f'''
+    <div class="container-fluid border py-4 px-4 border-primary">
+        <h5><strong>{app_name}</strong></h5>
+        <p>{app_desc}</p>
+        <a target="_self" href="{authorization_url}">
+            <img class="img-fluid" src="https://i.imgur.com/YTxsnUl.png" alt="streamlit">
+        </a>
+    </div>
+    '''
+    st.markdown(container, unsafe_allow_html=True)
+
+
+def logout_button(button_text):
+    if st.button(button_text):
+        asyncio.run(
+            revoke_token(
+                client=st.session_state.client,
+                token=st.session_state.token["access_token"],
+            )
+        )
+        st.session_state.user_email = None
+        st.session_state.user_id = None
+        st.session_state.token = None
+        st.experimental_rerun()
+
+
+def login(
+    client_id,
+    client_secret,
+    redirect_uri,
+    app_name="Continue with Google",
+    app_desc="",
+    logout_button_text="Logout",
+):
+    st.session_state.client = GoogleOAuth2(client_id, client_secret)
+    authorization_url = asyncio.run(
+        write_authorization_url(
+            client=st.session_state.client, redirect_uri=redirect_uri
+        )
+    )
+    app_desc
+    if "token" not in st.session_state:
+        st.session_state.token = None
+
+    if st.session_state.token is None:
+        try:
+            code = st.experimental_get_query_params()["code"]
+        except:
+            login_button(authorization_url, app_name, app_desc)
+        else:
+            # Verify token is correct:
+            try:
+                token = asyncio.run(
+                    write_access_token(
+                        client=st.session_state.client,
+                        redirect_uri=redirect_uri,
+                        code=code,
+                    )
+                )
+            except:
+                login_button(authorization_url, app_name, app_desc)
+            else:
+                # Check if token has expired:
+                if token.is_expired():
+                    login_button(authorization_url, app_name, app_desc)
+                else:
+                    st.session_state.token = token
+                    st.session_state.user_id, st.session_state.user_email = asyncio.run(
+                        get_user_info(
+                            client=st.session_state.client, token=token["access_token"]
+                        )
+                    )
+                    logout_button(button_text=logout_button_text)
+                    return (st.session_state.user_id, st.session_state.user_email)
+    else:
+        logout_button(button_text=logout_button_text)
+        return (st.session_state.user_id, st.session_state.user_email)

app.py

@@ -0,0 +1,33 @@
+import streamlit as st
+import os
+# from dotenv import load_dotenv
+import streamlit_google_oauth as oauth
+
+# load_dotenv()
+client_id = "657069732284-mbcb1l7ra4pask18n2cc7a29123ldsm3.apps.googleusercontent.com"
+client_secret = "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"
+redirect_uri = "https://agiveon-ath.streamlit.app/?embed=True"
+
+
+if __name__ == "__main__":
+    app_name = '''
+    Streamlit Google Authentication Demo
+    '''
+    app_desc = '''
+    A streamlit application that authenticates users by <strong>Google Oauth</strong>.
+    The user must have a google account to log in into the application.
+    '''
+    login_info = oauth.login(
+        client_id=client_id,
+        client_secret=client_secret,
+        redirect_uri=redirect_uri,
+        app_name=app_name,
+        app_desc=app_desc,
+        logout_button_text="Logout",
+    )
+    if login_info:
+        user_id, user_email = login_info
+        st.write(f"Welcome {user_email}")
+
+# streamlit run app.py --server.port 8080
+

requirements.txt

@@ -0,0 +1,5 @@
+protobuf<=3.20
+streamlit
+httpx-oauth
+python-dotenv
+typing_extensions

setup.py

@@ -0,0 +1,19 @@
+from setuptools import setup
+
+# from streamlit_google_oauth import __version__
+
+setup(
+    name="streamlit_google_oauth",
+    version="0.1",
+    url="https://github.com/hunkim/streamlit-google-oauth",
+    author="Sung Kim",
+    author_email="hunkim@gmail.com",
+    py_modules=["streamlit_google_oauth"],
+    packages=["streamlit_google_oauth"],
+    install_requires=[
+        "protobuf<=3.20",
+        "streamlit",
+        "httpx-oauth",
+        "typing-extensions",
+    ],
+)