Code Editor for app.py

import streamlit as st

# Initialize step state
if 'step' not in st.session_state:
    st.session_state.step = 1

# Function to display steps
def display_step(step):
    if step == 1:
        st.write("This is step 1: Press the button to proceed.")
        if st.button("Next"):  
            st.session_state.step = 2
    elif step == 2:
        st.write("This is step 2: You've moved to the next step!")
        if st.button("Go Back to Step 1"):  
            st.session_state.step = 1

# Display current step
display_step(st.session_state.step)