from tkinter import *
import backend


def train():
    sentence = train_entry.get()
    ac = backend.AutoComplete()
    ac.train(sentence)


def predict_word():
    word = predict_word_entry.get()
    ac = backend.AutoComplete()
    print(ac.predict(word))


if __name__ == "__main__":
    root = Tk()
    root.title("Input note")
    root.geometry("300x300")

    train_label = Label(root, text="Train")
    train_label.pack()
    train_entry = Entry(root)
    train_entry.pack()

    train_button = Button(root, text="train", command=train)
    train_button.pack()

    predict_word_label = Label(root, text="Input term to predict")
    predict_word_label.pack()
    predict_word_entry = Entry(root)
    predict_word_entry.pack()

    predict_button = Button(root, text="predict", command=predict_word)
    predict_button.pack()

    root.mainloop()