from handler import SweetCommander
import gradio as gr

controller = SweetCommander()

title = "BlueDice - Practice Space"

# demo = gr.Interface(
#     fn = controller,
#     inputs = [
#         gr.Textbox(label="Name", placeholder="Enter your name"),
#         gr.Textbox(label="Input", placeholder="Enter your message")
#     ],
#     outputs = gr.Textbox(label="Result")
# )
# demo.launch()

def predict(bot_name, bot_response, user_name, user_input, history = []):
    history.append(f"{bot_name}: {bot_response}")
    history.append(f"{user_name}: {user_input}")
    response_generator = controller(user_name, history)
    response_message = ""
    for response_token in response_generator:
        response_message += controller.tokenizer.decode(response_token, skip_special_tokens = True)
        yield response_message, history
    return response_message, history

gr.Interface(
    fn = predict,
    title = title,
    allow_flagging = "manual",
    inputs = ["BotName", "Response", "UserName", "Input", "State"],
    outputs = ["Response", "State"],
    theme = 'gradio/seafoam'
).launch()