Project_GRANITE / app.py
mswhite's picture
Update app.py
93387c6 verified
# =================================================
# Main Entry Page for Application
# =================================================
# =================================================
# Imports
# =================================================
import os, sys
sys.path.append(os.getcwd()) # to fix ModuleNotFoundError for Pages code
# print("sys.path:", sys.path)
from openai import OpenAI
from pages import load_pages
import streamlit as st
import streamlit.components.v1 as components
# =================================================
# Constants
# =================================================
# =================================================
# Functions
# =================================================
@st.dialog("Market Global Risk Monitor", width="medium") # 'large'
def app_login(intro_video):
if intro_video is not None: # Video has been loaded - use it
st.video(intro_video, loop=True, autoplay=False, muted=False)
else: # No video loaded, use static image
st.image(os.path.join(st.session_state["start_dir"],"pages","assets","logo.jpg"), width=1200)
user_name = st.text_input(label="User Name", placeholder="Please enter your User Name",label_visibility="collapsed",icon=":material/person_edit:")
user_password = st.text_input(label="Password", type="password", placeholder="Please enter your password",label_visibility="collapsed",icon=":material/key:")
speak_easy = os.environ["SPEAK_EASY"] #
users = os.environ["USERS"] #
if st.button("Log in") and (user_password.lower()==speak_easy) and user_name in users: #
st.session_state.logged_in = True
st.session_state.user_name = user_name
print("User Name is: ",st.session_state.user_name) # log user name
st.rerun() # rerun application with new session state
# =================================================
# Load State Variables
# Consistent across pages e.g. user / logged in / etc.
# =================================================
if "wide_layout" not in st.session_state:
st.set_page_config(layout="wide") # set default to wide page mode
st.session_state["wide_layout"] = True
if "start_dir" not in st.session_state:
st.session_state["start_dir"] = os.getcwd() # start directory for file hierarchy
if "openai_model" not in st.session_state:
st.session_state["openai_model"] = "gpt-5.2-pro" # advanced reasoning model with webtool support
if "openai_client" not in st.session_state:
API_KEY = os.environ["API_KEY"] # api key from environ / secrets
st.session_state["openai_client"] = OpenAI(api_key=API_KEY)
if "messages" not in st.session_state:
st.session_state.messages = []
# Load intro video - Optional
##intro_file = open(os.path.join(st.session_state["start_dir"],"pages","assets","ascend.mp4"),"rb")
##
##if intro_file is None:
## print("ERROR: Intro video file did not load")
##
##intro_video = intro_file.read()
intro_video = None
# Define login once and no pages available are available until a user login
if st.query_params.get("test_more_pages", False):
st.session_state.logged_in = True
if "logged_in" not in st.session_state:
app_login(intro_video) # None to disable and revert back to splash image
if "logged_in" not in st.session_state:
pass
else:
# Load default first page - Overview of Program
# load_pages.load_page("Overview")
load_pages.load_page("Module_01")
# =================================================
if __name__ == "__main__":
pass
# load_pages.test_load() # tests for successful module import
# print("<<< Main APP Compile Successful >>>") # DEBUG