app.py

import gradio as gr
from transformers import pipeline

# تحميل الموديل
classifier = pipeline(
    "text-classification",
    model="MennatullahHany/Arabic-Speech-Guard",
    tokenizer="MennatullahHany/Arabic-Speech-Guard",
    return_all_scores=True
)

# mapping labels
label_map = {
    "LABEL_0": "SAFE ✅",
    "LABEL_1": "DANGER ⚠️"
}

def predict(text):
    if text.strip() == "":
        return "❌ Please enter Arabic text"

    results = classifier(text)[0]  # قائمة من القيم لكل label
    output = ""
    for r in results:
        mapped_label = label_map.get(r['label'], r['label'])
        output += f"{mapped_label}: {round(r['score']*100, 2)}%\n"
    return output

# واجهة Gradio
demo = gr.Interface(
    fn=predict,
    inputs=gr.Textbox(lines=4, placeholder="اكتب النص هنا..."),
    outputs="text",
    title="Arabic Speech Guard 🛡️",
    description="Detect offensive Arabic content"
)

demo.launch()