Update app.py

app.py

@@ -152,12 +152,29 @@ retriever = SimpleRetriever(
     index=index,
     model=model
 )
+custom_prompt = PromptTemplate(
+    input_variables=["context", "question"],
+    template="""
+شما فقط مجاز هستید از اطلاعات زیر پاسخ دهید. اگر اطلاعات لازم برای پاسخ وجود ندارد، بگویید: "متاسفم، اطلاعات لازم در فایل موجود نیست."
+
+اطلاعات:
+{context}
+
+سوال:
+{question}
+
+پاسخ دقیق و کامل بده:
+"""
+)
+
+
 
 # ----------------- ساخت Chain -----------------
 qa_chain = RetrievalQA.from_chain_type(
     llm=llm,
     retriever=retriever,
-    chain_type="stuff"
+    chain_type="stuff",
+    chain_type_kwargs={"prompt": custom_prompt}
 )
 
 # ----------------- چت استیت -----------------