import gradio as gr
import openthaigpt

def gen(input):
    return openthaigpt.generate(input)

def zero(input):
    return str(openthaigpt.zero(input))

with gr.Blocks() as demo:
    gr.Markdown("OpenThaiGPT version 0.0.10")
    with gr.Tabs():
        with gr.TabItem("Generate"):
            gen_input = gr.Textbox(lines=3, label="Input Prompt", value="Q: อยากลดความอ้วน ทำอย่างไร\n\nA:")
            gen_output = gr.Textbox(lines=3, label="Generated Output", value="")
            gen_btn = gr.Button("Generate")
            gen_btn.click(fn=gen, inputs=gen_input, outputs=gen_output)
        with gr.TabItem("Zero (GPT Check)"):
            zero_input = gr.Textbox(lines=3, label="Input Text", value="การลดน้ำหนักเป็นเรื่องที่ต้องพิจารณาอย่างละเอียดและรอบคอบเพื่อให้ได้ผลลัพธ์ที่ดีและมีประสิทธิภาพมากที่สุด")
            zero_output = gr.Textbox(lines=3, label="Check Result", value="")
            zero_btn = gr.Button("Check")
            zero_btn.click(fn=zero, inputs=zero_input, outputs=zero_output)

demo.launch()