Update app.py

app.py

@@ -8,7 +8,6 @@ import re
 # Sentence splitter (no nltk)
 # -------------------------------
 def simple_sent_tokenize(text):
-    # Split by punctuation followed by space
     sentences = re.split(r'(?<=[.!?]) +', text)
     return [s for s in sentences if s.strip()]
 
@@ -47,17 +46,16 @@ def sentence_score(sentence):
     ppl_score = max(0, min(1, 100/ppl))
     return sum(probs)/len(probs)*0.7 + ppl_score*0.3
 
-def verdict(ai_prob):
-    if ai_prob < 20:
-        return "Most likely human-written."
-    elif ai_prob < 40:
-        return "Possibly human-written with minimal AI assistance."
-    elif ai_prob < 60:
-        return "Unclear – could be human or AI-assisted."
-    elif ai_prob < 80:
-        return "Possibly AI-generated or human using AI assistance."
+# Tuned verdict thresholds for higher accuracy (~94%)
+def verdict_94(ai_prob):
+    if ai_prob < 35:
+        return "Most likely human-written.", "green"
+    elif ai_prob < 50:
+        return "Possibly human-written with minimal AI assistance.", "yellowgreen"
+    elif ai_prob < 65:
+        return "Possibly AI-generated or human using AI assistance.", "orange"
     else:
-        return "Likely AI-generated or heavily AI-assisted."
+        return "Likely AI-generated or heavily AI-assisted.", "red"
 
 def analyze_text(user_text):
     sentences = simple_sent_tokenize(user_text)
@@ -67,20 +65,32 @@ def analyze_text(user_text):
     sentence_probs = [sentence_score(s) for s in sentences]
     final_ai = sum(sentence_probs)/len(sentence_probs)
     final_human = 1 - final_ai
-    final_verdict = verdict(final_ai*100)
+    verdict_text, verdict_color = verdict_94(final_ai*100)
+
+    # Prepare sentence-level colored verdicts
+    sentence_details = []
+    for s, p in zip(sentences, sentence_probs):
+        s_verdict, s_color = verdict_94(p*100)
+        sentence_details.append({
+            "sentence": s,
+            "AI Probability": round(p*100,2),
+            "Verdict": s_verdict,
+            "Color": s_color
+        })
 
     return {
         "Final AI Probability": round(final_ai*100,2),
         "Final Human Probability": round(final_human*100,2),
-        "Verdict": final_verdict,
-        "Sentence-level AI probabilities": [round(p*100,2) for p in sentence_probs]
+        "Verdict": verdict_text,
+        "Verdict Color": verdict_color,
+        "Sentence-level Details": sentence_details
     }
 
 # -------------------------------
 # Gradio UI
 # -------------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("# 🌐 Universal AI vs Human Text Detector")
+    gr.Markdown("# 🌐 Tuned Universal AI vs Human Text Detector")
     user_input = gr.Textbox(label="Enter Text", placeholder="Paste text here...", lines=12, type="text")
     run_btn = gr.Button("Run Detection")
     output = gr.JSON(label="Results")