Update app.py

app.py

@@ -82,6 +82,14 @@ def main():
 
     user_question = st.text_input("Ask a question about your documents:")
 
+    
+    
+    retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20)
+ 
+    if user_question:
+        handle_userinput(user_question,retriever)
+
+
     if "messages" not in st.session_state:
         st.session_state["messages"] = [
         {"role": "assistant", "content": "Hi, I'm a chatbot who is based on lithuanian law documents. How can I help you?"}
@@ -89,11 +97,6 @@ def main():
 
     for msg in st.session_state.messages:
         st.chat_message(msg["role"]).write(msg["content"])
-    
-    retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20)
- 
-    if user_question:
-        handle_userinput(user_question,retriever)
 
 
 def handle_userinput(user_question,retriever):