import gradio as gr
from transformers import pipeline

classifier = pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli")

def classify_email(text):
    labels = ["встреча", "отказ", "уточнение"]
    result = classifier(text, candidate_labels=labels)
    label = result["labels"][0]
    if label == "встреча":
        return "1"
    elif label == "отказ":
        return "2"
    elif label == "уточнение":
        return "3"
    else:
        return "?"

iface = gr.Interface(fn=classify_email, inputs="text", outputs="text", title="Email Classifier")
iface.launch()