Update app.py

app.py

@@ -242,6 +242,12 @@ def set_intro(css):
     #     "[Stefano Faralli](https://corsidilaurea.uniroma1.it/it/users/stefanofaralliuniroma1it), and [Roberto Navigli](https://www.diag.uniroma1.it/navigli/)."
     # )
 
+def write_candidates_to_file(text, candidates, selected_candidates):
+    # Write the selected candidates to a file
+    with open("logs/selected_candidates.tsv", "a") as file:
+        file.write(text + "\t" + str(candidates) + str(selected_candidates) + "\n")
+
+
 def run_client():
     with open(Path(__file__).parent / "style.css") as f:
         css = f.read()
@@ -341,7 +347,7 @@ def run_client():
                     st.markdown(text, unsafe_allow_html=True)
                 else:
                     if selection_list == "Top-k DB in Taxonomy Intervention" or selection_list == "Top-k DB in Taxonmy Outcome":
-                        response = relik_model.retrieve(text, k=100, batch_size=400, progress_bar=False)
+                        response = relik_model.retrieve(text, k=30, batch_size=400, progress_bar=False)
                            
                         candidates_text = []
                         for pred in response[0]:
@@ -361,13 +367,24 @@ def run_client():
     
                         dict_of_ents_candidates, options_candidates = get_retriever_annotations_candidates(text, candidates_text)
     
-                        
-                        text = """
-                        <h2 style='color: black;'>Possible Candidates:</h2>
-                        <ul style='color: black;'>
-                        """ + "".join(f"<li style='color: black;'>{candidate}</li>" for candidate in dict_of_ents_candidates["ents"]) + "</ul>"
+
+                        text = "<h2 style='color: black;'>Possible Candidates:</h2>"
+                        #<h2 style='color: black;'>Possible Candidates:</h2>
+                        # <ul style='color: black;'>
+                        # """ + "".join(f"<li style='color: black;'>{candidate}</li>" for candidate in dict_of_ents_candidates["ents"]) + "</ul>"
         
                         st.markdown(text, unsafe_allow_html=True)
+                       # Multiselect widget for selecting candidates
+                        selected_candidates = st.multiselect(
+                            "Select Candidates:",
+                            dict_of_ents_candidates["ents"],  # Options for the multiselect
+                            help="Select one or more candidates from the list."
+                        )
+                       # Button to save the selected candidates to a file
+                        if st.button("Save Selected Candidates"):
+                            write_candidates_to_file(text, dict_of_ents_candidates["ents"], selected_candidates)
+                            st.success("Selected candidates have been saved to file.")
+
                     else:
                         
                         text = "<h2 style='color: black;'>No Candidates Found</h2>"