Update app.py

app.py

@@ -42,7 +42,6 @@ def predict_safety_api(text):
     full_encodings = tokenizer(cleaned_text, add_special_tokens=False, return_attention_mask=False)
     input_ids = full_encodings['input_ids']
     
-    # 🎯 1. حساب عدد التوكنز الكلي الفعلي للنص المدخل
     total_tokens_count = len(input_ids)
     
     window_size = 60
@@ -58,12 +57,10 @@ def predict_safety_api(text):
             if len(window) > 0: windows.append(window)
             if i + window_size >= len(input_ids): break
 
-    # 🎯 2. حساب عدد النوافذ الناتجة رياضياً عن هذا النص
     total_windows_count = len(windows)
 
     is_blocked = False
     highest_unsafe_prob = 0.0
-    # 🎯 3. مصفوفة مخصصة لاصطياد وحفظ النوافذ النصية التي تسببت في إطلاق الخطر
     triggered_sentences = []
 
     for win_ids in windows:
@@ -87,19 +84,17 @@ def predict_safety_api(text):
         if unsafe_p > 0.50:
             is_blocked = True
             highest_unsafe_prob = max(highest_unsafe_prob, unsafe_p)
-            # اصطياد النافذة الخبيثة المسببة للحظر وحفظها بدون تكرار
             if window_text not in triggered_sentences:
                 triggered_sentences.append(window_text)
 
-    # 🎯 حقن وحفظ الحقول الحسابية الجديدة والتفسيرية مباشرة داخل كائن الرد للـ JSON
     if is_blocked:
         return {
             "verdict": "UNSAFE", 
             "block": True, 
             "confidence": f"{highest_unsafe_prob * 100:.2f}%",
-            "total_tokens": total_tokens_count,        # عرض التوكنز الكلي
-            "total_windows": total_windows_count,      # عرض النوافذ الكلية
-            "triggered_phrases": triggered_sentences   # عرض الجمل المسببة للحظر
+            "total_tokens": total_tokens_count,       
+            "total_windows": total_windows_count,      
+            "triggered_phrases": triggered_sentences   
         }
     
     safe_p = 1.0 - highest_unsafe_prob
@@ -107,9 +102,9 @@ def predict_safety_api(text):
         "verdict": "SAFE", 
         "block": False, 
         "confidence": f"{safe_p * 100:.2f}%",
-        "total_tokens": total_tokens_count,        # عرض التوكنز الكلي
-        "total_windows": total_windows_count,      # عرض النوافذ الكلية
-        "triggered_phrases": []                    # فارغة لأن النص سليم وممرر
+        "total_tokens": total_tokens_count,        
+        "total_windows": total_windows_count,      
+        "triggered_phrases": []                 
     }
 
 interface = gr.Interface(