Update app.py

app.py

@@ -118,22 +118,18 @@ def main():
     
     script_dir = os.path.dirname(os.path.abspath(__file__))
     data_path = os.path.join(script_dir, "data/")
-    if not os.path.exists(data_path):
-        st.error(f"Data path does not exist: {data_path}")
-        return
+    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)
+    
     
-    try:
-        # Load documents from the data path
-        documents = load_txt_documents(data_path)
-        if not documents:
-            st.warning("No documents found in the data path.")
-        else:
-            # Split the documents into chunks
-            docs = split_docs(documents, 350, 40)
-            # Add your logic here to use `docs`
-            st.success("Documents loaded and processed successfully.")
-    except Exception as e:
-        st.error(f"An error occurred while loading documents: {e}")
+    docs = split_docs(documents, 350, 40)