pages/composer.py

import streamlit as st
from utils.menu import menu_with_redirect,_logout
from pages import solution
from loguru import logger

menu_with_redirect()

# query_lib = 'query_library/'
# with st.container():
solution.compose_dataset()
    # with st.container():
    #     if 'retrieval_sql' in st.session_state and 'selected_db' in st.session_state:
    #         if st.button('Save Query'):
    #             database_name = st.session_state['selected_db']
    #             sql_saved = solution.save_sql_query_blob(st.session_state['retrieval_query'], st.session_state['retrieval_sql'], st.session_state['retrieval_query_no'], st.session_state['retrieval_result_structure'], query_lib, database_name)
    #             if sql_saved:
    #                 st.write(f"Query saved in the library with id {st.session_state['retrieval_query_no']}.")
    #                 logger.info("Query saved in the library with id {}.", st.session_state['retrieval_query_no'])

# st.success("You have been successfully entered in data composer.")
# run()