-
Notifications
You must be signed in to change notification settings - Fork 21
/
responses.py
50 lines (39 loc) · 1.71 KB
/
responses.py
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
import re
def process_message(message, response_array, response):
# Splits the message and the punctuation into an array
list_message = re.findall(r"[\w']+|[.,!?;]", message.lower())
# Scores the amount of words in the message
score = 0
for word in list_message:
if word in response_array:
score = score + 1
# Returns the response and the score of the response
# print(score, response)
return [score, response]
def get_response(message):
# Add your custom responses here
response_list = [
process_message(message, ['hello', 'hi', 'hey'], 'Hey there!'),
process_message(message, ['bye', 'goodbye'], 'Goodbye!'),
process_message(message, ['how', 'are', 'you'], 'I\'m doing fine thanks!'),
process_message(message, ['your', 'name'], 'My name is Mario, nice to meet you!'),
process_message(message, ['help', 'me'], 'I will do my best to assist you!')
# Add more responses here
]
# Checks all of the response scores and returns the best matching response
response_scores = []
for response in response_list:
response_scores.append(response[0])
# Get the max value for the best response and store it into a variable
winning_response = max(response_scores)
matching_response = response_list[response_scores.index(winning_response)]
# Return the matching response to the user
if winning_response == 0:
bot_response = 'I didn\'t understand what you wrote.'
else:
bot_response = matching_response[1]
print('Bot response:', bot_response)
return bot_response
# Test your system
# get_response('What is your name bruv?')
# get_response('Can you help me with something please?')