Update app.py

app.py

@@ -1478,19 +1478,14 @@ def perform_checks(message, description, key_message):
 
     # Perform LLM checks group 1
     for check_name, check_func in llm_checks_group1:
-        result = safe_check(check_func, message)
+        result = safe_check(check_func, message, description, key_message)
         checks[check_name] = result
         if result is False:
             return checks  # Stop on first failure
 
     # Perform remaining LLM checks
     for check_name, check_func in llm_checks_group2:
-        if check_name == "no_contradictions":
-            result = safe_check(check_func, message, description)
-        elif check_name == "contains_key_message":
-            result = safe_check(check_func, message, key_message)
-        else:
-            result = safe_check(check_func, message)
+        result = safe_check(check_func, message, description, key_message)
         checks[check_name] = result
         if result is False:
             return checks  # Stop on first failure