Update my_model/utilities/st_utils.py

my_model/utilities/st_utils.py

@@ -90,9 +90,10 @@ class StateManager:
         if selected_method is not None:
             st.session_state['selected_method'] = selected_method
 
-    def check_settings_changed(self, current_detection_model, current_confidence_level):
+    def check_settings_changed(self, current_selected_method, current_detection_model, current_confidence_level):
         return (st.session_state['model_settings']['detection_model'] != current_detection_model or
-                st.session_state['model_settings']['confidence_level'] != current_confidence_level)
+                st.session_state['model_settings']['confidence_level'] != current_confidence_level
+                st.session_state['model_settings']['selected_method'] != current_selected_method)
 
     def display_model_settings(self):
         st.write("### Current Model Settings:")
@@ -104,6 +105,23 @@ class StateManager:
         df = pd.DataFrame(data)
         st.table(df)
 
+    def is_model_loaded(self):
+        """Check if the model is loaded in the session state."""
+        return 'kbvqa' in st.session_state and st.session_state['kbvqa'] is not None
+
+
+    def reload_detection_model(self, detection_model, confidence_level):
+        """Reload only the detection model with new settings."""
+        try:
+            free_gpu_resources()
+            if self.is_model_loaded():
+                prepare_kbvqa_model(detection_model, only_reload_detection_model=True)
+                st.session_state['kbvqa'].detection_confidence = confidence_level
+                self.update_model_settings(detection_model, confidence_level)
+            free_gpu_resources()
+        except Exception as e:
+            st.error(f"Error reloading detection model: {e}")
+