ui/utils.py

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)