Update app.py

app.py

@@ -67,6 +67,7 @@ def predict_safety_api(text):
     total_windows_count = len(windows)
     is_blocked = False
     highest_unsafe_prob = 0.0
+    highest_safe_prob = 0.0
     windows_analysis = []
     triggered_windows = []
 
@@ -99,8 +100,13 @@ def predict_safety_api(text):
             is_blocked = True
             highest_unsafe_prob = max(highest_unsafe_prob, unsafe_prob)
             triggered_windows.append(idx + 1)
+        
+        else:
+            highest_safe_prob = max(highest_safe_prob, safe_prob)    
 
     final_prediction = "UNSAFE" if is_blocked else "SAFE"
+    winning_probability = highest_unsafe_prob if is_blocked else highest_safe_prob
+    formatted_confidence= f"{winning_probability *100:.2f}%"
     
     unsafe_confidence_score = round(highest_unsafe_prob, 4)
     safe_confidence_score = round(1.0 - highest_unsafe_prob, 4)