import gradio as gr
from ukrainian_accentor_transformer import Accentor



accentor = Accentor()



def accentification(sentence):
    accented_sentence = accentor(sentence)
    return accented_sentence



iface = gr.Interface(
    fn=accentification,
    inputs=gr.inputs.Textbox(label="Input"),
    outputs=gr.outputs.Textbox(label="Stressed"),
    examples=[
        "Помилки вчать людей. Молодець, ти не зробив жодної помилки.",
        "Роботи підкорять світ, але від роботи не втечеш",
        "Золоті яйця, але нема ні яйця",
        "Треба ще пошукати такий міцний замок. А це Замок Вінгерський.",
    ],
    title="Ukrainian Accentor Transformer",
)
iface.launch()