Update app.py

app.py

@@ -44,12 +44,11 @@ def get_vector_store(chunks):
 
 # ✅ Conversational chain
 def get_conversational_chain():
-    prompt_template = """
-    شما یک دستیار هوشمند هستید. لطفاً فقط بر اساس متن موجود در زمینه پاسخ دهید. 
-    اگر پاسخ دقیق در متن نبود، بگویید: "اطلاعات کافی در متن موجود نیست".
-    از اضافه‌گویی یا ساختن جواب خودداری کنید.
+    map_prompt_template = """
+    لطفاً بر اساس متن زیر فقط به سؤال پاسخ دهید.
+    اگر جواب دقیق نبود، بگویید: "اطلاعات کافی در متن موجود نیست".
 
-    --- زمینه:
+    --- متن:
     {context}
 
     --- سوال:
@@ -57,12 +56,34 @@ def get_conversational_chain():
 
     --- پاسخ:
     """
+    combine_prompt_template = """
+    شما یک دستیار هوشمند هستید. پاسخ نهایی را بر اساس نتایج تیکه‌های مختلف بنویس.
+    اگر پاسخی در متن نبود، بگویید: "اطلاعات کافی در متن موجود نیست".
+
+    --- نتایج جزئی:
+    {summaries}
+
+    --- سوال:
+    {question}
+
+    --- پاسخ نهایی:
+    """
+
     model = ChatGoogleGenerativeAI(model="gemini-2.5-pro", client=genai, temperature=0.3)
-    prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
-    chain = load_qa_chain(llm=model, chain_type="map_reduce", prompt=prompt)
+
+    map_prompt = PromptTemplate(template=map_prompt_template, input_variables=["context", "question"])
+    combine_prompt = PromptTemplate(template=combine_prompt_template, input_variables=["summaries", "question"])
+
+    chain = load_qa_chain(
+        llm=model,
+        chain_type="map_reduce",
+        map_prompt=map_prompt,
+        combine_prompt=combine_prompt
+    )
     return chain
 
 
+
 # ✅ Clear history
 def clear_chat_history():
     st.session_state.messages = [{"role": "assistant", "content": "در خدمتیم"}]