import os
from crewai import Agent, Task, Crew, Process

# تنظیم کلید مدل (OpenAI / Anthropic / Gemini)
os.environ["OPENAI_API_KEY"] = "sk-..."

# ۱. تعریف پرسنل هوش مصنوعی
researcher = Agent(
    role="پژوهشگر ارشد بازار و فناوری",
    goal="یافتن جدیدترین فرصت‌های تجاری و رقبای نوظهور در صنعت هدف",
    backstory="شما تحلیل‌گری زبردست با ذهن جستجوگر هستید که داده‌های عمیق را استخراج می‌کنید.",
    verbose=True,
    memory=True
)

copywriter = Agent(
    role="استراتژیست محتوا و کپی‌رایتر",
    goal="نگارش مقاله تحلیلی اثرگذار و متن بازاریابی متناسب با لحن برند",
    backstory="شما نویسنده‌ای چیره‌دست با درک عمیق از روانشناسی مخاطب و اصول سئو هستید.",
    verbose=True,
    memory=True
)

reviewer = Agent(
    role="سردبیر و بازرس کنترل کیفیت (QA)",
    goal="بررسی صحت منابع، رعایت اصول نگارش و اطمینان از نبود ادعاهای کذب",
    backstory="شما سردبیری سخت‌گیر هستید که تنها محتواهای بی‌نقص و مستند را تأیید می‌کنید.",
    verbose=True
)

# ۲. تعریف تسک‌های کاری
t1 = Task(
    description="تحقیق روی ۵ ابزار نوین هوش مصنوعی در حوزه اتوماسیون سازمانی و استخراج مزیت‌های هرکدام.",
    expected_output="گزارش ساختاریافته شامل نام ابزار، مدل قیمت‌گذاری و نقطه تمایز.",
    agent=researcher
)

t2 = Task(
    description="نگارش یک گزارش راهبردی ۸۰۰ کلمه‌ای جذاب بر مبنای یافته‌های پژوهشگر برای انتشار در وبلاگ شرکت.",
    expected_output="متن مقاله با فرمت Markdown، تیترهای جذاب و دعوت به اقدام (CTA).",
    agent=copywriter
)

t3 = Task(
    description="بررسی کامل پیش‌نویس مقاله، رفع نواقص سئو و اعتبارسنجی ارجاعات.",
    expected_output="نسخه ویرایش‌شده نهایی آماده انتشار به همراه گزارش رضایت کیفی.",
    agent=reviewer
)

# ۳. راه‌اندازی شرکت
ai_company = Crew(
    agents=[researcher, copywriter, reviewer],
    tasks=[t1, t2, t3],
    process=Process.sequential,
    verbose=True
)

if __name__ == "__main__":
    result = ai_company.kickoff()
    print("\n\n======== خروجی نهایی تیم هوشمند ========\n")
    print(result)

# کلیدهای مدل‌های هوش مصنوعی
OPENAI_API_KEY=your_openai_key_here
ANTHROPIC_API_KEY=your_anthropic_key_here
GEMINI_API_KEY=your_gemini_key_here

# پایگاه داده برداری (Vector DB)
CHROMA_PERSIST_DIRECTORY=./chroma_db
QDRANT_HOST=localhost
QDRANT_PORT=6333

# اتصالات بیرونی
GITHUB_TOKEN=ghp_...
SLACK_WEBHOOK_URL=https://hooks.slack.com/services/...
SERPER_API_KEY=your_google_search_key