Update my_model/state_manager.py

my_model/state_manager.py

@@ -42,10 +42,10 @@ class StateManager:
         Sets up user interface widgets for selecting models, settings, and displaying model settings conditionally.
         """
         disabled=st.session_state['loading_in_progress']
-        self.col1.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method', disabled=disabled)
-        detection_model = self.col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model', disabled=disabled)
+        self.col1.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method', disabled=self.is_widget_disabled)
+        detection_model = self.col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model', disabled=self.is_widget_disabled)
         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', col=self.col1, disabled=disabled)
+        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', col=self.col1, disabled=self.is_widget_disabled)
 
         # Conditional display of model settings
 
@@ -71,10 +71,13 @@ class StateManager:
         """
         
         if col is None:
-            return st.slider(text, min_value, max_value, value, step, key=slider_key_name)
+            return st.slider(text, min_value, max_value, value, step, key=slider_key_name, disabled=disabled)
         else:
-            return col.slider(text, min_value, max_value, value, step, key=slider_key_name)
+            return col.slider(text, min_value, max_value, value, step, key=slider_key_name, disabled=disabled)
 
+    @property
+    def is_widget_disabled(self):
+        return st.session_state['loading_in_progress']
         
     @property
     def settings_changed(self):
@@ -141,7 +144,6 @@ class StateManager:
         try:
             st.session_state['loading_in_progress'] = True
             self.delete_model()
-            self.initialize_state()
             free_gpu_resources()
             st.session_state['kbvqa'] = prepare_kbvqa_model(force_reload=True)
             st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level