utils/levels.py

import streamlit as st

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 st.session_state['level'] < level:
        st.error(f"You need to complete Level {st.session_state['level']} first!")
    else:
        page()

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.')