import gradio as gr
import openai
from osbot_utils.utils.Dev import pprint
from osbot_utils.utils.Misc import list_set

from cbr_athena.api.Chat_Predict import Chat_Predict
from cbr_athena.api.Open_API import Open_API




TITLE = "# Meet Athena. v0.3.0"

class Gradio_Test:


    def __init__(self):
        #self.demo = None
        pass

    def title(self):
        return TITLE

    def add_great(self):
        pass
        # chatbot = gr.Chatbot()
        # msg = gr.Textbox()
        # clear = gr.Button("Clear")
        #
        # def user(user_message, history):
        #     print('user_message', user_message)
        #     print('history', history)
        #     print('-------')
        #     return "", history + [[user_message, None]]
        #
        # def bot(history):
        #     from random import choice
        #     from time import sleep
        #     bot_message = choice(["How are you?", "I like you", "I'm very hungry"])
        #
        #     #sleep(2)
        #     print('bot_message', bot_message)
        #     print('history', history)
        #     history[-1][1] = bot_message + '____'
        #     #history.append(['from me','to you'])
        #     print('history', history)
        #     return history

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

    def add_chat_bot(self):
        default_text = "Hi, good morning"
        gr.Markdown(self.title())
        chat_predict = Chat_Predict()
        textbox_input = gr.Textbox(value=default_text, render=False)
        gr.ChatInterface(chat_predict.predict, textbox=textbox_input)

    def create_demo(self):
        with gr.Blocks() as demo:
            self.add_great()
            self.add_chat_bot()






        demo.queue()
        #self.demo = demo
        return demo

    # def launch(self):
    #     self.demo.queue()
    #     #self.demo.launch()