diff --git a/naive-nlu/tree_nlu/cli.py b/naive-nlu/tree_nlu/cli.py index 20b53e3..a4eee32 100644 --- a/naive-nlu/tree_nlu/cli.py +++ b/naive-nlu/tree_nlu/cli.py @@ -1,6 +1,9 @@ import logging from .knowledge_base import KnowledgeBase -from .visualization import show_knowledge +from .visualization import ( + show_knowledge, + show_samples, +) from .tests import gac_100 from .modifiable_property import ( ModifiableProperty, @@ -31,6 +34,9 @@ def main(): if data == '/show': show_knowledge(knowledge) continue + elif data == '/samples': + show_samples(knowledge) + continue ret = knowledge.process(data) if ret: diff --git a/naive-nlu/tree_nlu/visualization.py b/naive-nlu/tree_nlu/visualization.py index 4a95a55..6f07325 100644 --- a/naive-nlu/tree_nlu/visualization.py +++ b/naive-nlu/tree_nlu/visualization.py @@ -1,3 +1,8 @@ def show_knowledge(knowledge): for key in knowledge.knowledge: print("\x1b[1m{}\x1b[0m {}".format(key, knowledge.knowledge[key])) + + +def show_samples(knowledge): + for example in knowledge.originals: + print("{}".format(example))