diff --git a/bot-aiml.py b/bot-aiml.py index 0b9fcf5..a277beb 100644 --- a/bot-aiml.py +++ b/bot-aiml.py @@ -1,6 +1,7 @@ #!/usr/bin/env python2 # encoding=utf8 import os +import os.path import sys reload(sys) sys.setdefaultencoding('utf8') @@ -9,12 +10,25 @@ import time from datetime import datetime import pyAIML +import atexit TOKEN = os.environ['TELEGRAM_TOKEN'] +AIMLName = "comperBot" +BootName = "belinda" +BRAINNAME = "brain.sav" + k = pyAIML.Kernel() -k.learn(os.path.join(os.path.dirname(__file__), 'AIML/comperBot/') + "*.aiml") -k.setBotPredicate("bot_name", "Belinda") +if os.path.isfile(BRAINNAME): + k.loadBrain(BRAINNAME) +k.setTextEncoding("UTF-8") +k.learn(os.path.join(os.path.dirname(__file__), 'AIML/', AIMLName + "/") + "*.aiml") +k.loadSubs(os.path.join(os.path.dirname(__file__), 'AIML/', AIMLName + "/") + AIMLName+ ".ini") +k.setBotPredicate("bot_name", BootName) + + + +atexit.register(lambda : k.saveBrain(BRAINNAME)) def listener(*messages): f = open('chat.log', 'a') @@ -27,7 +41,7 @@ def listener(*messages): chatid = m.chat.id if m.content_type == 'text': text = m.text - response = k.respond(text) + response = k.respond(text, chatid) tb.send_message(chatid, "Escribiste: " + text) tb.send_message(chatid, response) f.write(str(datetime.now()) + "\t Chat: " + str(chatid) + "\t Msg: " + text + "\t Respuesta: " + response + "\n")