Update app.py

app.py

@@ -29,6 +29,30 @@ memory = ConversationBufferWindowMemory(
 
 
 
+
+
+query_text="what did alice say to rabbit"
+
+# Prepare the DB.
+#embedding_function = OpenAIEmbeddings() # main
+
+CHROMA_PATH = "chroma8"
+# call the chroma generated in a directory
+db = Chroma(persist_directory=CHROMA_PATH, embedding_function=embeddings)
+
+# Search the DB for similar documents to the query.
+results = db.similarity_search_with_relevance_scores(query_text, k=2)
+if len(results) == 0 or results[0][1] < 0.5:
+    print(f"Unable to find matching results.")
+
+
+
+
+
+
+
+
+
 from langchain import PromptTemplate
 query_text = "when did alice see mad hatter"