type

pages/1_Textiles_Disentanglement.py

@@ -80,7 +80,8 @@ with input_col_2:
     with st.form('text_form'):
         
         st.write('**Choose color to vary**')
-        type_col = st.selectbox('Color:', tuple(COLORS_LIST), value=st.session_state.concepts_ids)
+        type_col = st.selectbox('Color:', tuple(COLORS_LIST), value=st.session_state.concept_ids)
+        colors_button = st.form_submit_button('Choose the defined color')
         
         st.write('**Set range of change**')
         chosen_color_lambda_input = st.empty()
@@ -139,10 +140,10 @@ with header_col_1:
     st.write(f'Original image')
 
 with header_col_2:
-    color_separation_vector, performance_color = concept_vectors[concept_vectors['color'] == st.session_state.concepts_ids].loc[0, ['vector', 'score']]
+    color_separation_vector, performance_color = concept_vectors[concept_vectors['color'] == st.session_state.concept_ids].loc[0, ['vector', 'score']]
     saturation_separation_vector, performance_saturation = concept_vectors[concept_vectors['color'] == 'Saturation'].loc[0, ['vector', 'score']]
     value_separation_vector, performance_value = concept_vectors[concept_vectors['color'] == 'Value'].loc[0, ['vector', 'score']]
-    st.write(f'Change in {st.session_state.concepts_ids} of {np.round(color_lambda, 2)}, in saturation of {np.round(saturation_lambda, 2)}, in value of {np.round(value_lambda, 2)}. - Performance color vector: {performance_color}, saturation vector: {performance_saturation}, value vector: {performance_value}')
+    st.write(f'Change in {st.session_state.concept_ids} of {np.round(color_lambda, 2)}, in saturation of {np.round(saturation_lambda, 2)}, in value of {np.round(value_lambda, 2)}. - Performance color vector: {performance_color}, saturation vector: {performance_saturation}, value vector: {performance_value}')
 
 # ---------------------------- DISPLAY COL 2 ROW 1 ------------------------------