Update ai_text_detector_valid_final.py

ai_text_detector_valid_final.py

@@ -102,6 +102,20 @@ def run_hf_model(model_id, text):
 # ---------------------------
 # Main Detector
 # ---------------------------
+def verdict(ai_prob):
+    """Return a human-readable verdict based on AI probability"""
+    if ai_prob < 20:
+        return "Most likely human-written."
+    elif 20 <= ai_prob < 40:
+        return "Possibly human-written with minimal AI assistance."
+    elif 40 <= ai_prob < 60:
+        return "Unclear – could be either human or AI-assisted."
+    elif 60 <= ai_prob < 80:
+        return "Possibly AI-generated, or a human using AI assistance."
+    else:  # ai_prob >= 80
+        return "Likely AI-generated or heavily AI-assisted."
+
+
 def detect_text(text):
     results = {}
     # HuggingFace transformer models
@@ -112,7 +126,7 @@ def detect_text(text):
     results["SzegedAI Detector"] = classify_szegedai(text)
 
     # ---------------------------
-    # Improved Final Verdict
+    # Final Verdict (Hybrid Rule)
     # ---------------------------
     ai_probs = []
     strong_ai_detector = None
@@ -126,17 +140,16 @@ def detect_text(text):
     avg_ai = np.mean(ai_probs) if ai_probs else 0
 
     if strong_ai_detector:
-        verdict = f"Likely AI-generated"
+        final_verdict = verdict(strong_ai_detector["AI Probability"])
         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"
+            final_verdict += f" (Identified: {strong_ai_detector['Identified LLM']})"
     else:
-        verdict = "Likely human-written"
+        final_verdict = verdict(avg_ai)
 
-    results["Final Score"] = {"Verdict": verdict, "Average AI Probability": round(avg_ai, 2)}
+    results["Final Score"] = {
+        "Verdict": final_verdict,
+        "Average AI Probability": round(avg_ai, 2)
+    }
     return results
 
 if __name__ == "__main__":