Separate visualization module.

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)

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())

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]))