import gradio as gr
import random
import time
import os

openai_secret_key = os.environ.get('OPENAI_SECRET_KEY')
with open("Format_Library/Weldon_Handover_Note_Format.txt", "r") as f:
    role = f.read()
role_head = role[0:50]

with gr.Blocks() as demo:
    chatbot = gr.Chatbot()
    msg = gr.Textbox()
    clear = gr.Button("Clear")

    def user(user_message, history):
        return "", history + [[user_message, None]]

    def bot(history):
        bot_message = random.choice(["Yes", role_head])
        history[-1][1] = bot_message
        time.sleep(1)
        return history

    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
        bot, chatbot, chatbot
    )
    clear.click(lambda: None, None, chatbot, queue=False)

demo.launch()