Update app.py

app.py

@@ -30,39 +30,19 @@ def load_embeddings():
 
 hf = load_embeddings()
 
-@st.cache_data
-def load_txt_documents(data_path):
-    documents = []
-    for filename in os.listdir(data_path):
-        if filename.endswith('.txt'):
-            file_path = os.path.join(data_path, filename)
-            documents.extend(TextLoader(file_path).load())
-    return documents
 
-@st.cache_data
-def load_uploaded_documents(uploaded_files):
-    documents = []
-    for uploaded_file in uploaded_files:
-        content = uploaded_file.read().decode("utf-8")
-        documents.append({"content": content, "filename": uploaded_file.name})
-    return documents
 
 
 
 documents = load_documents(data_path)
 docs = split_docs(documents, 450, 20)
 
-@st.cache_resource
-def create_chroma_db(docs, hf):
-    return Chroma(docs, hf)
 
-chroma_db = create_chroma_db(docs, hf)
 
-@st.cache_resource
-def create_retriever(chroma_db):
-    return mmr_retriever(chroma_db, "mmr", 6)
+chroma_db = chroma_db(docs, hf)
+
 
-retriever = create_retriever(chroma_db)
+retriever = retriever_from_chroma(chroma_db,"mmr", 6)
 
 # Set up LlamaCpp model
 callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])