import gradio as gr
from pipe import pipe
import pytz
import datetime
import json

def fn(text):
    now = datetime.datetime.now(pytz.timezone('Asia/Tokyo'))
    print('\ntext:',text,now)
    result = pipe.run({
        "prompt_builder":{"text":text}
    })
    parsed = result['parser']['parsed']
    print(parsed)
    return json.dumps(parsed,indent=2,ensure_ascii=False)

title = '日本語 明確さ判定'

with gr.Blocks(title=title,css_paths='./app.css') as app:
    gr.HTML(f'<h1 class="title">{title}</h1>')
    inputs=gr.TextArea(label='テキスト',lines=12)
    submit = gr.Button("送信",variant="primary")
    reply =gr.JSON(label='回答')
    submit.click(lambda: gr.update(interactive=False),inputs=None, outputs=submit) \
        .then(fn=fn, inputs=inputs, outputs=reply ) \
        .then(fn=lambda: gr.update(interactive=True),inputs=None, outputs=submit)


if __name__ == "__main__":
    app.launch()