Merge branch 'main' of https://huggingface.co/spaces/BatuhanYilmaz/Whisper-Auto-Subtitled-Video-Generator

components/.env

@@ -1,4 +0,0 @@
-COGNITO_DOMAIN = "xxx"
-CLIENT_ID = "xxx"
-CLIENT_SECRET = "xxx"
-APP_URI = "xxx"

components/authenticate.py

@@ -1,158 +0,0 @@
-import os
-import streamlit as st
-from dotenv import load_dotenv
-import requests
-import base64
-import json
-
-load_dotenv()
-COGNITO_DOMAIN = os.environ.get("COGNITO_DOMAIN")
-CLIENT_ID = os.environ.get("CLIENT_ID")
-CLIENT_SECRET = os.environ.get("CLIENT_SECRET")
-APP_URI = os.environ.get("APP_URI")
-
-
-def init_state():
-    if "auth_code" not in st.session_state:
-        st.session_state["auth_code"] = ""
-    if "authenticated" not in st.session_state:
-        st.session_state["authenticated"] = False
-    if "user_cognito_groups" not in st.session_state:
-        st.session_state["user_cognito_groups"] = []
-
-# Get the authorization code after the user has logged in
-def get_auth_code():
-    auth_query_params = st.experimental_get_query_params()
-    try:
-        auth_code = dict(auth_query_params)["code"][0]
-    except (KeyError, TypeError):
-        auth_code = ""
-    return auth_code
-
-
-# Set the authorization code after the user has logged in
-def set_auth_code():
-    init_state()
-    auth_code = get_auth_code()
-    st.session_state["auth_code"] = auth_code
-
-
-# Get the access token from the authorization code
-def get_user_tokens(auth_code):
-    # Variables to make a post request
-    token_url = f"{COGNITO_DOMAIN}/oauth2/token"
-    client_secret_string = f"{CLIENT_ID}:{CLIENT_SECRET}"
-    client_secret_encoded = str(
-        base64.b64encode(client_secret_string.encode("utf-8")), "utf-8"
-    )
-    headers = {
-        "Content-Type": "application/x-www-form-urlencoded",
-        "Authorization": f"Basic {client_secret_encoded}",
-    }
-    body = {
-        "grant_type": "authorization_code",
-        "client_id": CLIENT_ID,
-        "code": auth_code,
-        "redirect_uri": APP_URI,
-    }
-
-    token_response = requests.post(token_url, headers=headers, data=body)
-    try:
-        access_token = token_response.json()["access_token"]
-        id_token = token_response.json()["id_token"]
-    except (KeyError, TypeError):
-        access_token = ""
-        id_token = ""
-
-    return access_token, id_token
-
-
-# Use access token to retrieve user info
-def get_user_info(access_token):
-    userinfo_url = f"{COGNITO_DOMAIN}/oauth2/userInfo"
-    headers = {
-        "Content-Type": "application/json;charset=UTF-8",
-        "Authorization": f"Bearer {access_token}",
-    }
-
-    userinfo_response = requests.get(userinfo_url, headers=headers)
-
-    return userinfo_response.json()
-
-
-# Decode access token to JWT to get user's cognito groups
-def pad_base64(data):
-    missing_padding = len(data) % 4
-    if missing_padding != 0:
-        data += "=" * (4 - missing_padding)
-    return data
-
-
-def get_user_cognito_groups(id_token):
-    user_cognito_groups = []
-    if id_token != "":
-        header, payload, signature = id_token.split(".")
-        printable_payload = base64.urlsafe_b64decode(pad_base64(payload))
-        payload_dict = json.loads(printable_payload)
-        try:
-            user_cognito_groups = list(dict(payload_dict)["cognito:groups"])
-        except (KeyError, TypeError):
-            pass
-    return user_cognito_groups
-
-
-# Set streamlit state variables
-def set_st_state_vars():
-    init_state()
-    auth_code = get_auth_code()
-    access_token, id_token = get_user_tokens(auth_code)
-    user_cognito_groups = get_user_cognito_groups(id_token)
-
-    if access_token != "":
-        st.session_state["auth_code"] = auth_code
-        st.session_state["authenticated"] = True
-        st.session_state["user_cognito_groups"] = user_cognito_groups
-
-
-# Login/ Logout HTML components
-login_link = f"{COGNITO_DOMAIN}/login?client_id={CLIENT_ID}&response_type=code&scope=email+openid&redirect_uri={APP_URI}"
-logout_link = f"{COGNITO_DOMAIN}/logout?client_id={CLIENT_ID}&logout_uri={APP_URI}"
-
-html_css_login = """
-<style>
-.button-login {
-  background-color: skyblue;
-  color: white !important;
-  padding: 1em 1.5em;
-  text-decoration: none;
-  text-transform: uppercase;
-}
-
-.button-login:hover {
-  background-color: #555;
-  text-decoration: none;
-}
-
-.button-login:active {
-  background-color: black;
-}
-
-</style>
-"""
-
-html_button_login = (
-    html_css_login
-    + f"<a href='{login_link}' class='button-login' target='_self'>Log In</a>"
-)
-html_button_logout = (
-    html_css_login
-    + f"<a href='{logout_link}' class='button-login' target='_self'>Log Out</a>"
-)
-
-
-def button_login():
-    return st.sidebar.markdown(f"{html_button_login}", unsafe_allow_html=True)
-
-
-def button_logout():
-    return st.sidebar.markdown(f"{html_button_logout}", unsafe_allow_html=True)