fix some bug

aws-samples · Jul 4, 2024 · 9dd887b · 9dd887b
1 parent 5b3dcdb
commit 9dd887b
Show file tree

Hide file tree

Showing 6 changed files with 19 additions and 0 deletions.
diff --git a/application/pages/3_🪙_Data_Profile_Management.py b/application/pages/3_🪙_Data_Profile_Management.py
@@ -22,6 +22,9 @@ def main():
     if 'profile_page_mode' not in st.session_state:
         st.session_state['profile_page_mode'] = 'default'
 
+    if 'current_profile' not in st.session_state:
+        st.session_state['current_profile'] = ''
+
     with st.sidebar:
         st.title("Data Profile Management")
         st.selectbox("My Data Profiles", ProfileManagement.get_all_profiles(),

diff --git a/application/pages/4_🪙_Schema_Description_Management.py b/application/pages/4_🪙_Schema_Description_Management.py
@@ -12,6 +12,9 @@ def main():
     st.set_page_config(page_title="Schema Management", )
     make_sidebar()
 
+    if 'current_profile' not in st.session_state:
+        st.session_state['current_profile'] = ''
+
     with st.sidebar:
         st.title("Schema Management")
         all_profiles_list = ProfileManagement.get_all_profiles()

diff --git a/application/pages/5_🪙_Prompt_Management.py b/application/pages/5_🪙_Prompt_Management.py
@@ -14,6 +14,9 @@ def main():
     st.set_page_config(page_title="Prompt Management")
     make_sidebar()
 
+    if 'current_profile' not in st.session_state:
+        st.session_state['current_profile'] = ''
+
     with st.sidebar:
         st.title("Prompt Management")
         all_profiles_list = ProfileManagement.get_all_profiles()

diff --git a/application/pages/6_📚_Index_Management.py b/application/pages/6_📚_Index_Management.py
@@ -24,6 +24,10 @@ def main():
     if 'profile_page_mode' not in st.session_state:
         st.session_state['index_mgt_mode'] = 'default'
 
+
+    if 'current_profile' not in st.session_state:
+        st.session_state['current_profile'] = ''
+
     with st.sidebar:
         st.title("Index Management")
         all_profiles_list = ProfileManagement.get_all_profiles()

diff --git a/application/pages/7_📚_Entity_Management.py b/application/pages/7_📚_Entity_Management.py
@@ -25,6 +25,9 @@ def main():
     if 'profile_page_mode' not in st.session_state:
         st.session_state['index_mgt_mode'] = 'default'
 
+    if 'current_profile' not in st.session_state:
+        st.session_state['current_profile'] = ''
+
     with st.sidebar:
         st.title("Entity Management")
         all_profiles_list = ProfileManagement.get_all_profiles()

diff --git a/application/pages/8_📚_Agent_Cot_Management.py b/application/pages/8_📚_Agent_Cot_Management.py
@@ -24,6 +24,9 @@ def main():
     if 'profile_page_mode' not in st.session_state:
         st.session_state['index_mgt_mode'] = 'default'
 
+    if 'current_profile' not in st.session_state:
+        st.session_state['current_profile'] = ''
+
     with st.sidebar:
         st.title("Agent Cot Management")
         all_profiles_list = ProfileManagement.get_all_profiles()