Update app.py

app.py

@@ -13,13 +13,15 @@ from langchain.document_loaders import PyPDFLoader
 from fastapi.encoders import jsonable_encoder
 from dotenv import load_dotenv
 
-load_dotenv()
-logging.basicConfig(level=logging.DEBUG)
+#load_dotenv()
+#logging.basicConfig(level=logging.DEBUG)
 
 ABS_PATH = os.path.dirname(os.path.abspath(__file__))
 DB_DIR = os.path.join(ABS_PATH, "db")
 
+vectorstore = None
 #embedding_function
+embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
 
 def replace_newlines_and_spaces(text):
     # Replace all newline characters with spaces
@@ -77,10 +79,10 @@ def query_chromadb():
 
     # Select which embeddings we want to use
     #embeddings = OpenAIEmbeddings()
-    embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
+    #embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
 
     # Load Vector store from local disk
-    vectorstore = Chroma(persist_directory=DB_DIR, embedding_function=embeddings)
+    #vectorstore = Chroma(persist_directory=DB_DIR, embedding_function=embeddings)
     #vectorstore.persist()
 
     result = vectorstore.similarity_search_with_score(query="how to use AI in water conservation?", k=4)