Update app.py

app.py

@@ -1,81 +1,47 @@
-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 */
+body, .gradio-container, .gr-blocks, .gr-box, .gr-panel {
+    background-color: #2b2b2b !important;
+}
 
-# ---------- UI (Blocks with Sidebar) ----------
-SIDEBAR_CSS = """
-#sidebar-panel {
-  background: #2b2b2b;  /* keep sidebar's grey */
-  border: 1px solid rgba(255,255,255,0.08);
-  border-radius: 10px;
-  padding: 18px;
+/* Keep text visible */
+* {
+    color: #fff !important;
 }
 
-/* Make textboxes transparent and match the page background */
+/* Style input fields */
 textarea, input[type="text"] {
-  background: transparent !important;
-  border: 1px solid rgba(255,255,255,0.2) !important;
-  color: white !important;
+    background-color: #1f1f1f !important;
+    border: 1px solid rgba(255,255,255,0.2) !important;
+    color: #fff !important;
 }
 """
 
-with gr.Blocks(css=SIDEBAR_CSS, theme="default") as demo:
+# ---------- UI ----------
+with gr.Blocks(css=CSS, theme="default") as demo:
     with gr.Row(equal_height=True):
-        # Sidebar (left)
         with gr.Column(scale=1):
-            with gr.Group(elem_id="sidebar-panel"):
-                gr.Markdown(
-                    """
-                    ## 🛡️ Arabic Content Safety Analyzer
+            gr.Markdown("""
+            ## 🛡️ Arabic Content Safety Analyzer
 
-                    **Purpose**  
-                    Analyze Arabic text for harmful or threatening language.
+            **Purpose**  
+            Analyze Arabic text for harmful or threatening language.
 
-                    **Features**
-                    - **Hate Speech Classification** (Offensive, Racism, Sexism, Religious Discrimination)
-                    - **Dialect Detection** (Gulf, Levant, Egyptian, MSA)
-                    - **Threat Severity Score** (0–1 based on label + confidence)
-                    - **Recommended Action** (rule-based suggestion)
+            **Features**
+            - **Hate Speech Classification** (Offensive, Racism, Sexism, Religious Discrimination)
+            - **Dialect Detection** (Gulf, Levant, Egyptian, MSA)
+            - **Threat Severity Score** (0–1 based on label + confidence)
+            - **Recommended Action** (rule-based suggestion)
 
-                    **How to Use**
-                    1. Enter Arabic text in the box.
-                    2. Click **Analyze**.
-                    3. Review the results and suggested action.
-                    """
-                )
+            **How to Use**
+            1) أدخل النص العربي في الحقل.  
+            2) اضغط **Analyze**.  
+            3) راجع النتائج والإجراء المقترح.
+            """)
 
-        # Main app (right)
         with gr.Column(scale=3):
             gr.Markdown("### Enter Arabic Text for Analysis")
-
             input_text = gr.Textbox(lines=4, placeholder="اكتب هنا...", label="Arabic Text")
 
             out_hate = gr.Textbox(label="Hate Speech Label", interactive=False)
@@ -87,11 +53,10 @@ with gr.Blocks(css=SIDEBAR_CSS, theme="default") as demo:
 
             analyze_btn = gr.Button("Analyze", variant="primary")
             analyze_btn.click(
-                fn=analyze,
+                analyze,
                 inputs=input_text,
-                outputs=[out_hate, out_hate_conf, out_dialect, out_dialect_conf, out_score, out_action],
+                outputs=[out_hate, out_hate_conf, out_dialect, out_dialect_conf, out_score, out_action]
             )
 
-# ---------- Launch ----------
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()