import home
import streamlit as st

st.set_page_config(layout="wide")
st.sidebar.image('assets/img/textgrad_logo_1x.png', width=200)
st.sidebar.title("TextGrad Playground")
st.sidebar.markdown("## Menu")

PAGES = {
    "🏠 Home": home,
}

page = st.sidebar.radio("", list(PAGES.keys()))
st.sidebar.markdown("## Links")

st.sidebar.markdown("""
                        📖 [Paper](https://arxiv.org/abs/2406.07496)
                        👨‍💻 [Code](https://github.com/zou-group/textgrad)
                        ⚙️ [API documentation](https://textgrad.readthedocs.io/en/latest/index.html)""")

PAGES[page].app()