import os

def fix_indentation(path):
    with open(path, 'r') as f:
        lines = f.readlines()
        
    new_lines = []
    in_imports = False
    
    for line in lines:
        stripped = line.strip()
        if stripped == "imports:":
            in_imports = True
            new_lines.append("imports:\n")
            continue
            
        if in_imports:
            if not stripped.startswith("-"):
                if stripped and not stripped.startswith("#"):
                    in_imports = False
                new_lines.append(line)
            else:
                # Force 2 spaces indent
                # content is stripped, so just prepend "  "
                new_lines.append(f"  {stripped}\n")
        else:
            new_lines.append(line)
            
    with open(path, 'w') as f:
        f.writelines(new_lines)

def process_directory(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(".yaml"):
                fix_indentation(os.path.join(root, file))

process_directory("schemas/20251121/linkml/modules/classes")