import time
import gradio as gr
from chatbot_multiagent import submitUserMessage


def chat(message:str, history):
    print(message)
    return submitUserMessage(message)


def slow_echo_chat(message, history):
    answer = submitUserMessage(message)
    for i in range(len(answer)):
        time.sleep(0.01)
        yield answer[: i+1]
        
        
with gr.Blocks() as demo:
    chatbot = gr.Chatbot(height=600)
    msg = gr.Textbox()
    clear = gr.ClearButton([msg, chatbot])

    def respond(message, chat_history):
        bot_message = submitUserMessage(message)
        chat_history.append((message, bot_message))
        return "", chat_history

    msg.submit(respond, [msg, chatbot], [msg, chatbot])
    

demo.launch()
        
# gr.ChatInterface(chat).launch()

# interface = gr.ChatInterface(chat)
# interface.launch()