utils/levels.py

import streamlit as st
from utils.login import get_login

def initialize_level():
    if 'level' not in st.session_state:
        st.session_state['level'] = 0

def get_level():
    return st.session_state['level']

def render_page(page, level):
    if get_login()["status"]:
        if st.session_state['level'] < level:
            st.error(f"You need to complete Level {st.session_state['level']} first!")
        else:
            page()
    else:
        st.error("You need to login first!")

def complete_level(level):
    if st.session_state['level'] > level:
        st.info(f'You have Already completed Level {level}!')
    else:
        st.session_state['level'] = level + 1
        st.balloons()
        st.success(f'You have completed Level {level}! You can now move on to the next level.')