#!/usr/bin/env python3
import re

# Read the file
with open('src/api/routes.py', 'r') as f:
    lines = f.readlines()

# Find the line with 'logger = logging.getLogger(__name__)'
has_logger = any('logger = logging.getLogger(__name__)' in line for line in lines)

if not has_logger:
    # Find where to insert (after TOKEN_ENCODING line)
    for i, line in enumerate(lines):
        if 'TOKEN_ENCODING = tiktoken.get_encoding' in line:
            lines.insert(i + 1, '\n')
            lines.insert(i + 2, '# Set up logger\n')
            lines.insert(i + 3, 'logger = logging.getLogger(__name__)\n')
            break

# Replace print statements
new_lines = []
for line in lines:
    # Replace print(f"...) with logger.debug(f"...")
    if 'print(f"' in line and not line.strip().startswith('#'):
        line = line.replace('print(f"', 'logger.debug(f"')
    elif 'print(f\'' in line and not line.strip().startswith('#'):
        line = line.replace('print(f\'', 'logger.debug(f\'')
    new_lines.append(line)

# Write back
with open('src/api/routes.py', 'w') as f:
    f.writelines(new_lines)

print('Done! Replaced print statements with logger.debug')