import gradio as gr
from ruaccent import RUAccent

accentizer = RUAccent()
accentizer.load(omograph_model_size='turbo3.1', use_dictionary=True, tiny_mode=False)

def accentize_text(text):
    return accentizer.process_all(text)

examples = [
    ["на двери висит замок."],
    ["мы можем все!"],
    ["я еду на машине в магазин, чтобы купить еду."],
    ["с нами живут сестры."],
    ["мы все живем в этом мире."],
    ["руки прочь!"],
    ["я хочу добиться твоей руки."]
]

iface = gr.Interface(
    fn=accentize_text,
    inputs=gr.Textbox(lines=5, placeholder="Введите текст для расстановки ударений..."),
    outputs="text",
    title="Расстановка ударений в русском тексте",
    description="Демонстрация работы библиотеки RUAccent для автоматической расстановки ударений в русском тексте, включая разрешение омографов.",
    examples=examples
)

iface.launch()