Update app.py

app.py

@@ -15,6 +15,8 @@ from sentence_transformers import SentenceTransformer
 import faiss
 from langchain.indexes import VectorstoreIndexCreator
 from langchain.vectorstores import FAISS
+from langchain.embeddings import SentenceTransformerEmbeddings
+
 
 # ----------------- تنظیمات صفحه -----------------
 st.set_page_config(page_title="چت‌ بات توانا", page_icon="🪖", layout="wide")
@@ -99,24 +101,20 @@ st.markdown("""
 
 # ----------------- لود PDF و ساخت ایندکس -----------------
 
+# ----------------- لود PDF و ساخت ایندکس -----------------
 @st.cache_resource
 def get_pdf_index():
     with st.spinner('📄 در حال پردازش فایل PDF...'):
         loader = PyPDFLoader('test1.pdf')
-        # documents = loader.load()
 
         splitter = RecursiveCharacterTextSplitter(chunk_size=2000, chunk_overlap=128)
-        # texts = []
-        # for doc in documents:
-        #     texts.extend(splitter.split_text(doc.page_content))
-
 
         embedding_function = SentenceTransformer("togethercomputer/m2-bert-80M-8k-retrieval", trust_remote_code=True)
-        
-        # ctorstore_index_creator.from_documents([Document(page_content=text) for text in texts])
+
+        embedding = SentenceTransformerEmbeddings(model=embedding_function)
 
         return VectorstoreIndexCreator(
-            embedding=embedding_function,
+            embedding=embedding,
             text_splitter=splitter
         ).from_loaders(loader)