Spaces:
Running
Running
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) | |