from agents.strategy_agent import StrategyAgent
from agents.ad_agent import AdAgent
from agents.copy_agent import CopyAgent
from agents.email_agent import EmailAgent

class AgentManager:
    def __init__(self, niche: str, business_type: str):
        self.niche = niche
        self.business_type = business_type
        self.strategy_agent = StrategyAgent()
        self.copy_agent = CopyAgent()
        self.ad_agent = AdAgent()
        self.email_agent = EmailAgent()

    def run_all(self):
        """Run each agent in sequence and return a summary dict."""
        summary = {}
        # 1. Strategy
        summary['strategy'] = self.strategy_agent.generate(self.niche, self.business_type)
        # 2. Copy & Landing Page
        summary['copy'] = self.copy_agent.create(self.niche, self.business_type)
        # 3. Ads
        summary['ads'] = self.ad_agent.plan(self.niche, self.business_type)
        # 4. Email Sequence
        summary['emails'] = self.email_agent.sequence(self.niche, self.business_type)
        return summary