Update my_model/state_manager.py

my_model/state_manager.py

@@ -25,23 +25,27 @@ class StateManager:
         
         # Create two columns with different widths
         col1, col2, col3 = st.columns([0.2, 0.6, 0.2])  # Adjust the ratio as needed
-     
-        col1.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
-        detection_model = col1.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
-        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=col1)
+
+        with col1:
+            st.selectbox("Choose a method:", ["Fine-Tuned Model", "In-Context Learning (n-shots)"], index=0, key='method')
+            detection_model = st.selectbox("Choose a model for objects detection:", ["yolov5", "detic"], index=1, key='detection_model')
+            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')
 
         # Conditional display of model settings
-     
-        show_model_settings = col3.checkbox("Show Model Settings", False)
-        if show_model_settings:
-            self.display_model_settings()
 
+        with col3:
+            show_model_settings = st.checkbox("Show Model Settings", False)
+            if show_model_settings:
+                self.display_model_settings()
 
-        
-    def set_slider_value(self, text, min_value, max_value, value, step, slider_key_name, col):
 
-        return col.slider(text, min_value, max_value, value, step, key=slider_key_name)
+        
+    def set_slider_value(self, text, min_value, max_value, value, step, slider_key_name, col=None):
+        if col is None:
+            return st.slider(text, min_value, max_value, value, step, key=slider_key_name)
+        else:
+            return col.slider(text, min_value, max_value, value, step, key=slider_key_name)
 
         
     @property
@@ -50,7 +54,7 @@ class StateManager:
 
     
     def display_model_settings(self):
-        st.write("#### Current Model Settings:")
+        st.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')]}])