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


def run(text):
    now = datetime.datetime.now(pytz.timezone("Asia/Tokyo"))
    print("\ntext:", text, now)
    result = pipe.run(
        {"retrieve_prompt_builder": {"text": text}, "prompt_builder": {"text": text}},
        include_outputs_from=["embedding_retriever", "parser"],
    )
    for document in result["embedding_retriever"]["documents"]:
        print(document.score, document.id)
        print(document.content)
    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=run, inputs=inputs, outputs=reply).then(
        fn=lambda: gr.update(interactive=True), inputs=None, outputs=submit
    )


if __name__ == "__main__":
    app.launch(share=True, debug=True)