import gradio as gr
import os

HF_TOKEN = os.getenv('HF_TOKEN')
hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo")

def greet(test, check):
    if check: test += 'heh'
    return "Hello " + test + "!!", "burp!"

iface = gr.Interface(fn=greet, inputs=["text", "checkbox"], outputs=[gr.Textbox(label='greeting'), "text"], allow_flagging='auto', flagging_callback=hf_writer)
iface.launch()