Update app.py

app.py

@@ -5,24 +5,31 @@ hate = pipeline("text-classification", model="hossam87/bert-base-arabic-hate-spe
 dialect = pipeline("text-classification", model="IbrahimAmin/marbertv2-arabic-written-dialect-classifier", tokenizer="IbrahimAmin/marbertv2-arabic-written-dialect-classifier", return_all_scores=False)
 
 def analyze(text):
+    # Hate speech detection
     hate_res = hate(text)[0]
     hate_label = hate_res['label']
     hate_conf = hate_res['score']
 
+    # Dialect detection
     dial_res = dialect(text)[0]
     dial_label = dial_res['label']
     dial_conf = dial_res['score']
 
+    # Threat score
     weight = {"Neutral":0, "Offensive":0.5, "Sexism":1, "Racism":1, "Religious Discrimination":1}
     score = hate_conf * weight.get(hate_label, 0)
-    action = "✅ Safe Content — No harmful language detected. No moderation needed."
-    if score > 0.8:
-        action =  "🚨 Immediate Review Required — This content contains severe hate speech or threats and should be escalated to moderators immediately."
+
+    # Recommended action (modified logic)
+    if hate_label != "Neutral":
+        action = "🚨 Immediate Review Required — This content contains severe hate speech or threats and should be escalated to moderators immediately."
     elif score >= 0.49:
         action = "⚠️ Potentially Harmful — The content may contain offensive or harmful language. Please review before taking further action."
+    else:
+        action = "✅ Safe Content — No harmful language detected. No moderation needed."
 
     return hate_label, f"{hate_conf:.2f}", dial_label, f"{dial_conf:.2f}", f"{score:.2f}", action
 
+
 iface = gr.Interface(
     fn=analyze,
     inputs=gr.Textbox(lines=4, placeholder="اكتب هنا...", label="Arabic Text"),