auth.py

from datetime import datetime
from time import sleep

import pytz
import streamlit as st

from src.utils.greeting import GetRandomWelcomeMessage, GreetUser


def unix_to_ist(timestamp):
  india_tz = pytz.timezone("Asia/Kolkata")
  format_str = "%I:%M:%S %p IST"
  return datetime.fromtimestamp(timestamp, pytz.utc).astimezone(india_tz).strftime(format_str)


def auth():
  if st.user and not st.user.is_logged_in:
    st.title("🔐 Login Required")
    st.write("Please authenticate using your Google account to access your profile.")
    if st.button("🔓 Authenticate with Google"):
      st.login("google")

  else:
    st.title(f"🙏 {GreetUser(st.user.given_name)}")
    st.success(GetRandomWelcomeMessage(), icon="🤝")
    st.image(st.user.picture, caption=st.user.name)
    st.write("Email:", st.user.email)

    if st.button("Log out"):
      st.toast(f"Goodbye, {st.user.name}! See you soon!", icon="🚪")
      sleep(2)
      st.logout()


auth()