main.py

from wheels.tabs.tab1 import *
from wheels.tabs.tab2 import *
from wheels.tabs.tab3 import *
from wheels.login.login import *

# ============================================================
# Main
# ============================================================

if 'logged_in' not in st.session_state:
    st.session_state.logged_in = False
if 'store_id' not in st.session_state:
    st.session_state.store_id = None
if 'store_name' not in st.session_state:
    st.session_state.store_name = ""

def main_app():
    add_custom_css()
    if st.sidebar.button("🔒 Log out", key="logout_main"):
        st.session_state.clear()
        st.rerun()
    st.sidebar.title("Navigation")
    nav_options = ["Top Performing Items", "Worst Packaged Beverage Brands", "Payment Type Comparison"]
    nav_choice = st.sidebar.radio("", nav_options, key="nav_options")
    if nav_choice == "Top Performing Items":
        top_performing_items_tab()
    elif nav_choice == "Worst Packaged Beverage Brands":
        worst_brands_tab_app2()
    elif nav_choice == "Payment Type Comparison":
        credit_vs_cash_tab_new()

if not st.session_state.logged_in:
    login_page()
else:
    main_app()