Update app.py

app.py

@@ -82,21 +82,23 @@ class TogetherEmbeddings(Embeddings):
 @st.cache_resource
 def get_pdf_index():
     with st.spinner('📄 در حال پردازش فایل PDF...'):
-        loader = PyPDFLoader('test1.pdf')
-        documents = loader.load()
+        loader = [PyPDFLoader('test1.pdf')]
+        embeddings = TogetherEmbeddings(
+            model_name="togethercomputer/m2-bert-80M-8k-retrieval",
+            api_key="0291f33aee03412a47fa5d8e562e515182dcc5d9aac5a7fb5eefdd1759005979"
+
 
-        text_splitter = RecursiveCharacterTextSplitter(chunk_size=300, chunk_overlap=0)
-        texts = text_splitter.split_documents(documents)
 
-        embeddings = HuggingFaceEmbeddings(
-            model_name="HooshvareLab/bert-fa-zwnj-base"
-        )
 
-        index = FAISS.from_documents(texts, embeddings)
-        return index
+        )
+        return VectorstoreIndexCreator(
+            embedding=embeddings,
+            text_splitter=RecursiveCharacterTextSplitter(chunk_size=300, chunk_overlap=0)
+        ).from_loaders(loader)
 
 index = get_pdf_index()
 
+
 llm = HuggingFaceHub(
     repo_id="HooshvareLab/bert-fa-zwnj-base",  
     model_kwargs={"temperature": 0.7, "max_new_tokens": 512}