Update app.py

app.py

@@ -103,8 +103,17 @@ def main():
     ]
 
     st.markdown("Hi, I am Birute, chat assistant, based on republic of Lithuania law documents. You can choose below information retrieval type and how many documents you want to be retrieved.")
-    search_type = st.selectbox("Choose search type. Options are [Max marginal relevance search(mmr) , Similarity search (similarity). Default value (mmr)]", options =["mmr", "similarity"], index ="mmr")
-    k = st.select_slider("Select amount of documents to be retrieved. Default value(5) : ", options=[2, 15], value = 5)
+    search_type = st.selectbox(
+        "Choose search type. Options are [Max marginal relevance search (mmr) , Similarity search (similarity). Default value (mmr)]", 
+        options=["mmr", "similarity"], 
+        index=0  # Default to "mmr"
+    )
+
+    k = st.select_slider(
+        "Select amount of documents to be retrieved. Default value (5): ", 
+        options=list(range(2, 16)),  # Creates a list [2, 3, 4, ..., 15]
+        value=5  # Default value
+    )
     retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=450, chunk_overlap=20)
     
     if user_question := st.text_input("Ask a question about your documents:"):