Update app.py

app.py

@@ -110,22 +110,15 @@ def main():
 
     
     retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='mmr', k=9, chunk_size=250, chunk_overlap=20)
-    user_question = st.text_input("Ask a question about your documents:")
-    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?"}
-    ]    
-
-    for msg in st.session_state.messages:
-        st.chat_message(msg["role"]).write(msg["content"])
+    if user_question := st.text_input("Ask a question about your documents:"):
+        handle_userinput(user_question,retriever)
+        
+    
     
     
     
  
-    if user_question:
-        handle_userinput(user_question,retriever)
-
-
+