add profile select

Author: Pinyu Su

application/pages/1_🌍_Generative_BI_Playground.py

@@ -156,12 +156,17 @@ def main():
         # all_profiles.update(demo_profile)
         st.session_state['profiles'] = all_profiles
     else:
-        all_profiles = ProfileManagement.get_all_profiles_with_info()
-        st.session_state['profiles'] = all_profiles
+        if st.session_state.update_profile:
+            all_profiles = ProfileManagement.get_all_profiles_with_info()
+            st.session_state['profiles'] = all_profiles
+            st.session_state.update_profile = False
 
     if "vision_change" not in st.session_state:
         st.session_state["vision_change"] = False
 
+    if "update_profile" not in st.session_state:
+        st.session_state.update_profile = False
+
     if 'selected_sample' not in st.session_state:
         st.session_state['selected_sample'] = ''
 

application/pages/3_🪙_Data_Profile_Management.py

@@ -19,6 +19,9 @@ def main():
     st.set_page_config(page_title="Data Profile Management", )
     make_sidebar()
 
+    if "update_profile" not in st.session_state:
+        st.session_state.update_profile = False
+
     if 'profile_page_mode' not in st.session_state:
         st.session_state['profile_page_mode'] = 'default'
 
@@ -49,6 +52,7 @@ def main():
             comments = st.text_input("Comments")
 
             if st.button('Create Profile', type='primary'):
+                st.session_state.update_profile = True
                 if not selected_tables:
                     st.error('Please select at least one table.')
                     return
@@ -89,6 +93,7 @@ def main():
                                 "Your sample question 2")
 
         if st.button('Update Profile', type='primary'):
+            st.session_state.update_profile = True
             if not selected_tables:
                 st.error('Please select at least one table.')
                 return
@@ -99,6 +104,7 @@ def main():
                 st.success('Profile updated. Please click "Fetch table definition" button to continue.')
 
         if st.button('Fetch table definition'):
+            st.session_state.update_profile = True
             if not selected_tables:
                 st.error('Please select at least one table.')
             with st.spinner('fetching...'):
@@ -109,6 +115,7 @@ def main():
                 st.session_state.profile_page_mode = 'default'
 
         if st.button('Delete Profile'):
+            st.session_state.update_profile = True
             ProfileManagement.delete_profile(profile_name)
             st.success(f"{profile_name} deleted successfully!")
             st.session_state.profile_page_mode = 'default'