Update agent_manager.py

agent_manager.py

@@ -1,26 +1,26 @@
-from agents.strategy_agent import run_strategy
-from agents.ad_agent import run_ads
-from agents.copy_agent import run_copy
-from agents.email_agent import run_email
-from memory.database import log_action
+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_agents(niche, business_type):
-    summary = {}
-    # Strategy
-    strat = run_strategy(niche, business_type)
-    log_action("StrategyAgent", "generate_strategy", strat)
-    summary['strategy'] = strat
-    # Copy
-    copy = run_copy(strat)
-    log_action("CopyAgent", "generate_copy", copy)
-    summary['copy'] = copy
-    # Ads
-    ads = run_ads(strat)
-    log_action("AdAgent", "generate_ads", ads)
-    summary['ads'] = ads
-    # Email
-    emails = run_email(strat)
-    log_action("EmailAgent", "generate_emails", emails)
-    summary['emails'] = emails
-    return summary
+    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