import gradio as gr
from functions import random_plot, add_message, chat_mem, print_like_dislike

fig = random_plot()

# Membuat antarmuka Gradio
with gr.Blocks(fill_height=True) as demo:
    chatbot = gr.Chatbot(
        elem_id="chatbot",
        bubble_full_width=False,
        scale=1,
    )

    chat_input = gr.Textbox(interactive=True,
                            placeholder="Enter message...", show_label=False)

    chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
    bot_msg = chat_msg.then(chat_mem, inputs=[chat_input, chatbot], outputs=[chat_input, chatbot], api_name="bot_response")
    bot_msg.then(lambda: gr.Textbox(interactive=True), None, [chat_input])

    chatbot.like(print_like_dislike, None, None)

demo.launch()