Change interface again

app.py

@@ -1,23 +1,24 @@
 import gradio as gr
 import openthaigpt
-demo = gr.Blocks()
 
 def gen(input):
     return openthaigpt.generate(input)
 
 def zero(input):
-    return openthaigpt.zero(input)
+    return str(openthaigpt.zero(input))
 
-
-gen_interface = gr.Interface(fn=gen, inputs=gr.Textbox(lines=3, label="Input Prompt", value="Q: อยากลดความอ้วน ทำอย่างไร\n\nA:"), outputs="text")
-zero_interface = gr.Interface(fn=zero, inputs=gr.Textbox(lines=3, label="Input Text", value="การลดน้ำหนักเป็นเรื่องที่ต้องพิจารณาอย่างละเอียดและรอบคอบเพื่อให้ได้ผลลัพธ์ที่ดีและมีประสิทธิภาพมากที่สุด"), outputs="text")
-
-with demo:
+with gr.Blocks() as demo:
     gr.Markdown("OpenThaiGPT version 0.0.10")
     with gr.Tabs():
         with gr.TabItem("Generate"):
-            gen_interface
+            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_interface
+            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()   
+demo.launch()