Update app/vector_store.py

app/vector_store.py

@@ -34,6 +34,14 @@ def store_pdf_image(pdf_path):
     Chroma.from_documents(chunks, embeddings, persist_directory=CHROMA_IMG_DIR)
 
 
+def store_pdf_image_text(text):
+    doc = Document(page_content=text)
+    splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=50)
+    chunks = splitter.split_documents([doc])
+    embeddings = SentenceTransformerEmbeddings(model_name='thenlper/gte-large')
+    Chroma.from_documents(chunks, embeddings, persist_directory=CHROMA_IMG_DIR)
+
+
 # images = convert_from_path("your_file.pdf", poppler_path="/opt/homebrew/bin")