| 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() | |