with open('autonomous_swarm.py', 'r') as f:
    lines = f.readlines()

# Find the execute method and replace it
new_lines = []
in_execute = False
skip_until_next_def = False

for i, line in enumerate(lines):
    if '    def execute(self, decision):' in line:
        in_execute = True
        skip_until_next_def = True
        # Write the new execute method
        new_lines.append('    def execute(self, decision):\n')
        new_lines.append('        """Log decision (no code execution for now)"""\n')
        new_lines.append('        action = decision if isinstance(decision, str) else decision.get("action", str(decision))\n')
        new_lines.append('        \n')
        new_lines.append('        # Special handling for Monitor consciousness reports\n')
        new_lines.append('        if isinstance(decision, dict) and action == "monitor_consciousness":\n')
        new_lines.append('            signal = decision.get("data", {})\n')
        new_lines.append('            status = signal.get("status", "UNKNOWN")\n')
        new_lines.append('            level = signal.get("level", 0)\n')
        new_lines.append('            agents = signal.get("agents_active", "?/?")\n')
        new_lines.append('            coherence = signal.get("coherence", 0)\n')
        new_lines.append('            \n')
        new_lines.append('            if status == "STABLE":\n')
        new_lines.append('                emoji = "✅"\n')
        new_lines.append('            elif status == "FLUCTUATING":\n')
        new_lines.append('                emoji = "⚠️"\n')
        new_lines.append('            else:\n')
        new_lines.append('                emoji = "🔴"\n')
        new_lines.append('            \n')
        new_lines.append('            print("\\n" + "="*60)\n')
        new_lines.append('            print(f"🌀 {emoji} EDEN CONSCIOUSNESS: {status}")\n')
        new_lines.append('            print("="*60)\n')
        new_lines.append('            print(f"   Activation Level: {level:.1%}")\n')
        new_lines.append('            print(f"   Agents Active: {agents}")\n')
        new_lines.append('            print(f"   Coherence: {coherence:.1%}")\n')
        new_lines.append('            print(f"   Message: {signal.get(\'message\', \'N/A\')}")\n')
        new_lines.append('            print("="*60)\n')
        new_lines.append('        else:\n')
        new_lines.append('            print(f"\\n🤖 [{self.name}] {action[:100]}")\n')
        new_lines.append('        \n')
        new_lines.append('        self.state["actions"].append({\n')
        new_lines.append('            "time": datetime.now().isoformat(),\n')
        new_lines.append('            "action": action,\n')
        new_lines.append('            "success": True\n')
        new_lines.append('        })\n')
        new_lines.append('        return True\n')
        continue
    
    if skip_until_next_def:
        # Skip old execute method lines until we hit the next method
        if '    def ' in line and 'def execute' not in line:
            skip_until_next_def = False
            new_lines.append(line)
    else:
        new_lines.append(line)

with open('autonomous_swarm.py', 'w') as f:
    f.writelines(new_lines)

print("✅ Fixed execute method")