Update ai_text_detector_valid_final.py

ai_text_detector_valid_final.py

@@ -111,19 +111,32 @@ def detect_text(text):
     # SzegedAI ModernBERT ensemble
     results["SzegedAI Detector"] = classify_szegedai(text)
 
-    # Final verdict
+    # ---------------------------
+    # Improved Final Verdict
+    # ---------------------------
     ai_probs = []
-    for v in results.values():
+    strong_ai_detector = None
+
+    for k, v in results.items():
         if "AI Probability" in v:
             ai_probs.append(v["AI Probability"])
+            if v["AI Probability"] > 90:   # strong AI flag
+                strong_ai_detector = v
+
     avg_ai = np.mean(ai_probs) if ai_probs else 0
-    if avg_ai > 80:
-        verdict = "Likely AI-generated"
-    elif avg_ai > 40:
-        verdict = "Possibly human-written with AI assistance"
+
+    if strong_ai_detector:
+        verdict = f"Likely AI-generated"
+        if "Identified LLM" in strong_ai_detector:
+            verdict += f" (Identified: {strong_ai_detector['Identified LLM']})"
+    elif avg_ai > 60:
+        verdict = "Possibly AI-generated (with assistance)"
+    elif 40 <= avg_ai <= 60:
+        verdict = "Uncertain / Mixed signals"
     else:
         verdict = "Likely human-written"
-    results["Final Score"] = {"Verdict": verdict}
+
+    results["Final Score"] = {"Verdict": verdict, "Average AI Probability": round(avg_ai, 2)}
     return results
 
 if __name__ == "__main__":