app top n content added

app.py

@@ -169,7 +169,12 @@ def main():
     if uploaded_file is not None:
         # Select token size:
         st.sidebar.success("Note: 1 Token ~ 4 Characters.")
-        token_size = st.sidebar.slider('Select a token size (when we scrape the document)', 5, 150, 45)
+        token_size = st.sidebar.slider(
+            "Select a token size (when we scrape the document)", 5, 150, 45
+        )
+        top_n_content = st.sidebar.slider(
+            "Select top n content(s) you want to display as reference", 3, 30, 5
+        )
 
         # To read file as bytes:
         bytes_data = uploaded_file.getvalue()
@@ -233,7 +238,7 @@ def main():
 
         # User input
         query = st.text_input("Ask me anything!", "What is the document about?")
-        results = chroma_collection.query(query_texts=[query], n_results=5)
+        results = chroma_collection.query(query_texts=[query], n_results=top_n_content)
         retrieved_documents = results["documents"][0]
         results_as_table = pd.DataFrame(
             {