History map

def insert_departments(history_map, notes_text): from io import StringIO

# Convert history_map to a dictionary for lookup
user_department_map = [entry["department"] for entry in history_map]
user_ids = [entry["userid"] for entry in history_map]

# Read notes line by line
lines = notes_text.strip().split("\n")
updated_lines = []

user_index = 0
i = 0
while i < len(lines):
    line = lines[i]
    updated_lines.append(line)

    if line.startswith("userid:"):
        # After userid line, insert department line
        dept = history_map[user_index]["department"]
        updated_lines.append(f"department: {dept}")
        user_index += 1

    i += 1

return "\n".join(updated_lines)