Update my_model/state_manager.py

my_model/state_manager.py

@@ -18,12 +18,12 @@ class StateManager:
     def initialize_state(self):
         if "previous_state" not in st.session_state:
             st.session_state['previous_state'] = {'method': None, 'detection_model': None, 'confidence_level': None}
-        if 'method' not in st.session_state:
-            st.session_state['method'] = "Fine-Tuned Model"
-        if 'confidence_level' not in st.session_state:
-            st.session_state['confidence_level'] = 0.4
-        if 'detection_model' not in st.session_state:
-            st.session_state['detection_model'] = 'detic'
+        #if 'method' not in st.session_state:
+        #    st.session_state['method'] = "Fine-Tuned Model"
+        #if 'confidence_level' not in st.session_state:
+        #    st.session_state['confidence_level'] = 0.4
+        #if 'detection_model' not in st.session_state:
+        #    st.session_state['detection_model'] = 'detic'
         if 'images_data' not in st.session_state:
             st.session_state['images_data'] = {}
         if 'kbvqa' not in st.session_state:
@@ -198,7 +198,8 @@ class StateManager:
         for key in st.session_state['previous_state']:
             if key == 'confidence_level':
                 continue  # confidence_level tracker is separate
-            if st.session_state[key] != st.session_state['previous_state'][key]:
+            if key in st.session_state and st.session_state[key] != st.session_state['previous_state'][key]:
+                st.write("change check", key)
                 return True  # Found a change
         else: return False  # No changes found