Develop

application/console.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+from dialogue_system.bot import Bot
+
+
+if __name__ == '__main__':
+    bot = Bot()
+
+    print('S: 料理のジャンルや場所をおっしゃってください。')
+    while True:
+        sent = input('U: ')
+        if sent == 'ありがとう':
+            print('S: どういたしまして')
+            break
+
+        reply = bot.reply(sent)
+        print('S: {0}'.format(reply))

application/plugins/slack.py

@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+
+from slackbot.bot import respond_to
+
+from dialogue_system.bot import Bot
+
+bots = {}
+
+
+def create_or_read(user_id):
+    return bots[user_id] if user_id in bots else Bot()
+
+
+def save_bot(bot, user_id):
+    bots[user_id] = bot
+
+
+@respond_to('(.*)')
+def food(message, something):
+    body = message.body
+    text, ts, user_id = body['text'], body['ts'], body['user']
+    bot = create_or_read(user_id)
+    reply_message = bot.reply(text)
+    save_bot(bot, user_id)
+    message.reply(reply_message)
+

application/server.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+import tornado.ioloop
+import tornado.web
+import tornado.websocket
+
+from dialogue_system.bot import Bot
+
+
+class MessageServer(tornado.websocket.WebSocketHandler):
+    bots = {}
+
+    def check_origin(self, origin):
+        return True
+
+    def open(self):
+        print('on open')
+        self.bots[self] = Bot()
+        self.write_message('料理のジャンルや場所をおっしゃってください。')
+
+    def on_message(self, message):
+        print('on message')
+        print(message)
+        bot = self.bots[self]
+        self.write_message(bot.reply(message))
+
+    def on_close(self):
+        print('on close')
+        del self.bots[self]
+
+application = tornado.web.Application([(r'/ws', MessageServer)])
+
+if __name__ == '__main__':
+    application.listen(8080)
+    tornado.ioloop.IOLoop.current().start()

application/slack_bot.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+from slackbot.bot import Bot
+
+
+def main():
+    bot = Bot()
+    bot.run()
+
+if __name__ == "__main__":
+    main()

application/slackbot_settings.py

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+import os
+
+
+API_TOKEN = os.environ.get('SLACK_API_KEY', '')
+
+default_reply = "スイマセン。其ノ言葉ワカリマセン"
+
+PLUGINS = [
+    'plugins',
+]

dialogue_system/backend/apis/docomo_dialogue.py

@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+import os
+
+from doco.client import Client
+
+
+class DocomoDialogAPI(object):
+
+    def __init__(self, api_key=None):
+        api_key = os.environ.get('DOCOMO_DIALOGUE_API_KEY', api_key)
+        self.__client = Client(apikey=api_key)
+
+    def reply(self, text):
+        response = self.__client.send(utt=text, apiname='Dialogue')
+        utt = response['utt']
+
+        return utt

modules/BackEnd/APIs/hotpepper.py → dialogue_system/backend/apis/hotpepper.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 import os
+
 import requests
 
 
@@ -8,7 +9,6 @@ class AreaNotFoundException(BaseException):
 
 
 class HotPepperGourmetAPI(object):
-
     BASE_URL = 'http://webservice.recruit.co.jp/hotpepper/{0}/v1/'
 
     def __init__(self, api_key=None):
@@ -79,4 +79,4 @@ def search_food(self, **kwargs):
 
 if __name__ == '__main__':
     api = HotPepperGourmetAPI()
-    api.search_food()
+    api.search_food()

dialogue_system/bot.py

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+from dialogue_system.dialogue_management.manager import DialogueManager
+from dialogue_system.language_generation.generator import LanguageGenerator
+from dialogue_system.language_understanding.language_understanding import RuleBasedLanguageUnderstanding
+
+
+class Bot(object):
+
+    def __init__(self):
+        self.generator = LanguageGenerator()
+        self.language_understanding = RuleBasedLanguageUnderstanding()
+        self.manager = DialogueManager()
+
+    def reply(self, sent):
+        dialogue_act = self.language_understanding.execute(sent)
+
+        self.manager.update_dialogue_state(dialogue_act)
+        sys_act_type = self.manager.select_action(dialogue_act)
+
+        sent = self.generator.generate_sentence(sys_act_type)
+
+        return sent

modules/DialogueManagement/manager.py → dialogue_system/dialogue_management/manager.py

@@ -1,6 +1,9 @@
 # -*- coding: utf-8 -*-
-from modules.DialogueManagement.state import DialogueState
-from modules.BackEnd.APIs.hotpepper import HotPepperGourmetAPI
+from copy import deepcopy
+
+from dialogue_system.dialogue_management.state import DialogueState
+from dialogue_system.backend.apis.hotpepper import HotPepperGourmetAPI
+from dialogue_system.backend.apis.docomo_dialogue import DocomoDialogAPI
 
 
 class DialogueManager(object):
@@ -12,9 +15,13 @@ def update_dialogue_state(self, dialogue_act):
         self.dialogue_state.update(dialogue_act)
 
     def select_action(self, dialogue_act):
-        from copy import deepcopy
         sys_act = deepcopy(dialogue_act)
-        if not self.dialogue_state.has('LOCATION'):
+        if dialogue_act['user_act_type'] == 'other':
+            api = DocomoDialogAPI()
+            reply = api.reply(dialogue_act['utt'])
+            sys_act['sys_act_type'] = 'CHAT'
+            sys_act['utt'] = reply
+        elif not self.dialogue_state.has('LOCATION'):
             sys_act['sys_act_type'] = 'REQUEST_LOCATION'
         elif not self.dialogue_state.has('GENRE'):
             sys_act['sys_act_type'] = 'REQUEST_GENRE'
@@ -28,5 +35,6 @@ def select_action(self, dialogue_act):
             restaurant = api.search_restaurant(area=area, food=food,budget=budget)
             sys_act['sys_act_type'] = 'INFORM_RESTAURANT'
             sys_act['restaurant'] = restaurant
+            self.dialogue_state.clear()
 
-        return sys_act
+        return sys_act

modules/DialogueManagement/state.py → dialogue_system/dialogue_management/state.py

@@ -12,7 +12,7 @@ def update(self, dialogue_act):
         self.__state['MAXIMUM_AMOUNT'] = dialogue_act.get('MAXIMUM_AMOUNT', self.__state['MAXIMUM_AMOUNT'])
 
     def has(self, name):
-        return self.__state[name] != None
+        return self.__state.get(name, None) != None
 
     def get_area(self):
         return self.__state['LOCATION']
@@ -23,6 +23,9 @@ def get_food(self):
     def get_budget(self):
         return self.__state['MAXIMUM_AMOUNT']
 
+    def clear(self):
+        self.__init__()
+
     def __str__(self):
         import pprint
         return pprint.pformat(self.__state)

dialogue_system/knowledge/__init__.py

@@ -0,0 +1 @@
+__author__ = 'h-nakayama'