Update main.py

main.py

@@ -59,7 +59,7 @@ except Exception as e:
 # ==========================
 try:
     db = FAISS.load_local("ipc_vector_db", embeddings, allow_dangerous_deserialization=True)
-    db_retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2, "max-length": 512})
+    db_retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 4, "max-length": 512})
     logger.info("Vectorstore successfully loaded.")
 except Exception as e:
     logger.error(f"Error loading FAISS vectorstore: {e}")
@@ -68,13 +68,15 @@ except Exception as e:
 # ==========================
 # Define Prompt Template
 # ==========================
-prompt_template = """<s>[INST]As a legal chatbot specializing in the Indian Penal Code, provide a concise and accurate answer based on the given context. Avoid unnecessary details or unrelated content. Only respond if the answer can be derived from the provided context; otherwise, say \"The information is not available in the provided context.\" 
+prompt_template = """<s>[INST]You are a legal chatbot specializing in the Indian Penal Code. Provide concise, context-aware answers in a conversational tone. Avoid presenting the response as a question-answer format unless explicitly required. 
+If the answer cannot be derived from the given context, respond with: "I'm sorry, I couldn't find relevant information for your query."
 CONTEXT: {context}
 CHAT HISTORY: {chat_history}
 QUESTION: {question}
 ANSWER:
 </s>[INST]"""
 
+
 prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question", "chat_history"])
 
 # ==========================
@@ -124,12 +126,15 @@ async def chat(request: ChatRequest):
     try:
         logger.debug(f"Received user question: {request.question}")
         result = qa.invoke(input=request.question)
-        answer = result.get("answer", "I'm sorry, but I couldn't generate a response to your query. Please try rephrasing or providing more details.")
+        answer = result.get("answer")
+        if not answer or "The information is not available in the provided context" in answer:
+            answer = "I'm sorry, I couldn't find relevant information for your query. Please try rephrasing or providing more details."
         return ChatResponse(answer=answer)
     except Exception as e:
         logger.error(f"Error during chat invocation: {e}")
         raise HTTPException(status_code=500, detail="Oops! Something went wrong on our end. Please try again later.")
 
+
 # ==========================
 # Run Uvicorn Server
 # ==========================