diff --git a/naive-nlu/depth_meter.py b/naive-nlu/depth_meter.py index 8bc3073..7d8bb61 100644 --- a/naive-nlu/depth_meter.py +++ b/naive-nlu/depth_meter.py @@ -1,5 +1,12 @@ import sys def show_depth(depth: int): - sys.stdout.write("\r\x1b[K" + (u'█' * int(depth / 2)) + (u'▌' * int(depth % 2)) + "\x1b[7m \x1b[0m\b") + multiplier = 3 + max_depth = 5 + offset = int((max_depth - depth) / (2 / multiplier)) + + depth = depth * multiplier + offset -= int(depth % 2) + sys.stdout.write("\r|\x1b[K" + (u'█' * int(depth / 2)) + (u'▌' * int(depth % 2)) + ' ' * offset + "|\x1b[7m \x1b[0m\b") + sys.stdout.flush()