import gradio as gr
from transformers import pipeline

# Inizializza il modello DistilBERT per la classificazione di testo
classifier = pipeline("text-classification", model="distilbert-base-uncased")

def classify_email(text):
    result = classifier(text)
    # `result` è una lista con un solo dizionario, prendiamo il primo elemento e restituiamo solo la label
    label = result[0]['label']
    score = result[0]['score']
    return {label: score}

# Interfaccia con Gradio
iface = gr.Interface(fn=classify_email,
                     inputs="text",
                     outputs="label",
                     title="ZeroSpam Email Classifier",
                     description="Inserisci l'email da analizzare per determinare se è spam o phishing.")

iface.launch(share=True)