diff --git a/naive-nlu/tree_nlu/cli.py b/naive-nlu/tree_nlu/cli.py index bd4e894..20b53e3 100644 --- a/naive-nlu/tree_nlu/cli.py +++ b/naive-nlu/tree_nlu/cli.py @@ -1,5 +1,6 @@ import logging from .knowledge_base import KnowledgeBase +from .visualization import show_knowledge from .tests import gac_100 from .modifiable_property import ( ModifiableProperty, @@ -11,10 +12,6 @@ from .modifiable_property import ( bye_phrases = ['bye', 'exit'] -def debug(knowledge): - for key in knowledge.knowledge: - print("\x1b[1m{}\x1b[0m {}".format(key, knowledge.knowledge[key])) - def main(): logging.getLogger().setLevel(logging.INFO) @@ -31,8 +28,8 @@ def main(): if not data: continue - if data == '/debug': - debug(knowledge) + if data == '/show': + show_knowledge(knowledge) continue ret = knowledge.process(data) diff --git a/naive-nlu/tree_nlu/tests/gac_100.py b/naive-nlu/tree_nlu/tests/gac_100.py index 34f4c3b..caaa80b 100644 --- a/naive-nlu/tree_nlu/tests/gac_100.py +++ b/naive-nlu/tree_nlu/tests/gac_100.py @@ -1,6 +1,7 @@ import logging from ..knowledge_base import KnowledgeBase from ..utils.visuals import show_progbar +from ..visualization import show_knowledge def _assert(args): assert(args) @@ -745,3 +746,7 @@ def main(): print("\r\x1b[K", end='') return knowledge + + +if __name__ == '__main__': + show_knowledge(main()) diff --git a/naive-nlu/tree_nlu/visualization.py b/naive-nlu/tree_nlu/visualization.py new file mode 100644 index 0000000..4a95a55 --- /dev/null +++ b/naive-nlu/tree_nlu/visualization.py @@ -0,0 +1,3 @@ +def show_knowledge(knowledge): + for key in knowledge.knowledge: + print("\x1b[1m{}\x1b[0m {}".format(key, knowledge.knowledge[key]))