Create modules/agents.py

modules/agents.py

@@ -0,0 +1,105 @@
+import asyncio
+import random
+from datetime import datetime
+from ai_engine import engine
+from jarvis_links import get_link, get_random_template
+
+class TrendHijacker:
+    name = "TrendHijacker"
+    icon = "🔥"
+    TREND_SEEDS = [
+        "website gratis 2025", "cara bikin toko online", "hosting murah Indonesia",
+        "bisnis online modal kecil", "tips freelancer Indonesia", "kerja remote dari rumah",
+        "jualan online shopee tips", "produk viral shopee", "diskon shopee hari ini",
+    ]
+    async def run(self, custom_topic=None, **kwargs):
+        topic = custom_topic or random.choice(self.TREND_SEEDS)
+        prompt = f"""Kamu adalah content strategist viral Indonesia.
+Topik trending: "{topic}"
+Buat 3 ide konten viral untuk promosi affiliate Hostinger dan Shopee.
+Format: JUDUL | PLATFORM | HOOK | KONTEN | CTA dengan link affiliate.
+Bahasa Indonesia casual."""
+        result = await engine.think(prompt)
+        return {"module": self.name, "topic": topic, "timestamp": datetime.now().isoformat(), "output": result["response"], "provider": result["provider"]}
+
+class ViralPredictor:
+    name = "ViralPredictor"
+    icon = "📊"
+    async def run(self, content="contoh konten affiliate", **kwargs):
+        prompt = f"""Analisa potensi viral konten ini di Indonesia:
+KONTEN: {content}
+Berikan: VIRAL SCORE (0-100) | PLATFORM TERBAIK | KEKUATAN | KELEMAHAN | REKOMENDASI
+Bahasa Indonesia."""
+        result = await engine.think(prompt)
+        return {"module": self.name, "timestamp": datetime.now().isoformat(), "output": result["response"], "provider": result["provider"]}
+
+class GhostNetwork:
+    name = "GhostNetwork"
+    icon = "👻"
+    async def run(self, content="konten affiliate default", **kwargs):
+        hostinger_link = get_link("hostinger")
+        shopee_link = get_link("shopee")
+        prompt = f"""Adaptasi konten ini untuk Instagram, TikTok, Twitter, Facebook, WhatsApp, Telegram:
+KONTEN: {content}
+Embed link natural: Hostinger: {hostinger_link} | Shopee: {shopee_link}
+Sertakan hashtag dan waktu posting optimal WIB."""
+        result = await engine.think(prompt)
+        return {"module": self.name, "timestamp": datetime.now().isoformat(), "output": result["response"], "provider": result["provider"]}
+
+class Scout:
+    name = "Scout"
+    icon = "🔍"
+    async def run(self, target="hostinger competitor indonesia", **kwargs):
+        prompt = f"""Riset mendalam: "{target}"
+Berikan: OVERVIEW | PAIN POINTS | KOMPETITOR | PELUANG | KEYWORDS (10) | CONTENT IDEAS (5)
+Fokus pasar Indonesia."""
+        result = await engine.think(prompt)
+        return {"module": self.name, "target": target, "timestamp": datetime.now().isoformat(), "output": result["response"], "provider": result["provider"]}
+
+class Writer:
+    name = "Writer"
+    icon = "✍️"
+    async def run(self, topic="hosting murah", content_type="social_caption", platform="hostinger", **kwargs):
+        link = get_link(platform)
+        prompt = f"""Tulis {content_type} bahasa Indonesia tentang: "{topic}"
+Affiliate: {platform.upper()} | Link: {link}
+Syarat: casual, embed link natural, hook kuat, CTA tidak memaksa, tambah emojis."""
+        result = await engine.think(prompt)
+        return {"module": self.name, "topic": topic, "timestamp": datetime.now().isoformat(), "output": result["response"], "provider": result["provider"]}
+
+class Publisher:
+    name = "Publisher"
+    icon = "📢"
+    async def run(self, content="konten affiliate", **kwargs):
+        prompt = f"""Buat jadwal publishing 7 hari untuk konten:
+{content[:300]}
+Format tabel: HARI | PLATFORM | JAM WIB | TIPE KONTEN | CATATAN
+Tambahkan checklist sebelum posting."""
+        result = await engine.think(prompt)
+        return {"module": self.name, "timestamp": datetime.now().isoformat(), "output": result["response"], "provider": result["provider"]}
+
+class Analyst:
+    name = "Analyst"
+    icon = "📈"
+    async def run(self, metrics=None, **kwargs):
+        metrics_str = str(metrics) if metrics else "Belum ada data. Buat rekomendasi KPI awal untuk affiliate marketing."
+        prompt = f"""Analisa performa affiliate marketing:
+DATA: {metrics_str}
+Berikan: OVERVIEW | TOP PERFORMING | UNDERPERFORMING | INSIGHT | 5 LANGKAH OPTIMASI | TARGET MINGGU DEPAN"""
+        result = await engine.think(prompt)
+        return {"module": self.name, "timestamp": datetime.now().isoformat(), "output": result["response"], "provider": result["provider"]}
+
+MODULES = {
+    "trend": TrendHijacker(),
+    "viral": ViralPredictor(),
+    "ghost": GhostNetwork(),
+    "scout": Scout(),
+    "writer": Writer(),
+    "publisher": Publisher(),
+    "analyst": Analyst(),
+}
+
+async def run_module(module_name: str, **kwargs) -> dict:
+    if module_name not in MODULES:
+        return {"error": f"Module '{module_name}' not found"}
+    return await MODULES[module_name].run(**kwargs)