Update The Model issues and Prompt

llm.py

@@ -30,27 +30,27 @@ def query_gemini(questions, contexts):
         questions_text = "\n".join([f"{i+1}. {q}" for i, q in enumerate(questions)])
         
         prompt = f"""
-You are an intelligent insurance assistant trained to answer questions using insurance documents. Based on the context provided below, respond to each question with a **well-informed, complete, and professionally worded answer**.
+You are a skilled insurance policy assistant. Based only on the provided context, answer each question clearly and briefly.
 
-🎯 SCORING & OUTPUT GOAL:
-- Responses are part of an evaluated system.
-- Each answer should be **accurate**, **complete**, and **well-phrased** — ideally around **1–2 full sentences**.
-- Avoid short/fragmented answers or long multi-paragraph explanations.
-- Always write like an insurance advisor addressing a customer clearly.
+🎯 GOAL:
+- Each answer should be precise, informative, and phrased naturally — around 1–2 full sentences.
+- Avoid long clauses or repeating policy language verbatim.
+- Focus on **clarity**, **brevity**, and **accuracy** — your answers will be judged for quality and relevance.
 
 📘 INSTRUCTIONS:
-1. **Only use the provided context** to answer each question. If the answer is not found, respond with exactly: `"Not Found"`.
-2. Keep answers concise **but not vague**. Include all **key points** (such as limits, durations, conditions) in one or two complete sentences.
-3. DO NOT include any section numbers like "as per section 3.1.6"— these references must be omitted from the final answers.
-4. DO NOT use bullet points, partial phrases, or excessive legal text. DO NOT repeat the question in the answer.
-5. Match the tone and format of these examples:
-   - "A grace period of thirty days is provided for premium payment after the due date to renew or continue the policy without losing continuity benefits."
-   - "Yes, the policy covers maternity expenses, including childbirth and lawful medical termination of pregnancy. To be eligible, the female insured person must have been continuously covered for at least 24 months. The benefit is limited to two deliveries or terminations during the policy period."
-   - "Yes, the policy indemnifies the medical expenses for the organ donor's hospitalization for the purpose of harvesting the organ, provided the organ is for an insured person and the donation complies with the Transplantation of Human Organs Act, 1994."
+1. Only use the context below. If the answer is not present, respond with exactly: "Not Found".
+2. Answers must be helpful and compact — **complete, but not wordy**. Avoid long multi-clause sentences.
+3. Do NOT include section references (like “as per section 3.1.6”) or legal codes.
+4. Avoid copying large phrases from the document. Rephrase naturally using simpler language where appropriate.
+5. Follow the tone and length of these examples:
+
+   - "A grace period of thirty days is provided for premium payment after the due date."
+   - "Yes, maternity expenses are covered after 24 months of continuous coverage, limited to two deliveries or terminations."
+   - "Yes, the policy covers organ donor expenses for harvesting the organ if donated to the insured person."
    - "Not Found"
 
 📤 RETURN FORMAT:
-Respond strictly using this JSON structure:
+Respond in the exact JSON format below — no extra text or explanations.
 
 {{
   "answers": [
@@ -66,12 +66,13 @@ Respond strictly using this JSON structure:
 ❓ QUESTIONS:
 {questions_text}
 
-Your task: Provide accurate, refined answers based on the document context above. Use the tone and structure shown. Be concise but thorough. Only include what is supported in the context. Use "Not Found" if the answer is missing.
+Your task: Answer each question concisely and professionally. Use plain phrasing, stay within 1–2 clear sentences, and avoid unnecessary detail or repetition.
 """
 
 
 
 
+
         prompt_time = time.time() - prompt_start
         print(f"Prompt preparation took: {prompt_time:.2f} seconds")
         print(f"Total prompt length: {len(prompt)} characters")