Update my_model/utilities/state_manager.py

my_model/utilities/state_manager.py

@@ -12,32 +12,15 @@ class StateManager:
     def initialize_state(self):
         if 'images_data' not in st.session_state:
             st.session_state['images_data'] = {}
-        #if 'method' not in st.session_state:
-         #   st.session_state['method'] = None
-       # if 'detection_model' not in st.session_state:
-        #    st.session_state['detection_model'] = None
         if 'kbvqa' not in st.session_state:
             st.session_state['kbvqa'] = None
-      #  if 'confidence_level' not in st.session_state:
-       #     st.session_state['confidence_level'] = None
 
 
-    
     def set_up_widgets(self):
         st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
         detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
-        st.write('detection model')
-        st.write(st.session_state.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'
-                                                    )
-        st.write("Detection Confidence")
-        st.write(st.session_state.confidence_level)
+        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')
 
     def set_slider_value(self, text, min_value, max_value, value, step, slider_key_name):
 
@@ -69,9 +52,19 @@ class StateManager:
             st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
             #self.update_model_settings(detection_model, confidence_level)
             st.text("Model is ready for inference.")
+            # Update the previous state with current session state values
+            st.session_state['previous_state'] = {'method': st.session_state.method, 'detection_model': st.session_state.detection_model, 'confidence_level': st.session_state.confidence_level}
+            
             free_gpu_resources()
         except Exception as e:
-            st.error(f"Error loading model: {e}")    
+            st.error(f"Error loading model: {e}")
+
+    # Function to check if any session state values have changed
+    def has_state_changed(self):
+        for key in st.session_state['previous_state']:
+            if st.session_state[key] != st.session_state['previous_state'][key]:
+                return True  # Found a change
+            else: return False  # No changes found   
 
     def get_model(self):
         """Retrieve the KBVQA model from the session state."""