jacob-mathison/doczyai-pipelines
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Activity

Updated util.py in multipage

Browse Source
This commit is contained in:
AARETE\agupta
2024-06-20 16:56:17 -05:00
parent 967ecb0e60
commit 7e2ce49592
1 changed files with 37 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
streamlit/multipage/util.py
+37 -7
View File
@@ -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"<a href='{auth_url}' target='_self'>Sign In</a>", 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)