Update my_model/state_manager.py

my_model/state_manager.py

@@ -32,6 +32,8 @@ class StateManager:
             st.session_state['force_reload_button_clicked'] = False
         if 'time_taken_to_load_model' not in st.session_state:
             st.session_state['time_taken_to_load_model'] = None
+        if 'confidence_level_changed' not in st.session_state:
+            st.session_state['confidence_level_changed'] = False
         if "settings_changed" not in st.session_state:
             st.session_state['settings_changed'] = self.settings_changed
         if 'model_loaded' not in st.session_state:
@@ -232,6 +234,7 @@ class StateManager:
                      
                         st.session_state['kbvqa'].detection_confidence = st.session_state.confidence_level
                         st.write("reload func", st.session_state['kbvqa'].detection_confidence )
+                        st.session_state['confidence_level_changed'] = True
                         return  # only update the confidence level
                     
                 prepare_kbvqa_model(only_reload_detection_model=True)
@@ -330,7 +333,7 @@ class StateManager:
             detected_objects_str (str): String representation of detected objects.
             analysis_done (bool): Flag indicating if analysis of the image is complete.
         """
-        if image_key in st.session_state['images_data']:
+        if image_key in st.session_state['images_data'] or st.session_state["confidence_level_changed"]:
             st.session_state['images_data'][image_key].update({
                 'caption': caption,
                 'detected_objects_str': detected_objects_str,