| import os |
| import re |
|
|
| def format_directory(directory): |
| output = [] |
| def helper(directory, level, output): |
| files = os.listdir(directory) |
| for i, item in enumerate(files): |
| is_folder = os.path.isdir(os.path.join(directory, item)) |
| joiner = "├── " if i < len(files) - 1 else "└── " |
| item_html = item + "/" if is_folder else f"<a href='file={os.path.join(directory, item)}'>{item}</a>" |
| output.append("│ " * level + joiner + item_html) |
| if is_folder: |
| helper(os.path.join(directory, item), level + 1, output) |
| output.append(os.path.basename(directory) + "/") |
| helper(directory, 1, output) |
| return "\n".join(output) |
|
|
| DOWNLOAD_OUTPUTS_JS = """ |
| () => { |
| const a = document.createElement('a'); |
| a.href = 'file=outputs.zip'; |
| a.download = 'outputs.zip'; |
| document.body.appendChild(a); |
| a.click(); |
| document.body.removeChild(a); |
| }""" |
|
|
| def remove_color(text): |
| ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])') |
| return ansi_escape.sub('', text) |
