Add query_number with slider.

SNOMED-CT_Assistant.py

@@ -23,6 +23,7 @@ else:
 
 st.title("🏥 SNOMED-CT Assistant")
 st.caption("👩‍⚕️ A smart medical assistant with SNOMED-CT knowledge.")
+query_number = st.sidebar.slider("Query Numbers", 5, 20, 5)
 
 # System prompt
 system_prompt = """You are a medical expert with rich experience in SNOMED-CT professional knowledge. 
@@ -145,7 +146,7 @@ def chat_input(prompt, med_text):
         st.session_state.messages.append({"role": "assistant", "content": msg})
         st.chat_message("assistant").write(msg)
         for entity in entity_list:
-            results = query_chroma_db(entity, 10)
+            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)
             st.session_state.messages.append({"role": "assistant", "content": results_table})