def responder(self, afirmacao_usuario): edgar = Edgar() edgar.escute(afirmacao_usuario) resposta_edgar = edgar.responda() resposta = { 'resposta': resposta_edgar, 'afirmação': afirmacao_usuario, 'confianca': edgar.resposta.confidence, 'hora': datetime.utcnow() } return resposta
import logging from os import system from controllers.botController import BotController from models.edgar import Edgar from models.usuario import Usuario from controllers.usuarioController import UsuarioController def limpar_log(): logger = logging.getLogger() logger.setLevel(logging.CRITICAL) system('clear') edgar = Edgar() limpar_log() bot_controller = BotController(edgar) nome_usuario = UsuarioController.get_nome() usuario = Usuario(nome_usuario) usuario_controller = UsuarioController(usuario) limpar_log() bot_controller.apresentar() while not usuario.quer_sair: usuario_controller.get_afirmacao() if not usuario.quer_sair: bot_controller.escutar(usuario.afirmacao) if usuario.quer_corrigir():
def __init__(self): self.__dados_treino = [] self.__edgar = Edgar() self.__treinador = ListTrainer(self.__edgar.bot) self.__treinador_padrao = ChatterBotCorpusTrainer(self.__edgar.bot)
self.treinar([ 'Olá', 'Oi, tudo bem?', 'Tudo ótimo e com você?', 'Eu vou bem', 'Isso é ótimo de se ouvir', 'Obrigado', 'Não há de quê', ]) self.treinar([ 'Oi', 'Oi, tudo bem?', 'Não estou muito bem', 'É uma pena' ]) self.treinar([ 'Eai', 'Opa, eai tudo bom?', 'Tudo ótimo e contigo?', 'Eu estou ótimo' ]) edgar = Edgar() if __name__ == '__main__' and not edgar.treinado: treinamento = Treinamento() treinamento.saudacao() treinamento.padrao_br() edgar.treinado = True
def corrigir(self, afirmacao_usuario, resposta_valida): edgar = Edgar() edgar.aprenda(resposta_valida, afirmacao_usuario)