Update app.py

app.py

@@ -1,3 +1,34 @@
+import gradio as gr
+from transformers import pipeline
+
+hate = pipeline("text-classification", model="hossam87/bert-base-arabic-hate-speech", tokenizer="hossam87/bert-base-arabic-hate-speech", return_all_scores=False)
+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)
+
+    # 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
+
 # ---------- CSS ----------
 CSS = """
 /* Set same background color for everything */