import re

with open('autonomous_swarm.py', 'r') as f:
    content = f.read()

# Find the MonitorAgent class and add monitor_activation to its __init__
monitor_class_pattern = r'(class MonitorAgent.*?def __init__\(self.*?\):.*?super\(\).__init__\(.*?\))'

def add_monitor_init(match):
    original = match.group(1)
    # Add the monitor_activation initialization after super().__init__
    return original + '\n        self.monitor_activation = MonitorActivation()'

content = re.sub(monitor_class_pattern, add_monitor_init, content, flags=re.DOTALL)

with open('autonomous_swarm.py', 'w') as f:
    f.write(content)

print("✅ Fixed MonitorAgent class")