Update app.py

app.py

@@ -1478,14 +1478,20 @@ 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, description, key_message)
+        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)
         checks[check_name] = result
         if result is False:
-            return checks  # Stop on first failure
+            return checks
+
 
     # Perform remaining LLM checks
     for check_name, check_func in llm_checks_group2:
-        result = safe_check(check_func, message, description, key_message)
+        result = safe_check(check_func, message)
         checks[check_name] = result
         if result is False:
             return checks  # Stop on first failure