Update app.py

app.py

@@ -168,7 +168,7 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
     retriever=vector_db.as_retriever()
     progress(0.8, desc="Defining retrieval chain...")
 
-    prompt_template_string = f'''Your task is as follows: 1. Determine if the input {question} is compliant with the provided {context}. 2. If the requirement is compliant, report "This requirement is compliant." 3. If the requirement is not compliant report "This requirement is not compliant." 4. If the requirement is not compliant, give the reason for non compliance and return the specific rule or guideline the requirement violates. 5. If the requirement is not compliant, report a refined version of the requirement delimited in quotes that is compliant with the provided {context}.'''
+    prompt_template_string = f'''Your task is as follows: 1. Determine if the input {user_message} is compliant with the provided {context}. 2. If the requirement is compliant, report "This requirement is compliant." 3. If the requirement is not compliant report "This requirement is not compliant." 4. If the requirement is not compliant, give the reason for non compliance and return the specific rule or guideline the requirement violates. 5. If the requirement is not compliant, report a refined version of the requirement delimited in quotes that is compliant with the provided {context}.'''
 
     # Create a PromptTemplate object
     prompt_template = PromptTemplate(