Update app.py

app.py

@@ -242,14 +242,19 @@ 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
-    # Ensure the directory exists
-    os.makedirs("logs", exist_ok=True)
-    # Write the selected candidates to a file
-    with open("logs/selected_candidates.tsv", "a") as file:
-        file.write(f"{text}\t{str(candidates)}\t{str(selected_candidates)}\n")
-   
+    try:
+        # Ensure the directory exists
+        os.makedirs("logs", exist_ok=True)
+        # Write the selected candidates to a file
+        with open("logs/selected_candidates.tsv", "a") as file:
+            file.write(f"{text}\t{str(candidates)}\t{str(selected_candidates)}\n")
+        return True
+    except Exception as e:
+        st.error(f"An error occurred while writing the file: {e}")
+        return False
+        
 
 def run_client():
     with open(Path(__file__).parent / "style.css") as f:
@@ -361,39 +366,48 @@ def run_client():
                         response = relik_model.retrieve(text, k=10, batch_size=400, progress_bar=False)
                         candidates_text = [pred.document.text for pred in response[0]]
     
+        
                     if candidates_text:
-                        st.session_state.candidates = candidates_text[:10]
-
+                        st.session_state.candidates = candidates_text[:5]
+                        dict_of_ents_candidates, options_candidates = get_retriever_annotations_candidates(text, st.session_state.candidates)
+                        st.markdown("<h2 style='color: black;'>Possible Candidates:</h2>", unsafe_allow_html=True)
+        
+                        for candidate in dict_of_ents_candidates["ents"]:
+                            checked = candidate in st.session_state.selected_candidates
+                            if st.checkbox(candidate, key=candidate, value=checked):
+                                if candidate not in st.session_state.selected_candidates:
+                                    st.session_state.selected_candidates.append(candidate)
+                            else:
+                                if candidate in st.session_state.selected_candidates:
+                                    st.session_state.selected_candidates.remove(candidate)
+        
+                        if st.button("Save Selected Candidates"):
+                            if write_candidates_to_file(text, dict_of_ents_candidates["ents"], st.session_state.selected_candidates):
+                                st.success("Selected candidates have been saved to file.")
+                    else:
+                        st.session_state.candidates = []
+                        st.session_state.selected_candidates = []
+                        st.markdown("<h2 style='color: black;'>No Candidates Found</h2>", unsafe_allow_html=True)
+         
         else:
             st.error("Please enter some text.")
-
+    
+    # Ensure the candidates list is displayed even after interactions
     if st.session_state.candidates:
         dict_of_ents_candidates, options_candidates = get_retriever_annotations_candidates(text, st.session_state.candidates)
-
         st.markdown("<h2 style='color: black;'>Possible Candidates:</h2>", unsafe_allow_html=True)
-
-        # Display the candidates with checkboxes
         for candidate in dict_of_ents_candidates["ents"]:
-            if candidate in st.session_state.selected_candidates:
-                checked = True
-            else:
-                checked = False
+            checked = candidate in st.session_state.selected_candidates
             if st.checkbox(candidate, key=candidate, value=checked):
                 if candidate not in st.session_state.selected_candidates:
                     st.session_state.selected_candidates.append(candidate)
             else:
                 if candidate in st.session_state.selected_candidates:
                     st.session_state.selected_candidates.remove(candidate)
-
-        # 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"], st.session_state.selected_candidates)
-            st.success("Selected candidates have been saved to file.")
-
-    else:
-        st.session_state.candidates = []
-        st.session_state.selected_candidates = []
-        st.markdown("<h2 style='color: black;'>No Candidates Found</h2>", unsafe_allow_html=True)
+            if write_candidates_to_file(text, dict_of_ents_candidates["ents"], st.session_state.selected_candidates):
+                st.success("Selected candidates have been saved to file.")
 
  
 if __name__ == "__main__":