import random
import os
from groq import Groq
from config import GROQ_API_KEY

class ContentGenerator:
    def __init__(self):
        """Инициализация генератора контента"""
        print("🔄 Инициализация ContentGenerator...")
        
        # Проверяем, есть ли ключ Groq
        if GROQ_API_KEY:
            try:
                self.client = Groq(api_key=GROQ_API_KEY)
                self.model = "llama-3.3-70b-versatile"
                self.use_groq = True
                print("✅ Groq API подключен!")
            except Exception as e:
                print(f"⚠️ Ошибка подключения Groq: {e}")
                self.use_groq = False
        else:
            print("⚠️ Groq API ключ не найден, использую шаблоны")
            self.use_groq = False
        
        # Шаблоны для запасного варианта (если Groq не работает)
        self.templates = {
            'instagram': [
                "🔥 {business_name} — это не просто бизнес, это миссия!\n\n"
                "Рассказываю, как мы пришли к тому, что делаем сегодня.\n"
                "В {years} лет мы прошли путь от идеи до реализации.\n\n"
                "А вы знали, что {business_name} помогает {problem_solution}?\n\n"
                "👉 Напиши в комментариях, сталкивался ли ты с этой проблемой!\n\n"
                "#бизнес #предприниматель #{hashtag} #историяуспеха",
                
                "📊 {business_name} — цифры, которые вдохновляют!\n\n"
                "За {years} лет мы:\n"
                "✅ Помогли {clients} клиентам\n"
                "✅ Увеличили прибыль на {profit}%\n"
                "✅ Выросли в {growth} раз\n\n"
                "Хочешь так же? Пиши в DM! 💬\n\n"
                "#бизнес #предпринимательство #{hashtag}"
            ],
            
            'linkedin': [
                "🚀 Как {business_name} изменил рынок в {years} году?\n\n"
                "Когда мы начинали, казалось, что {problem} — это непреодолимая стена.\n"
                "Но мы нашли решение: {solution}\n\n"
                "Результат: {result} 📈\n\n"
                "Какую бизнес-задачу вы решаете сегодня? Делитесь в комментариях! 💬\n\n"
                "#бизнес #лидерство #{hashtag} #growth",
                
                "💡 5 уроков за {years} лет в бизнесе\n\n"
                "1️⃣ {lesson1}\n"
                "2️⃣ {lesson2}\n"
                "3️⃣ {lesson3}\n"
                "4️⃣ {lesson4}\n"
                "5️⃣ {lesson5}\n\n"
                "Какой урок оказался для вас самым важным? 👇\n\n"
                "#бизнес #урокипредпринимательства #{hashtag}"
            ],
            
            'telegram': [
                "💬 Привет, друзья! 👋\n\n"
                "Сегодня хочу поделиться мыслями о {business_name}.\n\n"
                "Многие спрашивают: «Как ты пришел к этому?»\n"
                "Ответ прост: {short_story}\n\n"
                "А какой бизнес вдохновляет тебя? Пиши в комментариях! 👇\n\n"
                "#бизнес #предприниматель #{hashtag}",
                
                "📌 Важный пост для предпринимателей!\n\n"
                "В {business_name} мы верим, что {belief}\n\n"
                "Поэтому сегодня делюсь {tip} советом:\n"
                "{advice}\n\n"
                "А какой совет ты бы дал себе в начале пути? 🔥\n\n"
                "#бизнес #советы #{hashtag}"
            ],
            
            'twitter': [
                "🧵 Нить: Как {business_name} за {years} лет изменил подход к {industry}\n\n"
                "1/ Мы начинали с {start_point}\n"
                "2/ Главная проблема была: {problem}\n"
                "3/ Решение нашли через {solution}\n"
                "4/ Результат: {result}\n"
                "5/ Вывод: {lesson}\n\n"
                "Согласны? Репост, если цените бизнес-мышление! 🔄\n\n"
                "#бизнес #{hashtag}",
                
                "💭 Мысль дня от {business_name}:\n\n"
                "{quote}\n\n"
                "Как думаете, это правда? 👇\n\n"
                "#бизнес #мысли #{hashtag}"
            ],
            
            'ad': [
                "🎯 Хочешь получить {benefit}?\n\n"
                "В {business_name} мы знаем, как это сделать!\n\n"
                "За {years} лет мы помогли {clients} клиентам достичь {result}\n\n"
                "Напиши «ХОЧУ» в комментариях или DM, и я расскажу как! 💬\n\n"
                "#реклама #бизнес #{hashtag}"
            ]
        }
        
        print("✅ ContentGenerator готов к работе!")

    def generate_post(self, business_type, business_description, platform):
        """
        Генерирует контент для соцсетей
        
        Args:
            business_type (str): Тип бизнеса
            business_description (str): Описание бизнеса
            platform (str): Платформа (instagram, linkedin, telegram, twitter, ad)
        
        Returns:
            str: Сгенерированный пост
        """
        print(f"🔄 Генерация поста для {platform}...")
        
        # Пробуем сгенерировать через Groq
        if self.use_groq:
            try:
                post = self._generate_with_groq(business_type, business_description, platform)
                if post and len(post) > 20:
                    print("✅ Пост сгенерирован через Groq!")
                    return post
            except Exception as e:
                print(f"⚠️ Ошибка Groq: {e}, использую шаблоны")
        
        # Если Groq не работает - используем шаблоны
        print("🔄 Использую шаблоны...")
        return self._generate_with_templates(business_type, business_description, platform)
    
    def _generate_with_groq(self, business_type, business_description, platform):
        """Генерация через Groq API"""
        
        platform_names = {
            'instagram': 'Instagram',
            'linkedin': 'LinkedIn',
            'telegram': 'Telegram',
            'twitter': 'Twitter/X',
            'ad': 'рекламный пост'
        }
        
        prompt = f"""
Ты — профессиональный SMM-специалист и копирайтер.

Напиши интересный, живой пост для {platform_names.get(platform, 'соцсетей')} о бизнесе.

Тип бизнеса: {business_type}
Описание бизнеса: {business_description}

Требования к посту:
- Длина: 100-200 слов (но не больше 200 слов)
- Живой, разговорный стиль
- Полезный контент для предпринимателей
- Призыв к действию в конце
- Используй 2-3 эмодзи

Напиши только текст поста, без лишних пояснений.
Пост должен быть уникальным и интересным.
"""
        
        response = self.client.chat.completions.create(
            model=self.model,
            messages=[
                {"role": "system", "content": "Ты профессиональный копирайтер. Пиши ярко, полезно и вовлекающе."},
                {"role": "user", "content": prompt}
            ],
            temperature=0.9,
            max_tokens=500
        )
        
        return response.choices[0].message.content.strip()
    
    def _generate_with_templates(self, business_type, business_description, platform):
        """Генерация через шаблоны (запасной вариант)"""
        
        # Если платформы нет в шаблонах - используем instagram
        if platform not in self.templates:
            platform = 'instagram'
        
        # Подготовка данных для шаблона
        data = {
            'business_name': business_type,
            'industry': business_description[:30] if business_description else "твоей нише",
            'years': random.randint(2, 15),
            'clients': random.randint(50, 5000),
            'profit': random.randint(20, 300),
            'growth': random.randint(2, 10),
            'problem': random.choice([
                "не хватало знаний", "было сложно найти клиентов", 
                "конкуренты давили", "не было системы"
            ]),
            'solution': random.choice([
                "инновационный подход", "автоматизация процессов",
                "клиентоориентированность", "стратегическое планирование"
            ]),
            'result': random.choice([
                "выручка выросла в 3 раза", "число клиентов удвоилось",
                "бренд узнали в 5 городах", "создали команду из 30 человек"
            ]),
            'problem_solution': random.choice([
                "решает главную проблему предпринимателей",
                "помогает бизнесу расти быстрее",
                "упрощает сложные процессы",
                "делает мир лучше"
            ]),
            'lesson1': "Не бойся пробовать новое",
            'lesson2': "Клиент всегда прав (почти всегда)",
            'lesson3': "Команда — это всё",
            'lesson4': "Без маркетинга никуда",
            'lesson5': "Верь в своё дело",
            'short_story': random.choice([
                "я просто начал и не остановился",
                "увидел проблему и решил её",
                "поверил в идею, несмотря на сомнения",
                "просто делал то, что люблю"
            ]),
            'belief': random.choice([
                "успех приходит к тем, кто не боится ошибаться",
                "лучший способ предсказать будущее — создать его",
                "качество всегда побеждает количество"
            ]),
            'tip': random.choice(["простой", "важный", "ключевой"]),
            'advice': random.choice([
                "не распыляйся на всё подряд — фокусируйся на главном",
                "слушай своих клиентов — они лучшие советчики",
                "инвестируй в своё развитие — это окупается",
                "не бойся делегировать задачи"
            ]),
            'start_point': random.choice([
                "идеи и ноутбука", "небольшого стартапа",
                "мечты и энтузиазма", "первых 5 клиентов"
            ]),
            'quote': random.choice([
                "Бизнес — это не про деньги, а про ценность, которую ты создаёшь",
                "Каждый отказ приближает тебя к успеху",
                "Лучший момент начать — прямо сейчас",
                "Сложно — значит ты на правильном пути"
            ]),
            'benefit': random.choice([
                "прибыль без головной боли",
                "стабильный доход",
                "финансовую свободу",
                "масштабирование бизнеса"
            ]),
            'hashtag': random.choice([
                "бизнесрост", "предприниматель", "успех", 
                "бизнесидеи", "мойбизнес", "развитие"
            ])
        }
        
        # Выбираем случайный шаблон для платформы
        template = random.choice(self.templates[platform])
        
        # Заполняем шаблон данными
        try:
            post = template.format(**data)
        except KeyError as e:
            # Если какие-то ключи отсутствуют
            post = f"📝 Пост для {platform.upper()}\n\n" \
                   f"Бизнес: {business_type}\n" \
                   f"Описание: {business_description or 'Не указано'}\n\n" \
                   f"Расскажи о своём бизнесе в комментариях! 👇\n\n#бизнес #предприниматель"
        
        return post