From 7e2ce495922f267cff9c19f20da528ad8c04d9f4 Mon Sep 17 00:00:00 2001 From: "AARETE\\agupta" Date: Thu, 20 Jun 2024 16:56:17 -0500 Subject: [PATCH] Updated util.py in multipage --- streamlit/multipage/util.py | 44 +++++++++++++++++++++++++++++++------ 1 file changed, 37 insertions(+), 7 deletions(-) diff --git a/streamlit/multipage/util.py b/streamlit/multipage/util.py index 78c2782..89e2d64 100644 --- a/streamlit/multipage/util.py +++ b/streamlit/multipage/util.py @@ -1,17 +1,39 @@ - import streamlit as st import security import os import constants +import logging +from logging.handlers import RotatingFileHandler -def setup_page(REDIRECT_URI): +# Ensure the log directory exists +log_dir = '/home/ubuntu/doczy.ai/streamlit' +if not os.path.exists(log_dir): + os.makedirs(log_dir, exist_ok=True) + +# Configure logging with RotatingFileHandler +log_file = f'{log_dir}/interface.log' +rotating_handler = RotatingFileHandler( + log_file, + maxBytes=10*1024*1024, # 10 MB + backupCount=5 # Keep up to 5 backup files +) +rotating_handler.setLevel(logging.INFO) +formatter = logging.Formatter('%(asctime)s %(levelname)s [%(filename)s] %(message)s') +rotating_handler.setFormatter(formatter) + +logger = logging.getLogger() +logger.setLevel(logging.INFO) +logger.addHandler(rotating_handler) + + +def setup_page(redirect_uril): # st.set_page_config( # page_title=page_title, # page_icon="👋", # ) if st.query_params.get('code'): - security.handle_redirect(REDIRECT_URI) + security.handle_redirect(redirect_uril) access_token = st.session_state.get('access_token') @@ -21,15 +43,23 @@ def setup_page(REDIRECT_URI): return True else: st.write("Please sign-in to use this app.") - auth_url = security.get_auth_url(REDIRECT_URI) + auth_url = security.get_auth_url(redirect_uril) st.markdown(f"Sign In", unsafe_allow_html=True) st.stop() + def load_page_details(interface): env_var = os.environ.get('ENVIRONMENT', 'DEV') + logger.info(f"env_var={env_var}") if env_var == 'UAT': - return (constants.DOCZY_REDIRECT_URL_UAT + str(interface), constants.DOCZY_CREATE_BATCH_URL_UAT, constants.DOCZY_PIPELINE_URL_UAT) + logger.info(constants.DOCZY_REDIRECT_URL_UAT + str(interface)) + return (constants.DOCZY_REDIRECT_URL_UAT + str(interface), constants.DOCZY_CREATE_BATCH_URL_UAT, + constants.DOCZY_PIPELINE_URL_UAT) elif env_var == 'DEV': - return (constants.DOCZY_REDIRECT_URL_DEV + str(interface), constants.DOCZY_CREATE_BATCH_URL_DEV, constants.DOCZY_PIPELINE_URL_DEV) + logger.info(constants.DOCZY_REDIRECT_URL_DEV + str(interface)) + return (constants.DOCZY_REDIRECT_URL_DEV + str(interface), constants.DOCZY_CREATE_BATCH_URL_DEV, + constants.DOCZY_PIPELINE_URL_DEV) else: - return (constants.DOCZY_REDIRECT_URL_DEV + str(interface), constants.DOCZY_CREATE_BATCH_URL_DEV, constants.DOCZY_PIPELINE_URL_DEV) + logger.info(constants.DOCZY_REDIRECT_URL_DEV + str(interface)) + return (constants.DOCZY_REDIRECT_URL_DEV + str(interface), constants.DOCZY_CREATE_BATCH_URL_DEV, + constants.DOCZY_PIPELINE_URL_DEV)