fixed

pages/1_Textiles_Disentanglement.py

@@ -35,7 +35,6 @@ concept_vectors = pd.read_csv('./data/stored_vectors/scores_colors_hsv.csv')
 concept_vectors['vector'] = [np.array([float(xx) for xx in x]) for x in concept_vectors['vector'].str.split(', ')]
 concept_vectors['score'] = concept_vectors['score'].astype(float)
 concept_vectors = concept_vectors.sort_values('score', ascending=False).reset_index()
-print(concept_vectors[['vector', 'score']], concept_vectors.loc[0, 'vector'], concept_vectors.loc[0, 'vector'].shape)
 
 with dnnlib.util.open_url('./data/textile_model_files/network-snapshot-005000.pkl') as f:
     model = legacy.load_network_pkl(f)['G_ema'].to('cpu') # type: ignore
@@ -77,7 +76,7 @@ with input_col_1:
             st.session_state.image_id = int(image_id)
 
 with input_col_2:
-    with st.form('text_form'):
+    with st.form('text_form_1'):
         
         st.write('**Choose color to vary**')
         type_col = st.selectbox('Color:', tuple(COLORS_LIST))
@@ -88,7 +87,7 @@ with input_col_2:
         color_lambda = chosen_color_lambda_input.number_input('Lambda:', min_value=0, step=1, value=7)
         color_lambda_button = st.form_submit_button('Choose the defined lambda')
           
-        if choose_text_button:
+        if colors_button:
             st.session_state.concept_ids = type_col
             st.session_state.space_id = space_id
         
@@ -97,13 +96,13 @@ with input_col_3:
         
         st.write('**Saturation variation**')
         chosen_saturation_lambda_input = st.empty()
-        saturation_lambda = chosen_saturation_lambda_input.number_input('Lambda:', min_value=0, step=1)
-        saturation_lambda_button = st.form_submit_button('Choose the defined lambda')
+        saturation_lambda = chosen_saturation_lambda_input.number_input('Lambda:', min_value=0, step=1, key=0)
+        saturation_lambda_button = st.form_submit_button('Choose the defined lambda for Saturation')
         
         st.write('**Value variation**')
         chosen_value_lambda_input = st.empty()
-        value_lambda = chosen_value_lambda_input.number_input('Lambda:', min_value=0, step=1)
-        value_lambda_button = st.form_submit_button('Choose the defined lambda')
+        value_lambda = chosen_value_lambda_input.number_input('Lambda:', min_value=0, step=1, key=1)
+        value_lambda_button = st.form_submit_button('Choose the defined lambda for Value')
         
 # with input_col_4:
 #     with st.form('Network specifics:'):
@@ -140,9 +139,9 @@ 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.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']]
+    color_separation_vector, performance_color = concept_vectors[concept_vectors['color'] == st.session_state.concept_ids].reset_index().loc[0, ['vector', 'score']]
+    saturation_separation_vector, performance_saturation = concept_vectors[concept_vectors['color'] == 'Saturation'].reset_index().loc[0, ['vector', 'score']]
+    value_separation_vector, performance_value = concept_vectors[concept_vectors['color'] == 'Value'].reset_index().loc[0, ['vector', 'score']]
     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 ------------------------------