Update my_model/state_manager.py

my_model/state_manager.py

@@ -25,6 +25,11 @@ class StateManager:
             st.session_state['previous_state'] = {}
         if "settings_changed" not in st.session_state:
             st.session_state['settings_changed'] = self.settings_changed
+        if 'model_loaded' not in st.session_state:
+            st.session_state['model_loaded'] = False
+        if 'loading_in_progress' not in st.session_state:
+            st.session_state['loading_in_progress'] = False
+
             
 
 
@@ -87,7 +92,7 @@ class StateManager:
         self.col3.write("##### Current Model Settings:")
         data = [{'Key': key, 'Value': str(value)} for key, value in st.session_state.items() if key in ["confidence_level", 'detection_model', 'method', 'kbvqa', 'previous_state', 'settings_changed', ]]
         df = pd.DataFrame(data)
-        styled_df = df.style.set_properties(**{'background-color': 'black', 'color': 'white', 'border-color': 'white'}).set_table_styles([{'selector': 'th','props': [('background-color', 'black'), ('font-weight', 'bold')]}])
+        styled_df = df.style.set_properties(**{'background-color': 'white', 'color': 'black', 'border-color': 'black'}).set_table_styles([{'selector': 'th','props': [('background-color', 'gray'), ('font-weight', 'bold')]}])
         self.col3.table(styled_df)
 
     
@@ -117,13 +122,10 @@ class StateManager:
             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['button_label'] = "Reload Model"
-            #st.text('button changed')
-            #self.has_state_changed()
             free_gpu_resources()
         except Exception as e:
             st.error(f"Error loading model: {e}")