Update app.py

app.py

@@ -65,10 +65,9 @@ llm = GroqLLM(api_key=groq_api_key, model_name="deepseek-r1-distill-llama-70b")
 
 # ----------------- ساخت SimpleRetriever -----------------
 class SimpleRetriever(BaseRetriever):
-    def __init__(self, documents, embeddings):
-        documents, embeddings = build_pdf_index()
-        self.documents = documents
-        self.embeddings = embeddings
+    def __init__(self):
+        self.documents, self.embeddings = build_pdf_index()
+
 
     def _get_relevant_documents(self, query):
         query_embedding = nlp(query).vector  # تبدیل سوال به امبدینگ با استفاده از spaCy
@@ -82,7 +81,7 @@ class SimpleRetriever(BaseRetriever):
         return [doc for _, doc in ranked_docs[:5]]  # بازگرداندن 5 مستند مشابه
 
 
-retriever = SimpleRetriever(documents, embeddings)
+retriever = SimpleRetriever()
 
 chain = RetrievalQA.from_chain_type(
     llm=llm,