File size: 866 Bytes
6c89611 be09774 6c89611 e5f9801 6c89611 e5f9801 6c89611 e5f9801 6c89611 be09774 6c89611 be09774 6c89611 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | from openai import OpenAI
from commons.loggerfactory import LoggerFactory
from commons.utils import prompt_path, media_path
from scripts.components import OpenAIBot
from scripts.gradio_ui import GradioBotUI
from scripts.io.terminal import TerminalIO
class GradioBot:
exit_codes = ["done", "quit", "exit"]
client = OpenAI()
prompts = [prompt_path(file_name="rick_and_morty/System_prompt.txt")]
openai_bot = OpenAIBot(client, "gpt-4o-mini", prompts, 30, exit_codes=exit_codes)
gradio_bot = GradioBotUI(openai_bot, user_avatar=media_path("characters/morty.png"),
agent_avatar=media_path("characters/rick.png"))
def __init__(self):
self.logger = LoggerFactory.getLogger(self.__class__.__name__)
def start(self):
self.logger.info("Starting gradio bot ...")
self.gradio_bot.launch()
|