Add files via upload
Initial project upload
This commit is contained in:
commit
126e76963c
20 files changed
+1227
No files matched your search
@@ -0,0 +1,78 @@
|
||||
from database import get_user
|
||||
from ai_core.neural_network import ContentGenerator
|
||||
from keyboards.reply_keyboards import get_content_keyboard, BUSINESS_TYPES
|
||||
import time
|
||||
|
||||
# Создаем экземпляр генератора контента
|
||||
content_gen = ContentGenerator()
|
||||
|
||||
def handle_content_generation(message, bot, platform_text):
|
||||
"""Генерирует контент для выбранной платформы"""
|
||||
user_id = message.from_user.id
|
||||
user = get_user(user_id)
|
||||
|
||||
if not user:
|
||||
bot.send_message(user_id, "❌ Ты не зарегистрирован. Напиши /start")
|
||||
return
|
||||
|
||||
# Извлекаем платформу из текста кнопки
|
||||
platform_map = {
|
||||
"📝 Пост для Instagram": "instagram",
|
||||
"💼 Пост для LinkedIn": "linkedin",
|
||||
"📱 Пост для Telegram": "telegram",
|
||||
"🧵 Пост для Twitter/X": "twitter",
|
||||
"🎯 Рекламный пост": "ad"
|
||||
}
|
||||
|
||||
platform = platform_map.get(platform_text, "instagram")
|
||||
platform_emoji = {
|
||||
"instagram": "📸",
|
||||
"linkedin": "💼",
|
||||
"telegram": "📱",
|
||||
"twitter": "🐦",
|
||||
"ad": "📢"
|
||||
}
|
||||
|
||||
# Получаем данные пользователя
|
||||
business_type = BUSINESS_TYPES.get(str(user[4]), {}).get('name', 'Бизнес')
|
||||
description = user[5] or "не указано"
|
||||
|
||||
# Отправляем статус "печатает"
|
||||
bot.send_chat_action(user_id, "typing")
|
||||
|
||||
# Генерируем контент
|
||||
try:
|
||||
post = content_gen.generate_post(business_type, description, platform)
|
||||
|
||||
# Отправляем результат
|
||||
response_text = f"""
|
||||
{platform_emoji.get(platform, '📝')} **ГОТОВЫЙ ПОСТ ДЛЯ {platform.upper()}**
|
||||
|
||||
━━━━━━━━━━━━━━━━━━━━━━━
|
||||
|
||||
{post}
|
||||
|
||||
━━━━━━━━━━━━━━━━━━━━━━━
|
||||
|
||||
💡 **Совет:**
|
||||
• Добавь хештеги из поста
|
||||
• Прикрепи фото или видео
|
||||
• Укажи ссылку на свой сайт
|
||||
|
||||
📌 Хочешь другой вариант? Снова нажми "💡 Предложить контент"
|
||||
"""
|
||||
|
||||
bot.send_message(user_id, response_text, parse_mode='Markdown')
|
||||
|
||||
except Exception as e:
|
||||
bot.send_message(
|
||||
user_id,
|
||||
f"❌ Ошибка генерации: {e}\n\nПопробуй позже или обнови описание бизнеса."
|
||||
)
|
||||
|
||||
# Показываем меню выбора платформы для следующего поста
|
||||
bot.send_message(
|
||||
user_id,
|
||||
"📱 Для какой соцсети нужен следующий пост?",
|
||||
reply_markup=get_content_keyboard()
|
||||
)
|
||||
Reference in new issue
Block a user