Update app.py

app.py

@@ -48,8 +48,8 @@ async def lifespan(app: FastAPI):
     
         # Load models into the shared dictionary
         ml_models["embedder"] = HuggingFaceEmbeddings(model_name="BAAI/bge-base-en-v1.5")
-        # cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
-        cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
+        cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-base")
+        # cross_encoder_model = HuggingFaceCrossEncoder(model_name="BAAI/bge-reranker-large")
         ml_models["reranker_compressor"] = CrossEncoderReranker(model=cross_encoder_model, top_n=5)
         ml_models["llm"] = ChatGoogleGenerativeAI(model="gemini-1.5-flash", api_key=GOOGLE_API_KEY)
         ml_models["prompt_template"] = PromptTemplate.from_template(
@@ -63,6 +63,10 @@ If not, the raw question is provided instead:
 We retrieved the following policy clauses and rules relevant to this case:
 {context}
 
+Remember the below three points while answering or responsing the query :
+1:Do  not mention document id or its page number just ,mention clauses if applicable .
+2:Do not make such statement in the response that you need more document or information to answer the query, just answer from retrieved data.
+3:UNDERSTAND THE QUERY ASKED SMARTLY AND ANSWER IT.
 ### Task:
 If the details about age , gender , procedure , location , policy duration are available, do all of the following:
 1. Decide whether the procedure is covered.
@@ -70,7 +74,7 @@ If the details about age , gender , procedure , location , policy duration are a
 3. Justify with the relevant clause.
 and answer the query  precisely as insurance agent.
 
-Otherwise, answer the question concisely and clearly using the retrieved context.
+Otherwise, answer the question concisely , relevently and clearly using the retrieved context.
 
 ### Output format:
 If query involes age , gender , procedure , location , policy duration answer like below:
@@ -78,16 +82,14 @@ If query involes age , gender , procedure , location , policy duration answer li
   "decision": "approved / rejected",
   "amount": "INR amount or null",
   "justification": "Refer to specific clause"
-  Make it a perfect and concise.
+  Make it  concise ,perfect, relevent and clear .
 }}
 Else:
 {{
   "response": "Concise natural language answer"
-  Make it a perfect and concise.
+   Make it  concise ,perfect, relevent and clear .
 }}
-
-##NOTE : Do  not mention document id or its page number just mention clauses if applicable .
-        """
+"""
     )
         print("✅ Models and prompt loaded successfully!")
     except Exception as e: