# elizabot.py a cheezy Eliza knock-off by Joe Strout <joe@strout.net>,
# updated by Jeff Epler <jepler@inetnebr.com> and landonb@douglas.bc.ca
import string
import pyTTS
tts = pyTTS.Create()

#set the speech rate
tts.Rate = 4

#set the speech volume percentage (0-100%)
tts.Volume = 60

#get a list of all the available voice actors
#print tts.GetVoiceNames()

#explicitly set a voice
tts.SetVoiceByName('MSMary')

#speak the text
#tts.Speak('This is the sound of my voice.')


# translate a string by replacing any words found in dict.keys()
def translate(str,dict):
	words = string.split(string.lower(str))
	keys = dict.keys();
	for i in range(0,len(words)):
		if words[i] in keys:
			words[i] = dict[words[i]]
	return string.join(words)


# dict converts what you say into repsonses e.g. "I am" --> "you are"
reflections = {
	"am" 	: "are",
	"was"	: "were",
	"i"	: "you",
	"i'd"	: "you would",
	"i've"	: "you have",
	"i'll"	: "you will",
	"my"	: "your",
	"are"	: "am",
	"you've": "I have",
	"you'll": "I will",
	"your"	: "my",
	"yours"	: "mine",
	"you"	: "me",
	"me"	: "you" }


def need(s): # for handling things you say starting with "I need"
	r = s[7:]
	if r[-1] == ".": r = r[:-1]
	r = translate(r, reflections)
	try: return ineed.popitem()[1] + r + '?'
	except: return saymore(s)


# dict of I need --> responses possible
ineed = {
	"1" : "Why do you need ",
	"2" : "Would it really help you to get ",
	"3" : "Are you sure you need "}


def saymore(s): # for encouraging more conversation - backchannel
	if s != "quit" :
		try: r = nomatch.popitem()[1]
		except: r = "That is all I can say today, Good-bye." 
		return r


# dict of responses when there is no match - just prompting more conversation	
nomatch = {
	"1" : "Please tell me more.",
	"2" : "Let's change focus a bit... Tell me about your family.",
	"3" : "Can you elaborate on that?",
	"4" : "I see.",
	"5" : "Very interesting.",
	"6" : "I see.  And what does that tell you?",
	"7" : "How does that make you feel?",
	"8" : "How do you feel when you say that?"}


# Main program for elizabot as updated 12/14/02 by landonb@douglas.bc.ca
tts.Speak('Talk to Elizabot by typing in plain English.  Enter "quit" when done.')
print 'Talk to Elizabot by typing in plain English.  Enter "quit" when done.'
tts.Speak("Hello.  What do you need to feel good today?")
print "Hello.  What do you need to feel good today?"
s = ""
while s != "quit":
	try: s = raw_input(":> ")
	except EOFError:
		s = "quit"
		#tts.Speak(s)
		print s
	if s != "": # avoids crashing on blank input line
		if s.lower()[:7] == "i need " : r = need(s)
		else: r = saymore(s)
		# change to introduce yourself
		if s.lower() == "who is bruce landon?" : 
		    r = "He is the Cognitive Psychology course instructor."
	if r != None :
            tts.Speak(r)
            print r #response to the typed input string