Update app.py

app.py

@@ -25,8 +25,8 @@ def get_model():
         ).to(device).eval()
     return tokenizer, model
 
-# Threshold used ONLY for coloring (Red vs Green)
-THRESHOLD = 0.81 
+# UPDATED: Threshold changed from 0.81 to 0.59
+THRESHOLD = 0.59 
 
 # -----------------------------
 # UTILITIES
@@ -72,6 +72,7 @@ def analyze(text):
         return "—", "—", "<em>Please enter text...</em>", None
         
     word_count = len(text.split())
+    # Word count remains at 300
     if word_count < 300:
         warning_msg = f"⚠️ <b>Insufficient Text:</b> Your input has {word_count} words. Please enter at least 300 words for an accurate analysis."
         return "Too Short", "N/A", f"<div style='color: #b80d0d; padding: 20px; border: 1px solid #b80d0d; border-radius: 8px;'>{warning_msg}</div>", None
@@ -103,7 +104,7 @@ def analyze(text):
     weighted_avg = sum(p * l for p, l in zip(probs, lengths)) / total_words if total_words > 0 else 0
 
     # -----------------------------
-    # HTML RECONSTRUCTION (Unfiltered Probabilities)
+    # HTML RECONSTRUCTION
     # -----------------------------
     highlighted_html = "<div style='font-family: sans-serif; line-height: 1.8;'>"
     prob_map = {idx: probs[i] for i, idx in enumerate(pure_sents_indices)}
@@ -115,7 +116,7 @@ def analyze(text):
         
         if i in prob_map:
             score = prob_map[i]
-            # Color is still determined by the 81% threshold for visual aid
+            # Logic now uses the 0.59 threshold
             if score >= THRESHOLD:
                 color, bg = "#b80d0d", "rgba(184, 13, 13, 0.15)" # RED
             else:
@@ -130,7 +131,6 @@ def analyze(text):
             highlighted_html += block
     highlighted_html += "</div>"
 
-    # --- RAW RESULTS (No Masking) ---
     label = f"{weighted_avg:.1%} AI Probability"
     display_score = f"{weighted_avg:.2%}"
     
@@ -142,7 +142,7 @@ def analyze(text):
 # -----------------------------
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("## 🕵️ AI Detector Pro: Raw Mode")
-    gr.Markdown(f"Direct model output from **{MODEL_NAME}**. Visual highlight still triggers at **{THRESHOLD*100:.0f}%**.")
+    gr.Markdown(f"Direct model output from **{MODEL_NAME}**. Visual highlight now triggers at **{THRESHOLD*100:.0f}%**.")
     
     with gr.Row():
         with gr.Column(scale=3):