Update pages/type_text.py

pages/type_text.py

@@ -97,7 +97,7 @@ INTdesc_input = st.text_input("Type internal description", key="user_input")
 
 placeholder, right_column = st.columns(2)
 #placeholder_clicked = placeholder.button("Perform some action", key="user_placeholder")
-right_column.button("Reset", on_click=on_click)
+right_column.button("Reset description", on_click=on_click)
 
 numMAPPINGS_input = 5
 #numMAPPINGS_input = st.text_input("Type number of mappings and hit Enter", key="user_input_numMAPPINGS")
@@ -133,7 +133,7 @@ def load_model():
 model = load_model() 
 
 
-mapSBS_button = st.button("Map to SBS codes")
+mapSBS_button = st.button("Map to SBS codes", on_click=on_click, key="user_clickedSBS")
 
 
 INTdesc_embedding = model.encode(INTdesc_input)
@@ -157,7 +157,7 @@ rs_models = {
 }
  
 ## Create the select box
-selected_rs_model = st.selectbox('Choose a Reasoning model:', list(st_models.keys()))
+selected_rs_model = st.selectbox('Choose a Reasoning model:', list(rs_models.keys()))
 st.write("Current selection:", selected_rs_model)
 
 ## Get the selected model
@@ -174,7 +174,7 @@ pipe = load_pipe()
 dictA = {"Score": [], "SBS Code": [], "SBS Description V2.0": []}
 dfALL = pd.DataFrame.from_dict(dictA)
 
-if INTdesc_input is not None and createSBScodes_clicked == True: 
+if INTdesc_input is not None and mapSBS_button == True: 
     for i, result in enumerate(HF_model_results_displayed):
         dictA.update({"Score": "%.4f" % result[0]["score"], "SBS Code": df_SBS.loc[df_SBS["Long_Description"] == SBScorpus[result[0]["corpus_id"]],"SBS_Code_Hyphenated"].values[0], "SBS Description V2.0": SBScorpus[result[0]["corpus_id"]]})
         dfALL = pd.concat([dfALL, pd.DataFrame([dictA])], ignore_index=True)