Update pages/type_text.py

pages/type_text.py

@@ -84,7 +84,6 @@ dfALL = pd.DataFrame.from_dict(dictA)
 
 if INTdesc_input is not None and createSBScodes_clicked == True: 
     for i, result in enumerate(HF_model_results_displayed):
-    #for result in 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)
         dictA.update({"Score": "%.4f" % result[1]["score"], "SBS Code": df_SBS.loc[df_SBS["Long_Description"] == SBScorpus[result[1]["corpus_id"]],"SBS_Code_Hyphenated"].values[0], "SBS Description V2.0": SBScorpus[result[1]["corpus_id"]]})
@@ -103,7 +102,7 @@ if INTdesc_input is not None and createSBScodes_clicked == True:
     question = "Which one, if any, of the below Saudi Billing System descriptions A, B, C, D, or E corresponds best to " + INTdesc_input +"? " 
     shortlist = [SBScorpus[result[0]["corpus_id"]], SBScorpus[result[1]["corpus_id"]], SBScorpus[result[2]["corpus_id"]], SBScorpus[result[3]["corpus_id"]], SBScorpus[result[4]["corpus_id"]]] 
     prompt = [question + " " +"A: "+ shortlist[0] + " " +"B: " + shortlist[1] + " " + "C: " + shortlist[2] + " " + "D: " + shortlist[3] + " " + "E: " + shortlist[4]]
-    st.write(prompt)
+    st.write(prompt[0])
     
     messages = [
     {"role": "system", "content": "You are a knowledgable AI assistant who always answers truthfully and precisely!"},