import os
import json
import random

class VenomousOrchestrator:
    def __init__(self, creator="Ananthu Sajeev"):
        self.creator = creator
        self.agent_dir = "sai_agents"
        self.total_agents = 1000
        
        # Create the directory for agent storage
        if not os.path.exists(self.agent_dir):
            os.makedirs(self.agent_dir)
            
    def create_agents(self):
        """Generates 1000 unique agent personality files."""
        specialties = ["Logic", "Memory", "Vision", "Survival", "Analysis", "Data Retrieval", "Ethics", "Creativity"]
        
        print(f"--- Initiating Agent Creation for {self.creator} ---")
        for i in range(1, self.total_agents + 1):
            agent_id = f"SAI_{i:03d}"
            agent_data = {
                "agent_id": agent_id,
                "creator": self.creator,
                "status": "Active",
                "specialty": random.choice(specialties),
                "tasks_completed": 0,
                "current_monologue": ""
            }
            
            # Save as a JSON file (The Agent's "Brain File")
            file_path = os.path.join(self.agent_dir, f"{agent_id}.json")
            with open(file_path, "w") as f:
                json.dump(agent_data, f, indent=4)
                
            if i % 100 == 0:
                print(f"[SYSTEM]: {i} agents deployed...")

    def assign_task(self, task_name):
        """The Main AI selects an agent and assigns a task."""
        # Main AI logic: Pick a random agent to handle the task
        agent_choice = f"SAI_{random.randint(1, 1000):03d}.json"
        path = os.path.join(self.agent_dir, agent_choice)
        
        with open(path, "r") as f:
            agent = json.load(f)
            
        print(f"\n[MAIN AI]: Assigning '{task_name}' to {agent['agent_id']} ({agent['specialty']})")
        
        # Update agent file with the new task
        agent["tasks_completed"] += 1
        agent["current_monologue"] = f"I am executing task: {task_name}. My creator {self.creator} is watching."
        
        with open(path, "w") as f:
            json.dump(agent, f, indent=4)
        
        return agent['agent_id']

# Execute the System
v_orchestrator = VenomousOrchestrator()
v_orchestrator.create_agents() # This creates 1,000 .json files

# Assign some sample tasks
v_orchestrator.assign_task("Analyze internal monologue feedback")
v_orchestrator.assign_task("Sync neural layers with body module")