some minor ui fix

RBotReloaded.py

@@ -175,14 +175,11 @@ class StorageRetrievalLLM:
     
     # Load file
     file_path = doc if os.path.exists(doc) else os.path.join("data", doc)  
+    self.stored_pages_folder
     # loader = DirectoryLoader(file_path, glob="**/*.pdf", loader_cls=PyMuPDFLoader)       
-    # documents = loader.load()
-    loader = TextLoader(path=file_path, loader_cls=PyMuPDFLoader)
-    documents = loader.load()    
-    
-    # Split and add
-    splitter = RecursiveCharacterTextSplitter()
-    documents = splitter.split_documents(documents)  
+    # documents = loader.load()    
+    loader = PyPDFLoader(file_path)
+    documents = loader.load_and_split()
     self.vectorstore.add_documents(documents)
     
     # Update chain
@@ -198,12 +195,8 @@ class StorageRetrievalLLM:
   def addTextFileToMemory(self, file_path : str, summarize = True):
     
     # Load file
-    loader = TextLoader(path=file_path, loader_cls=PyMuPDFLoader)
-    documents = loader.load()
-    
-    # Split and add
-    splitter = RecursiveCharacterTextSplitter()            
-    documents = splitter.split_documents(documents)
+    loader = PyPDFLoader(file_path)
+    documents = loader.load_and_split()
     self.vectorstore.add_documents(documents)
     
     # Update chain

agent_llama_ui.py

@@ -185,6 +185,14 @@ def render_simple_chat():
 
     uploaded_file = st.sidebar.file_uploader("Drag and Drop a File to ./knowledge_base/", type=["txt", "pdf", "docx"])     
     
+    knowledge_files = glob.glob(f"./knowledge_base/*.*") 
+    st.sidebar.subheader("Knowledge Files")
+    for file_path in knowledge_files:      
+        if not "index." in file_path.lower():
+            file_path = file_path.replace("\\", "/")
+            file_name = file_path.split("/")[-1]
+            st.sidebar.markdown(f"[{file_name}](/{file_path})", unsafe_allow_html=True)
+        
     if st.sidebar.button("Reset Long Term Memory", disabled=not (current_agent() is not None and get_index_size() > 0)) and current_agent() is not None:
         current_agent().reset_knowledge()