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()