app.py

import streamlit as st
from regular_user import regular_user
from admin_user import admin_user
from loginpage import login
from login_page_admin import admin_login
def main():
    if 'page' not in st.session_state:
        st.session_state.page = 'app'
    if st.session_state.page == 'app':
        if 'loginbtns' not in st.session_state:
            st.session_state.loginbtns = True
        col1,col2 = st.columns(2)
        if st.session_state.loginbtns:
            with col1:
                if st.button('User Login'):
                    st.session_state.page = 'login'
                    st.session_state.loginbtns = False
                    st.rerun()
            with col2:
                if st.button('Admin Login'):
                    st.session_state.page = 'admin_login'
                    st.session_state.loginbtns = False
                    st.rerun()

    if st.session_state.page == 'regular_user':
        regular_user()
    elif st.session_state.page == 'admin_user':
        admin_user()
    elif st.session_state.page=='login':
        login()
    elif st.session_state.page=='admin_login':
        admin_login()
if __name__=='__main__':
    main()