Update app.py

app.py

@@ -16,35 +16,42 @@ from utills import load_txt_documents, split_docs, load_uploaded_documents, retr
 from langchain.text_splitter import TokenTextSplitter, RecursiveCharacterTextSplitter
 from langchain_community.document_loaders.directory import DirectoryLoader
 
-def main():
-    st.set_page_config(page_title="Conversational RAG Chatbot", page_icon=":robot:")
-    st.title("Conversational RAG Chatbot")
-
-    if "documents" not in st.session_state:
-        st.session_state.documents = []
-
-    if "conversation_chain" not in st.session_state:
-        st.session_state.conversation_chain = None
-
-    data_path = "data"
+data_path = "data"
 
 
     
-    documents = []
+documents = []
     
-    for filename in os.listdir(data_path):
+for filename in os.listdir(data_path):
 
-        if filename.endswith('.txt'):
+    if filename.endswith('.txt'):
 
-            file_path = os.path.join(data_path, filename)
+        file_path = os.path.join(data_path, filename)
 
-            documents = TextLoader(file_path).load()
+        documents = TextLoader(file_path).load()
 
-            documents.extend(documents)
+        documents.extend(documents)
     
     
     docs = split_docs(documents, 350, 40)
     vectorstore = get_vectorstore(docs)
+
+
+
+
+
+
+def main(vectorstore):
+    st.set_page_config(page_title="Conversational RAG Chatbot", page_icon=":robot:")
+    st.title("Conversational RAG Chatbot")
+
+    if "documents" not in st.session_state:
+        st.session_state.documents = []
+
+    if "conversation_chain" not in st.session_state:
+        st.session_state.conversation_chain = None
+
+    
     if "conversation_chain" not in st.session_state:
         st.session_state.conversation_chain = create_conversational_rag_chain(vectorstore)