fix: session state handling

app.py

@@ -4,7 +4,6 @@ from utilis import Processing
 
 initial_page = "pages/upload_url.py"
 
-
 @st.cache_resource()
 def initialize_models():
     processing = Processing(
@@ -35,13 +34,13 @@ def disable_sidebar(page_title: str):
     """
     st.markdown(no_sidebar_style, unsafe_allow_html=True)
 
-    if "metadata" not in st.session_state:
-        st.switch_page("app.py")
-
 
 def main():
     disable_sidebar("Drake")
-    initialize_models()
+    st.title("Drake")
+
+    with st.spinner("Loading models..."):
+        processing, drake = initialize_models()
     st.switch_page(initial_page)
 
 

pages/upload_file.py

@@ -16,8 +16,8 @@ if st.button("Youtube/Video URL"):
 st.subheader('Upload the file')
 uploaded_file = st.file_uploader(label="Choose a file", type=['pdf', 'doc'])
 allow_make_notes = st.toggle('Make Complete Notes!')
-llm_model = st.selectbox('Choose LLM Model', 'gemini-pro')
-# llm_model = st.selectbox('Choose LLM Model', ('gemini-pro', 'llama'))
+llm_model = st.selectbox('Choose LLM Model', ('gemini-pro', 'llama'))
+st.caption("Note: Llama support to be added soon!")
 drake.llm_model = llm_model
 
 

pages/upload_url.py

@@ -13,8 +13,8 @@ if st.button("PDF/Transcript"):
 
 st.subheader('Enter the Video URL')
 video_url = st.text_input(label="Enter the URL")
-llm_model = st.selectbox('Choose LLM Model', 'gemini-pro')
-# llm_model = st.selectbox('Choose LLM Model', ('gemini-pro', 'llama'))
+llm_model = st.selectbox('Choose LLM Model', ('gemini-pro', 'llama'))
+st.caption("Note: Llama support to be added soon!")
 drake.llm_model = llm_model
 
 allow_make_notes = st.toggle('Make Complete Notes!')