Update my_model/state_manager.py

my_model/state_manager.py

@@ -41,11 +41,11 @@ class StateManager:
         """
         Sets up user interface widgets for selecting models, settings, and displaying model settings conditionally.
         """
-
-        self.col1.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
-        detection_model = self.col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
+        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)
         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)
+        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)
 
         # Conditional display of model settings
 
@@ -56,7 +56,7 @@ class StateManager:
 
 
         
-    def set_slider_value(self, text, min_value, max_value, value, step, slider_key_name, col=None):
+    def set_slider_value(self, text, min_value, max_value, value, step, slider_key_name, col=None, disabled=False):
         """
         Creates a slider widget with the specified parameters, optionally placing it in a specific column.
 
@@ -123,14 +123,17 @@ class StateManager:
         """
         
         try:
+            st.session_state['loading_in_progress'] = True
             free_gpu_resources()
             st.session_state['kbvqa'] = prepare_kbvqa_model()
             st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
             # 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}
             st.session_state['model_loaded'] = True
+            st.session_state['loading_in_progress'] = False
             st.session_state['button_label'] = "Reload Model"
             free_gpu_resources()
+            
         except Exception as e:
             st.error(f"Error loading model: {e}")