Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -26,7 +26,13 @@ class StateManager:
         detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
         default_confidence = 0.2 if st.session_state.detection_model == "yolov5" else 0.4
         self.set_slider_value(text="Select minimum detection confidence level", min_value=0.1, max_value=0.9, value=default_confidence, step=0.1, slider_key_name='confidence_level')
+        show_session_state = st.sidebar.checkbox("Show Session State", False)
 
+        # Conditional display of model settings
+        if show_session_state:
+            display_model_settings()
+
+        
     def set_slider_value(self, text, min_value, max_value, value, step, slider_key_name):
 
         return st.slider(text, min_value, max_value, value, step, key=slider_key_name)
@@ -37,7 +43,7 @@ class StateManager:
 
     def display_model_settings(self):
         st.write("#### Current Model Settings:")
-        data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa']]
+        data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa', 'previous_state', 'settings_changed', ]]
         df = pd.DataFrame(data)
         styled_df = df.style.set_properties(**{'background-color': 'black', 'color': 'white', 'border-color': 'white'}).set_table_styles([{'selector': 'th','props': [('background-color', 'black'), ('font-weight', 'bold')]}])
         st.table(styled_df)