layout.py

import streamlit as st

def init_session_state():
    if 'dark_mode' not in st.session_state:
        st.session_state.dark_mode = False
    if 'code' not in st.session_state:
        st.session_state.code = ""
    if 'stdin' not in st.session_state:
        st.session_state.stdin = ""