in UI/nl2sqlstudio_ui.py [0:0]
def sidebar_components() -> None:
"""
UI Controls in the Sidebar panel
"""
with st.sidebar.container():
column_1, column_2 = st.columns(2)
with column_1:
# st.write('v1.2')
st.image('google.png')
with column_2:
url = 'https://googlecloudplatform.github.io/nl2sql-studio/'
st.markdown("[User Guide](%s)" % url)
logout_state = st.button("Logout")
with st.sidebar.container():
st.write(" ")
gen_engine = st.sidebar.selectbox(
"Choose NL2SQL framework",
(LITE, CORE)
)
logger.info(f"Generation using : {gen_engine}")
if gen_engine == CORE:
st.session_state.generation_engine = CORE
with st.sidebar.container(height=140):
st.session_state.model = st.radio('Select Prompting Technique',
[LINEAR, COT])
elif gen_engine == LITE:
st.session_state.generation_engine = LITE
with st.sidebar.container(height=115):
st.session_state.lite_model = st.radio(
'Select Prompting Technique',
[FEW_SHOT, ZERO_SHOT])
else:
st.session_state.generation_engine = None
with st.sidebar.expander("Configuration Settings"):
proj_conf = st.button("Project Configuration")
rag_input = st.button("Questions & Queries", disabled=False)
with st.sidebar.container(height=60):
st.session_state.execution = st.checkbox(
"Generate and Execute",
disabled=False
)
if proj_conf:
pc_modal = st.session_state.pc_modal
pc_modal.open()
if rag_input:
qa_modal = st.session_state.qa_modal
qa_modal.open()
if logout_state:
logger.info("Logging out")
st.session_state.token = None
st.session_state.login_status = False
back_to_login_page()
st.query_params.clear()