Update app.py

app.py

@@ -121,6 +121,7 @@ def evaluate_answer(answer, question, kb):
         }
     }
 
+    # ---------------- SCHEMA ----------------
     key = hash_key(kb, question)
     if key not in SCHEMA_CACHE:
         SCHEMA_CACHE[key] = generate_schema_with_llm(kb, question)
@@ -128,6 +129,7 @@ def evaluate_answer(answer, question, kb):
     schema = SCHEMA_CACHE[key]
     logs["schema"] = schema
 
+    # ---------------- ANSWER CLAIMS ----------------
     claims = decompose_answer(answer)
     logs["claims"] = claims
 
@@ -149,35 +151,21 @@ def evaluate_answer(answer, question, kb):
 
     logs["coverage"] = coverage
 
-    # ---------------- CONTRADICTION CHECK ----------------
-    # contradictions = []
-    # kb_sents = split_sentences(kb)
+    # ---------------- CONTRADICTION CHECK (FIXED) ----------------
+    contradictions = []
+    relevant_kb = schema.get("required_concepts", [])
 
-    # for claim in claims:
-    #     for sent in kb_sents:
-    #         probs = softmax_logits(nli_model.predict([(sent, claim)]))
-    #         if probs[0] > CONTRADICTION_THRESHOLD:
-    #             contradictions.append({
-    #                 "claim": claim,
-    #                 "sentence": sent,
-    #                 "confidence": round(probs[0] * 100, 1)
-    #             })
-
-    # logs["contradictions"] = contradictions
-
-
-relevant_kb = schema.get("required_concepts", [])
-
-for claim in claims:
-    for sent in relevant_kb:
-        probs = softmax_logits(nli_model.predict([(sent, claim)]))
-        if probs[0] > CONTRADICTION_THRESHOLD:
-            contradictions.append({
-                "claim": claim,
-                "sentence": sent,
-                "confidence": round(probs[0] * 100, 1)
-            })
+    for claim in claims:
+        for sent in relevant_kb:
+            probs = softmax_logits(nli_model.predict([(sent, claim)]))
+            if probs[0] > CONTRADICTION_THRESHOLD:
+                contradictions.append({
+                    "claim": claim,
+                    "sentence": sent,
+                    "confidence": round(probs[0] * 100, 1)
+                })
 
+    logs["contradictions"] = contradictions
 
     # ---------------- FINAL DECISION ----------------
     if contradictions:
@@ -190,6 +178,7 @@ for claim in claims:
     logs["final_verdict"] = verdict
     return verdict, logs
 
+
 # ============================================================
 # GRADIO UI
 # ============================================================