Generate a Beautiful Folder Tree Visualization in Python

import os
from pathlib import Path

class FolderTree:
    def __init__(self, root_path=".", ignore_list=None, max_depth=3):
        self.root = Path(root_path)
        self.ignore = set(ignore_list or [".git", "__pycache__", ".DS_Store"])
        self.max_depth = max_depth
        
    def generate(self):
        tree_lines = [f"{self.root.name}/"]
        self._add_children(self.root, "", 0, tree_lines)
        return "\n".join(tree_lines)
    
    def _add_children(self, path, prefix, depth, tree_lines):
        if depth >= self.max_depth:
            return
        try:
            items = sorted(path.iterdir(), key=lambda x: (x.is_file(), x.name))
            items = [item for item in items if item.name not in self.ignore]
            for i, item in enumerate(items):
                is_last = i == len(items) - 1
                connector = "└── " if is_last else "├── "
                tree_lines.append(f"{prefix}{connector}{item.name}")
                if item.is_dir() and depth < self.max_depth - 1:
                    extension = "    " if is_last else "│   "
                    self._add_children(item, prefix + extension, depth + 1, tree_lines)
        except PermissionError:
            tree_lines.append(f"{prefix}└── [Permission Denied]")

if __name__ == "__main__":
    import sys
    path = sys.argv[1] if len(sys.argv) > 1 else "."
    depth = int(sys.argv[2]) if len(sys.argv) > 2 else 3
    tree = FolderTree(path, max_depth=depth)
    print(tree.generate())