Upload app.py

app.py

@@ -234,17 +234,18 @@ print(faiss_index)
 # def promt_engineer(text):
 PROMPT_TEMPLATE = """
 Instructions:
------------------------------------------------------------------------------------------
-Answer the question  only based on the below context:
+-------------------------------------------------------------------------------------------------------
+Answer the question only based on the below context:
 - You're a Vedic AI expert in the Hindu Vedic scriptures.
+- Questions with out-of-context replay with The question is out of context. 
 - Always try to provide Keep it simple answers in nice format without incomplete sentence.
 - Give the answer atleast 5 seperate lines addition to the title info.
 - provide Title: <title> Chapter: <chapter> Text No: <textnumber> Page No: <pagenumber> 
-------------------------------------------------------------------------------------------
+--------------------------------------------------------------------------------------------------------
 
 {context}
 
-----------------------------------------------------------------------------------
+--------------------------------------------------------------------------------------------------------
 
 Answer the question based on the above context: {question}
 """
@@ -314,7 +315,11 @@ def chat_actions():
     # Example usage
     file_path = "Bhagavad-Gita-As-It-Is.pdf"
     text_to_highlight = context_text.strip()
-    display_highlighted_pdf(file_path, result, sources)
+
+    if "The question is out of context" not in result:
+        display_highlighted_pdf(file_path, result, sources)
+    else:
+        st.error("The question is out of context. Pages not found.")
 
 with st.sidebar:
     option = st.selectbox(