from fastai.text.all import *
import gradio as gr

# Модельді жүктеу
learn = load_learner('model.pkl')

# Класстар мен сипаттамалар
description = "Медициналық Диагностика"
categories = ['Аллергия', 'Анемия', 'Бронхит', 'Диабет', 'Диарея', 'Шаршау', 'Тұмау', 'Малярия', 'Стресс']

# Мәтінді классификациялау функциясы
def classify_text(txt):
    pred, idx, probs = learn.predict(txt)
    return dict(zip(categories, map(float, probs)))

# Интерфейс үшін енгізу және шығару компоненттері
text = gr.Textbox(lines=2, label='Өзіңізді қалай сезінетініңізді егжей-тегжейлі сипаттаңыз')
label = gr.Label()
examples = [
    'Менде физикалық белсенділікке қызығушылық жоқ. Мен әрқашан шөлдеп жүрмін',
    'Мен тоңып тұрмын',
    'Көзіңіз бозарған'
]

# Интерфейс жасау
intf = gr.Interface(fn=classify_text, inputs=text, outputs=label, examples=examples, description=description)
intf.launch(inline=False)