PYTHON   21

bot.py

Guest on 6th June 2021 05:17:34 PM

  1. # Flask
  2. from chatterbot import ChatBot
  3. from chatterbot.trainers import ChatterBotCorpusTrainer
  4. # Database
  5. from .db import DB
  6.  
  7.  
  8. class Bot:
  9.     def __init__(self, *args, **kwargs):
  10.         db = DB()
  11.  
  12.         self.chatbot = chatbot = ChatBot(
  13.             'Chatbot Sudamericana',
  14.             read_only=True,
  15.             storage_adapter='chatterbot.storage.MongoDatabaseAdapter',
  16.             preprocessors=[
  17.                 'services.preprocessors.remove_stopwords'
  18.             ],
  19.             logic_adapters=[
  20.                 {
  21.                     'import_path': 'chatterbot.logic.BestMatch',
  22.                     'default_response': 'Disculpa pero no comprendo tu pregunta. Por favor reformula tu consulta. Utiliza frases como "ver mi estado de cuentas", "quiero ver mis pĆ³lizas", "quiero ver mis siniestros", "necesito un mĆ©dico" o "tengo un siniestro"'ro"',
  23.                 },
  24.                 {
  25.                     'import_path': 'services.adapters.BotLogicAdapter',
  26.                 },
  27.                 {
  28.                     'import_path': 'services.adapters.JustGreetingsAdapter',
  29.                 },
  30.                 {
  31.                     'import_path': 'services.adapters.GoodbayAdapter',
  32.                 },
  33.                 {
  34.                     'import_path': 'services.adapters.ContactRequestAdapter',
  35.                 },
  36.                 {
  37.                     'import_path': 'services.adapters.SinisterAdapter',
  38.                 },
  39.                 {
  40.                     'import_path': 'services.adapters.MedicalAdapter',
  41.                 },
  42.                 {
  43.                     'import_path': 'services.adapters.ExecutiveRequestAdapter',
  44.                 }
  45.             ],
  46.             database_uri=db.uri)
  47.  
  48.     def train(self):
  49.         trainer = ChatterBotCorpusTrainer(self.chatbot)
  50.  
  51.         trainer.train('./trainers/conversations1.json')
  52.         trainer.train('./trainers/conversations2.json')
  53.  
  54.     def response(self, question):
  55.         response = None
  56.  
  57.         try:
  58.             response = self.chatbot.get_response(question)
  59.         except:
  60.             print('Something error.')
  61.  
  62.         return resp

Raw Paste


Login or Register to edit or fork this paste. It's free.