Update app.py

app.py

@@ -19,7 +19,8 @@ from HTML_templates import css, bot_template, user_template
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.runnables import RunnablePassthrough
 from langchain import hub
-
+from langchain.retrievers import ContextualCompressionRetriever
+from langchain.retrievers.document_compressors import LLMChainExtractor
 
 lang_api_key = os.getenv("lang_api_key")
 
@@ -31,7 +32,7 @@ os.environ["LANGCHAIN_PROJECT"] = "Chat with multiple PDFs"
 
 
 def create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='mmr', k=7, chunk_size=250, chunk_overlap=20):
-    data_path = "data"
+    data_path = "data/"
     model_name = "Alibaba-NLP/gte-base-en-v1.5"
     model_kwargs = {'device': 'cpu',
                    "trust_remote_code" : 'True'}
@@ -85,6 +86,13 @@ def create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='m
     
     retriever=vectorstore.as_retriever(search_type = search_type, search_kwargs={"k": k})
 
+    compressor = LLMChainExtractor.from_llm(llm)
+
+    compression_retriever = ContextualCompressionRetriever(
+    base_compressor=compressor,
+    base_retriever=retriever
+)
+    
     rephrase_prompt = hub.pull("langchain-ai/chat-langchain-rephrase")
 
     ha_retriever = create_history_aware_retriever(llm, retriever, rephrase_prompt)