Update app.py

app.py

@@ -91,12 +91,17 @@ def get_conversation_chain(vectorstore):
 
     # Define a strict prompt template that makes the model answer only based on the document
     prompt_template = """
-    You are a helpful assistant. Answer the question using only the provided document content.
-    If the answer is not in the document, simply respond with "I cannot provide an answer based on the document.".
-    Question: {question}
-    """
+                    You are a helpful assistant. Use the following document context to answer the question.
+                    If the answer is not in the document, simply respond with "I cannot provide an answer based on the document."
+
+                    Document: {context}
+
+                    Question: {question}
+                    """
 
-    prompt = PromptTemplate(input_variables=["question"], template=prompt_template)
+    prompt = PromptTemplate(
+        input_variables=["context", "question"], template=prompt_template
+    )
 
     try:
         llm = ChatOpenAI(model_name="gpt-4o")
@@ -109,6 +114,7 @@ def get_conversation_chain(vectorstore):
             memory=memory,
             return_source_documents=True,
             combine_docs_chain_kwargs={"prompt": prompt},
+            document_variable_name="context",  # Specify the variable name for the document context
         )
         return conversation_chain
     except Exception as e:
@@ -241,6 +247,11 @@ def chat(slug, user_id):
     Restricts chat based on user group and chat count.
     """
 
+    # Show the user instruction at the top of the chat interface
+    st.write(
+        "**Please note:** Due to processing limitations, the chat may not fully comprehend the whole document."
+    )
+
     text_chunks = get_text_chunks(get_pdf_text(slug))
     vectorstore = get_vectorstore(text_chunks)
     st.session_state.conversation = get_conversation_chain(vectorstore)