from src.unmask import umaskall
import ipywidgets as widgets

text = widgets.Textarea(value="it _is a [RB] useful _", layout=widgets.Layout(width='100%', height="10em"))
top_k = widgets.IntText(layout=widgets.Layout(width='10%',))
top_k2 = widgets.IntSlider(min=1, max=100, value=10, step=1, description="Top-K", readout=False, 
                           layout=widgets.Layout(width='90%',), style = {'description_width': 'auto'})
widgets.jslink((top_k2, 'value'), (top_k, 'value'))
split_stences = widgets.Checkbox(value=True, description="stence by stence", indent=False)
single_mask = widgets.Checkbox(value=False, description="one blank at a time", indent=False)

import codecs
file = widgets.FileUpload(multiple=False, description="Load File", accept=".txt")
def on_upload_change(change):
    c = get_content(change['owner'])
    if c:
        text.value = c
        on_button_clicked(button)
file.observe(on_upload_change, names='_counter')

def get_content(file):
    file = file.value
    if isinstance(file, dict):
        c = next(iter(file.values()))['content']
    else:
        c = file[0]['content']
    return codecs.decode(c, encoding="utf-8")

import sys
class MyWriter:
    def __init__(self, file):
        self.logfile = file
    def write(self, text):
        sys.stdout.write(text)
        self.logfile.write(text)
    def close(self):
        sys.stdout.close()
        self.logfile.close()

button = widgets.Button(description="Go", button_style="success", 
                        icon="arrow-alt-circle-right", layout=widgets.Layout(width='30%'))
out = widgets.Output()

@out.capture()
def on_button_clicked(b):
    out.clear_output()
    with open("output-holly_cloze.txt", 'w', encoding="utf-8") as f:
        umaskall(text.value, top_k=top_k.value, 
                 split_stences=split_stences.value, single_mask=single_mask.value,
                 io=MyWriter(f))
    
    
button.on_click(on_button_clicked)

tpk = widgets.HBox(children=[top_k2, top_k])
chb = widgets.HBox(children=[split_stences, single_mask])
btn = widgets.HBox(children=[button, file])
display(tpk, chb, text, btn, out)