Update app.py

app.py

@@ -8,15 +8,19 @@ import streamlit as st
 embedding_function = SentenceTransformerEmbeddings(model_name="all-mpnet-base-v2")
 chdb = Chroma(persist_directory="./chroma_db_info", embedding_function=embedding_function)
 
-text = st.text_area("enter text")
+query = st.text_area("enter text")
+disnum = 3
 if text:
-    docs = chdb.similarity_search_with_score(text, k=3)
+    docs = chdb.similarity_search_with_score(query)
     docnum = len(docs)
     index = 0
-    ret = ''
+    ret = f"Query:{query}\n"
     for ii in range(docnum):    
         doc = docs[ii][0]
         score = docs[ii][1]
         ret += f"Return {index} ({score:.4f}) :\n{doc.page_content}\n"
+        index += 1
+        if index > disnum:
+            break
     st.text(ret)