cbr-athena / cbr_athena /Gradio_Test.py
DinisCruz's picture
changed version number on Gradio app
9ee53d4
raw
history blame
2 kB
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()