Update app.py

app.py

@@ -29,17 +29,22 @@ def create_embeddings():
     return embeddings
 
 # Function to create vector store
-def create_vector_store(text_chunks, embeddings):
+def create_vector_store(text_chunks, embeddings, nombre_vector):
     vector_store = FAISS.from_documents(text_chunks, embeddings)
+    vector_store.save_local("cache") #Guardarlo en un 
     return vector_store
 
+# Function to create vector store
+def load_vector_store(nombre_vector, embeddings):
+    return FAISS.load_local(nombre_vector, embeddings)
+
 # Function to create LLMS model
 def create_llms_model():
     llm = CTransformers(model='TheBloke/Mistral-7B-Instruct-v0.1-GGUF', config={'max_new_tokens': 128, 'temperature': 0.01})
     return llm
 
 # Initialize Streamlit app
-st.title("Chatbot usando mistral, solo test")
+st.title("Chatbot usando mistral")
 
 # loading of documents
 documents = load_documents()
@@ -50,9 +55,10 @@ text_chunks = split_text_into_chunks(documents)
 # Create embeddings
 embeddings = create_embeddings()
 
-# Create vector store
-vector_store = create_vector_store(text_chunks, embeddings)
-
+try:#load vector store from local
+    vector_store = load_vector_store("cache",embeddings)
+except:# Create vector store
+    vector_store = create_vector_store(text_chunks, embeddings, "cache")
 # Create LLMS model
 llm = create_llms_model()