Spaces:
Sleeping
Sleeping
File size: 2,022 Bytes
f027c05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
# From https://docs.streamlit.io/develop/tutorials/multipage/st.page_link-nav
import streamlit as st
import os
import project_config
def authenticated_menu():
# Insert profile picture
pfp_path = str(project_config.MEDIA_DIR / 'pfp' / f"{st.session_state.profile_pic}.png")
if not os.path.exists(pfp_path):
pfp_path = str(project_config.MEDIA_DIR / 'pfp' / "cipher.png")
st.sidebar.image(pfp_path, use_column_width=True)
st.sidebar.markdown("---")
# Show a navigation menu for authenticated users
# st.sidebar.page_link("app.py", label="Switch Accounts", icon="🔒")
st.sidebar.page_link("pages/about.py", label="About", icon="📖")
st.sidebar.page_link("pages/input.py", label="Input", icon="💡")
st.sidebar.page_link("pages/predict.py", label="Predict", icon="🔍",
disabled=("query" not in st.session_state))
st.sidebar.page_link("pages/validate.py", label="Validate", icon="✅",
disabled=("query" not in st.session_state))
# st.sidebar.page_link("pages/explore.py", label="Explore", icon="🔍")
if st.session_state.role in ["admin"]:
st.sidebar.page_link("pages/admin.py", label="Manage Users", icon="🔧")
# Show the logout button
st.sidebar.markdown("---")
st.sidebar.button("Log Out", on_click=lambda: st.session_state.clear())
def unauthenticated_menu():
# Show a navigation menu for unauthenticated users
st.sidebar.page_link("app.py", label="Log In", icon="🔒")
def menu():
# Determine if a user is logged in or not, then show the correct navigation menu
if "role" not in st.session_state or st.session_state.role is None:
unauthenticated_menu()
return
authenticated_menu()
def menu_with_redirect():
# Redirect users to the main page if not logged in, otherwise continue to
# render the navigation menu
if "role" not in st.session_state or st.session_state.role is None:
st.switch_page("app.py")
menu() |