import gradio as gr
#import main
from dateutil.utils import today


def pomoch(age, gender, height, weight, ap_hi, ap_lo, cholesterol, gluc, smoke, alco, active):
    X = [int(age), gender, int(height), int(weight), int(ap_hi), int(ap_lo), float(cholesterol), float(gluc), smoke,
         alco, active]
    X[0] = today().year - X[0]
    if X[1] == "Мужской":
        X[1] = 0
    else:
        X[1] = 1
    for i in range(8, 11):
        if X[i] == "Да":
            X[i] = 1
        else:
            X[i] = 0
    if X[6] <= 5:
        X[6] = 1
    else:
        if X[6] >= 7.8:
            X[6] = 3
        else:
            X[6] = 2

    if X[7] <= 5.5 and X[7] >= 3.3:
        X[7] = 1
    else:
        if (X[7] > 5.5 and X[7] < 11) or X[7] < 3.3:
            X[7] = 2
        else:
            X[7] = 3
    return X
#    return main.webai(X)


demo = gr.Interface(
    pomoch,
    [
        gr.Slider(1900, 2010, value=1990, step=1, label="Год рождения"),
        gr.Radio(["Мужской", "Женский"], label="Пол", ),
        gr.Textbox(label="Рост(см)"),
        gr.Textbox(label="Вес(кг)"),
        gr.Textbox(label="Верхнее Давление"),
        gr.Textbox(label="Нижнее Давление"),
        gr.Textbox(label="Холестерин(ммоль/л)"),
        gr.Textbox(label="Глюкоза(ммоль/л)", ),
        gr.Radio(["Да", "Нет"], label="Курение", ),
        gr.Radio(["Да", "Нет"], label="Алкоголь", ),
        gr.Radio(["Да", "Нет"], label="Активность", ),
    ],
    'text')

demo.launch()