import os
import warnings
warnings.filterwarnings('ignore')
os.chdir('../') # set the working directory to the root of the project
from symai import *
from symai.components import *
from IPython.display import display
Writing a chatbot is fairly easy with our framework. All we need to do is basically derive from the ChatBot class and implement the forward method. The base class ChatBot has already some helper capabilities and context detection dictionaries. All we have to do is use the self.narrate method to instruct our chatbot to say what we want.
Afterwards, we can use the self.context_choice method to classify the context of the user input. This is done by using a dictionary of context keywords. The self.context_choice method returns the context key that matches the user input. This key can then be used to determine the next action / condition of the chatbot.
See the following example:
By creating an instance of the SymbiaChat and calling the forward method, we can start a chat with our chatbot. The forward method takes a user input and returns a chatbot response.
from symai.chat import ChatBot
from symai.interfaces import Interface
class SymbiaChat(ChatBot):
def forward(self) -> str:
message = self.narrate('Symbia introduces herself, writes a greeting message and asks how to help.')
while True:
# query user
usr = self.input(message)
# detect context
ctxt = self.context_choice(usr)
if 'option 3' in ctxt: # exit
self.narrate('Symbia writes goodbye message.', end=True)
break # end chat
elif 'option 4' in ctxt: # help
message = self.narrate('Symbia writes for each capability one sentence.',
context=self.capabilities)
elif 'option 1' in ctxt: # chit chat
message = self.narrate('Symbia replies to the user question in a casual way.')
elif 'option 2' in ctxt:
# detect command
option = self.capabilities_choice(usr)
if 'option 1' in option:
q = usr.extract('user query request')
rsp = self.search(q)
message = self.narrate('Symbia replies to the user based on the online search results.',
context=rsp)
elif 'option 2' in option:
q = usr.extract('URL')
site = self.crawler(q)
site.save('tmp.html')
message = self.narrate('Symbia explains that the website is downloaded to the `tmp.html` file.')
elif 'option 3' in option:
pass
# TODO ...
from symai.chat import SymbiaChat
chat = SymbiaChat()
chat()
Symbia: Goodbye! If you have any more questions in the future, don't hesitate to reach out. Have a great day!
The above bot can answer some trivia, use the Google search engine to retrieve information and