Update app.py

app.py

@@ -361,31 +361,32 @@ def run_client():
                     if candidates_text:
                         st.session_state.candidates = candidates_text[:10]
     
-                        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 st.checkbox(candidate, key=candidate):
-                                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)
-    
+
         else:
             st.error("Please enter some text.")
 
+    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 st.checkbox(candidate, key=candidate):
+                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.")
 
 if __name__ == "__main__":
     run_client()