Web   ·   Wiki   ·   Activities   ·   Blog   ·   Lists   ·   Chat   ·   Meeting   ·   Bugs   ·   Git   ·   Translate   ·   Archive   ·   People   ·   Donate
summaryrefslogtreecommitdiffstats
path: root/Speak.activity/brain.py
blob: e39e91e8f6c761b7d9904a56122eeb639170184c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
# HablarConSara.activity
# A simple hack to attach a chatterbot to speak activity
# Copyright (C) 2008 Sebastian Silva Fundacion FuenteLibre sebastian@fuentelibre.org
#
# Style and structure taken from Speak.activity Copyright (C) Joshua Minor
#
#     HablarConSara.activity is free software: you can redistribute it and/or modify
#     it under the terms of the GNU General Public License as published by
#     the Free Software Foundation, either version 3 of the License, or
#     (at your option) any later version.
# 
#     HablarConSara.activity is distributed in the hope that it will be useful,
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#     GNU General Public License for more details.
# 
#     You should have received a copy of the GNU General Public License
#     along with HablarConSara.activity.  If not, see <http://www.gnu.org/licenses/>.

import gtk
import gobject
from gettext import gettext as _

import logging
logger = logging.getLogger('speak')

from port.widgets import ToolComboBox

import bot.aiml
import voice

BOTS = {
    _('Spanish'): { 'name': 'Sara',
                    'brain': 'bot/sara.brn',
                    'predicates': { 'nombre_bot': 'Sara',
                                    'botmaster': 'la comunidad Azucar' } },
    _('English'): { 'name': 'Alice',
                    'brain': 'bot/alice.brn',
                    'predicates': { 'name': 'Alice',
                                    'master': 'the Sugar Community' } } }

DEFAULT = voice.defaultVoice()
if not BOTS.has_key(DEFAULT):
    DEFAULT = voice.allVoices()[_('English')]

class Toolbar(gtk.Toolbar):
    def __init__(self, activity):
        gtk.Toolbar.__init__(self)
        self.activity = activity
        self._kernels = {}

        self.voices = ToolComboBox()

        for lang in sorted(BOTS.keys()):
            self.voices.combo.append_item(lang, lang)

        self.voices.combo.set_active(0)
        self.voices.combo.connect('changed', self._changed_cb)
        self.voices.show()

        self.insert(self.voices, -1)

    def _load_brain(self, voice, sorry=''):
        def load_brain(old_cursor):
            is_first_session = (len(self._kernels) == 0)

            try:
                brain = BOTS[voice]
                logger.debug('Load bot: %s' % brain)

                kernel = bot.aiml.Kernel()
                kernel.loadBrain(brain['brain'])
                for name, value in brain['predicates'].items():
                    kernel.setBotPredicate(name, value)
                self._kernels[voice] = kernel
            finally:
                self.activity.set_cursor(old_cursor)

            if is_first_session:
                hello = _("Hello, I'am a robot \"%s\". Ask me any question.") \
                        % BOTS[voice]['name']
                hello += ' ' + sorry
                self.activity.face.say_notification(hello)
            elif sorry:
                self.activity.face.say_notification(sorry)

        old_cursor = self.activity.get_cursor()
        self.activity.set_cursor(gtk.gdk.WATCH)
        gobject.idle_add(load_brain, old_cursor)

    def _changed_cb(self, combo):
        voice = combo.props.value
        self.activity.change_voice(voice, False)
        if not self._kernels.has_key(voice):
            self._load_brain(voice)

    def update_voice(self):
        voice = self.activity.voice_combo.props.value.friendlyname
        new_voice = BOTS.has_key(voice) and voice or DEFAULT.friendlyname

        if voice != new_voice:
            self.activity.change_voice(new_voice, True)
        self.voices.combo.select(new_voice, silent_cb=self._changed_cb)

        sorry = _("Sorry, I can speak %s, let's speak %s instead.") \
                % (voice, new_voice)

        if not self._kernels.has_key(new_voice):
            self._load_brain(new_voice, (voice != new_voice) and sorry or '')
        elif voice != new_voice:
            self.activity.face.say_notification(sorry)

    def respond(self, text):
        voice = self.voices.combo.props.value
        return self._kernels[voice].respond(text)