Update app.py

app.py

@@ -19,15 +19,25 @@ def detect_ai(text):
         with torch.no_grad():
             outputs = model(**inputs)
             probs = torch.softmax(outputs.logits, dim=1)
-        ai_score = float(probs[0][1])   # fix here
+        ai_score = float(probs[0][1])
         results.append({"sentence": sent, "ai_score": ai_score})
 
+    # Build highlighted HTML
     highlighted = ""
     for r in results:
         color = f"rgba(255,0,0,{r['ai_score']})"
         highlighted += f"<span style='background-color:{color}; padding:2px'>{r['sentence']} </span>"
 
-    return highlighted, results
+    # Compute total AI percentage
+    if results:
+        avg_ai = sum(r['ai_score'] for r in results) / len(results)
+        total_percent = round(avg_ai * 100, 2)
+        highlighted += f"<p><b>Total AI Probability: {total_percent}%</b></p>"
+    else:
+        total_percent = 0.0
+
+    return highlighted, {"sentences": results, "total_ai_percent": total_percent}
+
 
 
 with gr.Blocks() as demo: