Check SNOMED CT description is FSN

Check SNOMED CT description is FSN (contain the parentheses "(" & ")" )

SNOMED-CT_Assistant.py

@@ -127,7 +127,7 @@ def generate_entity_mapping_prompt(entity, query_result_dict):
 def query_chroma_db(query_text, query_number):
     results = collection.query(
         query_texts=[query_text],
-        n_results=query_number,
+        n_results=query_number+20,
         include=["distances", "metadatas", "documents"]
     )
     return results
@@ -156,12 +156,12 @@ def chat_input(prompt, med_text):
         for entity in entity_list:
             results = query_chroma_db(entity, query_number)
             results_dict = get_dict_from_chroma_results(results)
-            results_table = entity_mapping_result_to_table(entity, results_dict)
+            results_table = entity_mapping_result_to_table(entity, results_dict, query_number)
             st.session_state.messages.append({"role": "assistant", "content": results_table})
             st.chat_message("assistant").write(results_table)
             
 # Conver entity mapping result to markdown table
-def entity_mapping_result_to_table(entity, results_dict):
+def entity_mapping_result_to_table(entity, results_dict, query_number):
     
     ids = results_dict['ids']
     concept_ids = results_dict['concept_ids']
@@ -175,8 +175,12 @@ def entity_mapping_result_to_table(entity, results_dict):
     # table
     rows = []
     for id, distance, concept_id, description in zip(ids, distances, concept_ids, descriptions):
+        if '(' not in description or ')' not in description:
+            continue
         row = f"| {entity} | {distance:.3f} | {id} | {concept_id} | {description} |"
         rows.append(row)
+        if len(rows) >= query_number:
+            break
     
     # merge
     markdown_table = "\n".join([header, seperator] + rows)