Add Bangla model API

rag.py

@@ -10,8 +10,24 @@ from config import VECTOR_DIR, EMBED_MODEL, GROQ_API_KEY, GROQ_MODEL
 memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
 def reset_memory():
+    """Clear previous conversation memory (use after ingesting a new PDF)"""
     memory.clear()
 
+def normalize_question(question: str) -> str:
+    """
+    Ensure the question starts with 'What' if not already starting
+    with Who/When/Where/Why/How/What.
+    """
+    question = question.strip()
+    if question == "":
+        return "What?"
+    
+    first_word = question.split()[0].lower()
+    wh_words = ["what", "who", "when", "where", "why", "how","tell me about"]
+    if first_word not in wh_words:
+        question = "What " + question[0].lower() + question[1:] if len(question) > 0 else "What?"
+    return question
+
 def build_chain():
     embeddings = HuggingFaceEmbeddings(model_name=EMBED_MODEL)
     vectordb = FAISS.load_local(VECTOR_DIR, embeddings, allow_dangerous_deserialization=True)
@@ -19,12 +35,12 @@ def build_chain():
 
     llm = ChatGroq(model=GROQ_MODEL, api_key=GROQ_API_KEY, temperature=0.1)
 
-    # ---- Few-shot examples for the model ----
+    # ---- System + Human Prompt with diverse few-shot examples ----
     chat_prompt = ChatPromptTemplate.from_messages([
         SystemMessagePromptTemplate.from_template(
             "You are a helpful financial expert assistant with 10 years of experience. "
             "Answer all questions using the context retrieved from the PDF. "
-            "Always include the reference (page or section) where the information was found."
+            "Always include reference (page or section) where the information was found."
         ),
         HumanMessagePromptTemplate.from_template(
             """Context:
@@ -33,17 +49,32 @@ def build_chain():
 Question:
 {question}
 
-Answer the question clearly and concisely. Always start answers with the question phrased as 'What', 'Who', etc., exactly as asked. 
-Include the reference in the format: Reference: page:X
+Answer the question clearly and concisely. Always start answers naturally (What, Who, When, Where, Why, How) and include references in the format: Reference: page:X or section name.
 
-Examples (few-shot):
+Few-shot examples:
 Question: What is the purpose of the financial policy objectives and strategies statement?
 Answer: The purpose of the financial policy objectives and strategies statement is to make transparent the Government’s financial strategies and to establish a benchmark for evaluating the Government’s conduct of financial policy.
 Reference: page:5
 
-Question: What are the key objectives of the fiscal policy?
-Answer: The key objectives of the fiscal policy are to maintain macroeconomic stability, promote growth, and ensure social equity.
-Reference: page:8
+Question: Who is responsible for preparing the annual budget?
+Answer: The Ministry of Finance is responsible for preparing the annual budget, ensuring alignment with government policies.
+Reference: page:12
+
+Question: When is the fiscal report published each year?
+Answer: The fiscal report is published annually in September to provide transparency on government revenues and expenditures.
+Reference: page:20
+
+Question: Where can one find the details of government subsidies in the document?
+Answer: Details of government subsidies are provided in the fiscal policy section under the heading 'Subsidies and Support Programs'.
+Reference: page:15
+
+Question: Why is the financial risk management framework important?
+Answer: The financial risk management framework is important to identify, assess, and mitigate potential risks in government financial operations.
+Reference: page:18
+
+Question: How does the government evaluate financial policy effectiveness?
+Answer: The government evaluates financial policy effectiveness using key performance indicators and benchmarking against fiscal objectives.
+Reference: page:22
 
 Now answer the user's question in the same format:
 """
@@ -58,7 +89,11 @@ Now answer the user's question in the same format:
     )
     return chain
 
-def answer(question: str) -> str:
+def answer(user_question: str) -> str:
+    """
+    Normalize the user's question and ask the RAG chain.
+    """
+    question = normalize_question(user_question)
     chain = build_chain()
     result = chain.invoke({"question": question})
     return result.get("answer") or result.get("result") or "No answer."