Update app.py

app.py

@@ -28,22 +28,23 @@ def main():
 
     data_path = "data"
 
-    if not os.path.exists(data_path):
-        st.error(f"Data path does not exist: {data_path}")
-        return
-
-    try:
-        documents = load_txt_documents(data_path)
-        if not documents:
-            st.warning("No documents found in the data path.")
-        else:
-            st.session_state.documents = documents
-            docs = split_docs(documents, 350, 40)
-            vectorstore = retriever_from_chroma(docs, HuggingFaceEmbeddings(), "mmr", 7)
-            st.session_state.conversation_chain = create_conversational_rag_chain(vectorstore)
-            st.success("Documents loaded and processed successfully.")
-    except Exception as e:
-        st.error(f"An error occurred while loading documents: {e}")
+
+    
+    documents = []
+    
+    for filename in os.listdir(data_path):
+
+        if filename.endswith('.txt'):
+
+            file_path = os.path.join(data_path, filename)
+
+            documents = TextLoader(file_path).load()
+
+            documents.extend(documents)
+    
+    
+    docs = split_docs(documents, 350, 40)
+    
 
     if prompt := st.text_input("Enter your question:"):
         msgs = st.session_state.get("chat_history", StreamlitChatMessageHistory(key="special_app_key"))