Você está na página 1de 2

from 

chatterbot import ChatBot
from chatterbot.trainers import ListTrainer

bot = ChatBot('Test')

convI = ['que bom','ok','','voce pode me ajudar','Claro, como posso 
te ajudar?','oi, tudo bem?','tudo, obrigado por perguntar?','obriga
do','disponha','oi!','oi','olá','como vai','estou bem','tchau','até 
mais']

convV = ['quando será a prova','dia tal','a prova e dificil?','não, 
basta estudar!','tenho duvidas sobre o vestibular','Ok, vamos lá','
até quando será a inscrição para o vestibular?','ate dia tal','e as 
inscrições','dia tal','quando é o vestibular?','dia tal', 'quais cu
rsos são oferecidos?','os cursos são: ADS -Análise e Desenvolviment
o de Sistemas e Gestão de Negócios',]

convN = ['Qual é o seu nome?','pode me chamar de Bot.','quem és tu'
,'pode me chamar de Bot.','o que você faz?','Sou o estagiário virtu
al da Fatec RP. Posso te ajudar com algumas dúvidas sobre o vestibu
lar']

convPG = ['qual o valor da inscrição','R$75,00','quanto custa','R$7
5,00','quantocusta a inscrição','R$75,00','perdi meu boleto','sem p
roblemas, você pode fazer a reimpressão pelo site','meu boleto venc
eu','sem problemas, você pode fazer a reimpressão pelo site','meus 
dados estão errados','sem problemas, faremos a correção no ato da m
atrícula','meu nome esta errado','sem problemas, faremos a correção 
no ato da matrícula','meu rg esta errado','sem problemas, faremos a 
correção no ato da matrícula']

convRec=['onde posso pagar a taxa do vestibular?','faça o pagamento 
no banco','voces recebem?','por favor faça o pagamento no banco','p
osso pagar ai?','por favor faça o pagamento no banco','posso pagar 
na fatec','não, por favor faça o pagamento no banco']

convise=['quem tem direito a isenção','funcionários e seus dependen
tes tem direito','como pedir a isenção','ainda nao sei','quais docu
menos sao necessariospara solicitar a isenção','ainda não sei']

trainer = ListTrainer(bot)
trainer.train(convI)
trainer.train(convV)
trainer.train(convN)
trainer.train(convPG)
trainer.train(convRec)         
trainer.train(convise)
         
while True:
  quest = input('Você: ')
  response = bot.get_response(quest)
  if float(response.confidence)>0.05:
    print('FateBot: ', response)
  else:
    print('FateBot: Desculpe, não entendi, por favor refaça a pergu
nta.')

Você também pode gostar