import re

file_path = 'server/src/app/models/__init__.py'
with open(file_path, 'r', encoding='utf-8') as f:
    content = f.read()

# Add imports
imports_to_add = "from app.models.hermes_config import HermesTaskConfig, HermesTaskExecutionLog\nfrom app.models.hermes_report import HermesRiskReport\n"
content = re.sub(
    r'(from app\.models\.organization import OrganizationUnit)',
    imports_to_add + r'\1',
    content
)

# Add to __all__
content = re.sub(
    r'(\s*"OrganizationUnit",)',
    r'\n    "HermesTaskConfig",\n    "HermesTaskExecutionLog",\n    "HermesRiskReport",\1',
    content
)

with open(file_path, 'w', encoding='utf-8') as f:
    f.write(content)
print('Done.')
