feat: add draw_module_tree()

README.md

@@ -138,6 +138,11 @@ def buggy_foo():
     foo(a=b)
 ```
 
+## Visualize your codebase
+
+You can put the function `draw_module_tree()` in any of your modules to visualize the tree structure of it. Call it in the `__init__.py` of your package is recommended.
+
+
 ## Installation
 
 You can install Moduler by

moduler/gui.py

@@ -1,5 +1,19 @@
 from moduler import Struct
 import plotly.graph_objects as go
+import inspect
+from moduler.core import build_module_tree
+
+def draw_module_tree():
+    # Get current module by inspecting the stack
+    frame = inspect.stack()[1]
+    module = inspect.getmodule(frame[0])
+    struct = build_module_tree(module)
+    draw_treemap(struct)
+
+
+"""
+## Draw treemap
+"""
 
 def draw_treemap(struct: Struct):
     ids, labels, parents, texts = extract_tree_ingredients(struct)
@@ -97,7 +111,6 @@ def hypenate_texts(texts: str, line_width=40):
 
 
 if __name__ == '__main__':
-    from moduler.core import build_module_tree
     import moduler
     struct = build_module_tree(moduler)
     draw_treemap(struct)