Update appStore/prep_utils.py

appStore/prep_utils.py

@@ -1,8 +1,10 @@
 from langchain_text_splitters import TokenTextSplitter
 from langchain.docstore.document import Document
 import pandas as pd
-
+from qdrant_client import QdrantClient
+path_to_qdrant = "/data/local_qdrant"
 path_to_data = "./docStore/"
+import streamlit as st
 
 def create_chunks(text, tokens_count =400, overlap_count=0):
     """TAKES A TEXT AND CERATES CREATES CHUNKS"""
@@ -20,4 +22,12 @@ def create_documents(df, text_column):
         metadata = {col: str(df.loc[i, col]) for col in list(set(df.columns) - {text_column})}
         placeholder.append(Document(page_content= df.loc[i,text_column], 
                                 metadata=metadata))
-    return placeholder    
+    return 
+    
+@st.cache_resource
+def get_client():
+    if 'qdrant' in st.session_state:
+        return st.session_state.qdrant
+    else:
+        client = QdrantClient(path=path_to_qdrant)
+        return client