Update app.py

app.py

@@ -1,7 +1,10 @@
 import gradio as gr
 from langchain_community.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain_chroma import Chroma
+# from langchain_chroma import Chroma
+from langchain_community.vectorstores import FAISS
+
+
 from langchain_groq import ChatGroq
 from langchain.chains import create_retrieval_chain
 from langchain.chains.combine_documents import create_stuff_documents_chain
@@ -67,11 +70,16 @@ class PDFQAProcessor:
             
             splits = self.text_splitter.split_documents(docs)
 
-            # Create vector store
-            self.vectorstore = Chroma.from_documents(
-                documents=splits,
-                embedding=self.embeddings,
-                # persist_directory = self.persist_directory
+            # # Create vector store
+            # self.vectorstore = Chroma.from_documents(
+            #     documents=splits,
+            #     embedding=self.embeddings,
+            #     # persist_directory = self.persist_directory
+            # )
+            # Replace Chroma with:
+            self.vectorstore = FAISS.from_documents(
+                splits,
+                self.embeddings
             )
             self.retriever = self.vectorstore.as_retriever(search_kwargs={"k": 10})
             return "PDFs processed successfully! Ask your questions now."