smart-bot-factory 0.2.0__tar.gz → 0.2.2__tar.gz

smart_bot_factory-0.2.2/.claude/settings.local.json

@@ -0,0 +1,10 @@
+{
+  "permissions": {
+    "allow": [
+      "Bash(awk:*)",
+      "Bash(python:*)"
+    ],
+    "deny": [],
+    "ask": []
+  }
+}

smart_bot_factory-0.2.2/.env.example

@@ -0,0 +1,40 @@
+# Telegram
+TELEGRAM_BOT_TOKEN=your_telegram_bot_token_here
+
+# Supabase
+SUPABASE_URL=https://your-project.supabase.co
+SUPABASE_KEY=your_supabase_anon_key
+
+# OpenAI
+OPENAI_API_KEY=sk-your-openai-api-key
+OPENAI_MODEL=gpt-5-mini
+OPENAI_MAX_TOKENS=1500
+OPENAI_TEMPERATURE=0.7
+
+# Промпты (каталог)
+PROMT_FILES_DIR=prompts
+
+# Файл после приветствия с подписью (если он есть - грузим его в папку welcome_file, если нет - ничего не делаем)
+WELCOME_FILE_URL=welcome_file/
+WELCOME_FILE_MSG=welcome_file_msg.txt
+
+# 🆕 Администраторы (через запятую)
+# Укажите Telegram ID админов
+ADMIN_TELEGRAM_IDS=123456789,987654321
+ADMIN_SESSION_TIMEOUT_MINUTES=30
+
+# 🆕 Режим отладки (показывать JSON пользователям)
+DEBUG_MODE=false
+
+# Дополнительные настройки
+MAX_CONTEXT_MESSAGES=50
+LOG_LEVEL=INFO
+MESSAGE_PARSE_MODE=Markdown
+
+# Настройки продаж
+LEAD_QUALIFICATION_THRESHOLD=7
+SESSION_TIMEOUT_HOURS=24
+
+# ⚠️ ВАЖНО: BOT_ID теперь НЕ нужен в .env!
+# Bot ID автоматически определяется из имени файла запускалки
+# Например: python growthmed-october-24.py → BOT_ID = growthmed-october-24

{smart_bot_factory-0.2.0 → smart_bot_factory-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: smart-bot-factory
-Version: 0.2.0
+Version: 0.2.2
 Summary: Библиотека для создания умных чат-ботов
 Author-email: Kopatych <kopatych@example.com>
 License: MIT

smart_bot_factory-0.2.2/best-valera.py

@@ -0,0 +1,240 @@
+import asyncio
+import logging
+
+from smart_bot_factory.router import EventRouter, TelegramRouter
+from smart_bot_factory.message import send_message_by_human, send_message_to_users_by_stage
+from smart_bot_factory.supabase import SupabaseClient
+from smart_bot_factory.creation import BotBuilder
+
+logger = logging.getLogger(__name__)
+
+# =============================================================================
+# СОЗДАНИЕ РОУТЕРОВ
+# =============================================================================
+
+# Роутер для событий (бизнес-логика)
+event_router = EventRouter("best-valera_events")
+
+# Роутер для Telegram (команды и сообщения)
+telegram_router_1 = TelegramRouter("best-valera_telegram")
+telegram_router_2 = TelegramRouter("best-valera_telegram_2")
+
+supabase_client = SupabaseClient("best-valera")
+
+# =============================================================================
+# ИНИЦИАЛИЗАЦИЯ BOT BUILDER (нужен для декоратора on_start)
+# =============================================================================
+
+bot_builder = BotBuilder("best-valera")
+
+# =============================================================================
+# TELEGRAM ОБРАБОТЧИКИ (используем прямой aiogram API)
+# =============================================================================
+
+from aiogram import F
+from aiogram.filters import Command
+from aiogram.types import Message, CallbackQuery, InlineKeyboardMarkup, InlineKeyboardButton
+from aiogram.fsm.context import FSMContext
+
+# =============================================================================
+# ОБРАБОТЧИК on_start (вызывается после стандартного /start)
+# =============================================================================
+
+@bot_builder.on_start
+async def custom_start_handler(user_id: int, session_id: str, message: Message, state: FSMContext):
+    """
+    Вызывается после стандартной логики /start
+    Здесь можно отправить дополнительные сообщения, кнопки и т.д.
+    """
+    # Пример: отправляем сообщение с кнопками
+    keyboard = InlineKeyboardMarkup(inline_keyboard=[
+        [InlineKeyboardButton(text="📖 Каталог", callback_data="catalog")],
+        [InlineKeyboardButton(text="💰 Цены", callback_data="prices")],
+        [InlineKeyboardButton(text="📞 Связаться", callback_data="contact")]
+    ])
+    
+    await message.answer(
+        "🎯 Что вас интересует?",
+        reply_markup=keyboard
+    )
+
+# =============================================================================
+# ХУКИ ДЛЯ КАСТОМИЗАЦИИ ОБРАБОТКИ СООБЩЕНИЙ
+# =============================================================================
+
+# ХУК 1: Валидация сообщений
+@bot_builder.validate_message
+async def validate_service_names(message, supabase_client):
+    """Проверяем корректность названий услуг"""
+    # Пример: блокируем неправильные названия
+    incorrect_names = ["массаш", "педекюр", "макияш"]
+    
+    if message.text:
+        for incorrect in incorrect_names:
+            if incorrect in message.text.lower():
+                await message.answer(
+                    f"❗ Возможно, вы имели в виду другую услугу?\n"
+                    f"Пожалуйста, уточните название"
+                )
+                return False  # Прерываем обработку AI
+    
+    return True  # Продолжаем обработку
+
+# ХУК 2: Обогащение промпта
+@bot_builder.enrich_prompt
+async def add_client_info_to_prompt(system_prompt, user_id, session_id, supabase_client):
+    """Добавляем информацию о клиенте в промпт"""
+    try:
+        session = await supabase_client.get_active_session(user_id)
+        
+        if session and session.get('metadata'):
+            phone = session['metadata'].get('phone')
+            name = session['metadata'].get('confirmed_name') or session['metadata'].get('telegram_name')
+            
+            if phone:
+                client_info = f"\n\nИНФОРМАЦИЯ О КЛИЕНТЕ:\n- Телефон: {phone}"
+                if name:
+                    client_info += f"\n- Имя: {name}"
+                client_info += "\n\n⚠️ ВАЖНО: При создании записи используй ЭТОТ телефон и имя!"
+                
+                return system_prompt + client_info
+    except Exception as e:
+        logger.error(f"Ошибка обогащения промпта: {e}")
+    
+    return system_prompt
+
+# ХУК 3: Обогащение контекста (например, данные из внешнего API)
+@bot_builder.enrich_context
+async def add_external_api_data(messages, user_id, session_id):
+    """Добавляем данные из внешних систем"""
+    # Пример: можно добавить расписание из YClients, данные из CRM и т.д.
+    # messages.append({
+    #     "role": "system",
+    #     "content": "Реальное доступное время: ..."
+    # })
+    return messages
+
+# ХУК 4: Обработка ответа AI
+@bot_builder.process_response
+async def add_promo_to_price_response(response_text, ai_metadata, user_id):
+    """Добавляем промо-код к ответам о ценах"""
+    if "цен" in response_text.lower() or "стоимост" in response_text.lower():
+        response_text += "\n\n🎁 Промокод FIRST10 для скидки 10% на первый визит!"
+    
+    return response_text, ai_metadata
+
+# ХУК 5: Фильтр отправки
+@bot_builder.filter_send
+async def allow_all_messages(user_id, response_text):
+    """Разрешаем все сообщения (можно добавить блокировку по условию)"""
+    # Пример: блокировка во время обработки
+    # if is_processing(user_id):
+    #     return False
+    return True
+
+@telegram_router_1.router.message(Command("price", "цена"))
+async def handle_price_command(message: Message, state: FSMContext):
+    """Обработчик команды /price"""
+    await message.answer(
+        "💰 Наши цены:\n\n"
+        "📦 Базовый пакет - 1000₽\n"
+        "📦 Стандартный - 2000₽\n"
+        "📦 Премиум - 5000₽"
+    )
+
+@telegram_router_2.router.message(F.text & (F.text.lower().contains("цена") | F.text.lower().contains("стоимость")))
+async def handle_price_question(message: Message, state: FSMContext):
+    """Обработчик вопросов о цене"""
+    await message.answer("💡 Напишите /price чтобы увидеть актуальные цены")
+
+# Обработчики callback'ов от кнопок в on_start
+@telegram_router_1.router.callback_query(F.data == "catalog")
+async def handle_catalog(callback: CallbackQuery, state: FSMContext):
+    """Обработка кнопки Каталог"""
+    await callback.answer()
+    await callback.message.answer("📖 Вот наш каталог товаров:\n\n1. Товар 1\n2. Товар 2\n3. Товар 3")
+
+@telegram_router_1.router.callback_query(F.data == "prices")
+async def handle_prices(callback: CallbackQuery, state: FSMContext):
+    """Обработка кнопки Цены"""
+    await callback.answer()
+    await callback.message.answer("💰 Наши цены:\n\n📦 Базовый - 1000₽\n📦 Премиум - 5000₽")
+
+@telegram_router_1.router.callback_query(F.data == "contact")
+async def handle_contact(callback: CallbackQuery, state: FSMContext):
+    """Обработка кнопки Связаться"""
+    await callback.answer()
+    await callback.message.answer("📞 Свяжитесь с нами:\n\nТелефон: +7 (999) 123-45-67\nEmail: info@example.com")
+
+# =============================================================================
+# ОБРАБОТЧИКИ СОБЫТИЙ (бизнес-логика)
+# =============================================================================
+
+@event_router.event_handler("example_event")
+async def handle_example_event(user_id: int, event_data: str):
+    """Пример обработчика события"""
+    await send_message_by_human(
+        user_id=user_id,
+        message_text=f"✅ Событие обработано! Данные: {event_data}"
+    )
+    
+    return {
+        "status": "success",
+        "message": "Событие обработано"
+    }
+
+# =============================================================================
+# ВРЕМЕННЫЕ ЗАДАЧИ ДЛЯ ОДНОГО ПОЛЬЗОВАТЕЛЯ
+# =============================================================================
+
+@event_router.schedule_task("send_reminder", delay="1h")
+async def send_user_reminder(user_id: int, reminder_text: str):
+    """Отправляет напоминание пользователю"""
+    await send_message_by_human(
+        user_id=user_id,
+        message_text=f"🔔 Напоминание: {reminder_text}"
+    )
+    
+    return {
+        "status": "reminder_sent",
+        "message": f"Напоминание отправлено пользователю {user_id}"
+    }
+
+# =============================================================================
+# ГЛОБАЛЬНЫЕ ОБРАБОТЧИКИ (для всех пользователей)
+# =============================================================================
+
+@event_router.global_handler("mass_notification", delay="1h", notify=True)
+async def send_global_announcement(announcement_text: str):
+    """Отправляет анонс всем пользователям бота"""
+   
+    await send_message_to_users_by_stage(
+        stage="introduction",
+        message_text=announcement_text,
+        bot_id="best-valera"
+    )
+
+# =============================================================================
+# ОСНОВНАЯ ФУНКЦИЯ
+# =============================================================================
+
+async def main():
+    """Основная функция запуска бота"""
+    try:
+        # bot_builder уже создан выше (для декоратора @bot_builder.on_start)
+        
+        # Регистрируем роутеры ПЕРЕД сборкой (можно по одному или несколько сразу)
+        bot_builder.register_routers(event_router)  # Роутеры событий
+        bot_builder.register_telegram_routers(telegram_router_1, telegram_router_2)  # Telegram роутеры
+        
+        await bot_builder.build()
+        
+        # Запускаем бота
+        await bot_builder.start()
+        
+    except Exception as e:
+        print(f"❌ Ошибка запуска бота: {e}")
+        raise
+
+if __name__ == "__main__":
+    asyncio.run(main())

{smart_bot_factory-0.2.0 → smart_bot_factory-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "smart-bot-factory"
-version = "0.2.0"
+version = "0.2.2"
 description = "Библиотека для создания умных чат-ботов"
 authors = [
     {name = "Kopatych", email = "kopatych@example.com"}

{smart_bot_factory-0.2.0 → smart_bot_factory-0.2.2}/smart_bot_factory/creation/bot_builder.py

@@ -46,6 +46,14 @@ class BotBuilder:
         self.prompt_loader: Optional[PromptLoader] = None
         self.router_manager: Optional[RouterManager] = None
         self._telegram_routers: List = []  # Список Telegram роутеров
+        self._start_handlers: List = []  # Список обработчиков on_start
+        
+        # Хуки для кастомизации process_user_message
+        self._message_validators: List = []  # Валидация ДО обработки
+        self._prompt_enrichers: List = []    # Обогащение системного промпта
+        self._context_enrichers: List = []   # Обогащение контекста для AI
+        self._response_processors: List = [] # Обработка ответа AI
+        self._send_filters: List = []        # Фильтры перед отправкой пользователю
         
         # Флаги инициализации
         self._initialized = False
@@ -326,6 +334,165 @@ class BotBuilder:
         
         logger.info(f"✅ Зарегистрировано {len(telegram_routers)} Telegram роутеров")
     
+    def on_start(self, handler):
+        """
+        Регистрирует обработчик, который вызывается после стандартной логики /start
+        
+        Обработчик получает доступ к:
+        - user_id: int - ID пользователя Telegram
+        - session_id: str - ID созданной сессии
+        - message: Message - Объект сообщения от aiogram
+        - state: FSMContext - Контекст состояния
+        
+        Args:
+            handler: Async функция с сигнатурой:
+                     async def handler(user_id: int, session_id: str, message: Message, state: FSMContext)
+        
+        Example:
+            @bot_builder.on_start
+            async def my_start_handler(user_id, session_id, message, state):
+                keyboard = InlineKeyboardMarkup(...)
+                await message.answer("Выберите действие:", reply_markup=keyboard)
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._start_handlers.append(handler)
+        logger.info(f"✅ Зарегистрирован обработчик on_start: {handler.__name__}")
+        return handler  # Возвращаем handler для использования как декоратор
+    
+    def get_start_handlers(self) -> List:
+        """Получает список обработчиков on_start"""
+        return self._start_handlers.copy()
+    
+    # ========== ХУКИ ДЛЯ КАСТОМИЗАЦИИ ОБРАБОТКИ СООБЩЕНИЙ ==========
+    
+    def validate_message(self, handler):
+        """
+        Регистрирует валидатор сообщений (вызывается ДО обработки AI)
+        
+        Если валидатор возвращает False, обработка прерывается
+        
+        Args:
+            handler: async def(message: Message, supabase_client) -> bool
+        
+        Example:
+            @bot_builder.validate_message
+            async def check_service_names(message, supabase_client):
+                if "неправильное название" in message.text:
+                    await message.answer("Пожалуйста, уточните название услуги")
+                    return False  # Прерываем обработку
+                return True  # Продолжаем
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._message_validators.append(handler)
+        logger.info(f"✅ Зарегистрирован валидатор сообщений: {handler.__name__}")
+        return handler
+    
+    def enrich_prompt(self, handler):
+        """
+        Регистрирует обогатитель системного промпта
+        
+        Args:
+            handler: async def(system_prompt: str, user_id: int, session_id: str, supabase_client) -> str
+        
+        Example:
+            @bot_builder.enrich_prompt
+            async def add_client_info(system_prompt, user_id, session_id, supabase_client):
+                session = await supabase_client.get_active_session(user_id)
+                phone = session.get('metadata', {}).get('phone')
+                if phone:
+                    return f"{system_prompt}\\n\\nТелефон клиента: {phone}"
+                return system_prompt
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._prompt_enrichers.append(handler)
+        logger.info(f"✅ Зарегистрирован обогатитель промпта: {handler.__name__}")
+        return handler
+    
+    def enrich_context(self, handler):
+        """
+        Регистрирует обогатитель контекста для AI (messages array)
+        
+        Args:
+            handler: async def(messages: List[dict], user_id: int, session_id: str) -> List[dict]
+        
+        Example:
+            @bot_builder.enrich_context
+            async def add_external_data(messages, user_id, session_id):
+                # Добавляем данные из внешнего API
+                messages.append({
+                    "role": "system", 
+                    "content": "Дополнительная информация..."
+                })
+                return messages
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._context_enrichers.append(handler)
+        logger.info(f"✅ Зарегистрирован обогатитель контекста: {handler.__name__}")
+        return handler
+    
+    def process_response(self, handler):
+        """
+        Регистрирует обработчик ответа AI (ПОСЛЕ получения ответа)
+        
+        Args:
+            handler: async def(response_text: str, ai_metadata: dict, user_id: int) -> tuple[str, dict]
+        
+        Example:
+            @bot_builder.process_response
+            async def modify_response(response_text, ai_metadata, user_id):
+                # Модифицируем ответ
+                if "цена" in response_text.lower():
+                    response_text += "\\n\\n💰 Актуальные цены на сайте"
+                return response_text, ai_metadata
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._response_processors.append(handler)
+        logger.info(f"✅ Зарегистрирован обработчик ответа: {handler.__name__}")
+        return handler
+    
+    def filter_send(self, handler):
+        """
+        Регистрирует фильтр отправки (может блокировать отправку пользователю)
+        
+        Если фильтр возвращает False, сообщение НЕ отправляется
+        
+        Args:
+            handler: async def(user_id: int, response_text: str) -> bool
+        
+        Example:
+            @bot_builder.filter_send
+            async def block_during_process(user_id, response_text):
+                if is_processing(user_id):
+                    return False  # Не отправляем
+                return True  # Отправляем
+        """
+        if not callable(handler):
+            raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
+        
+        self._send_filters.append(handler)
+        logger.info(f"✅ Зарегистрирован фильтр отправки: {handler.__name__}")
+        return handler
+    
+    def get_message_hooks(self) -> Dict[str, List]:
+        """Получает все хуки для обработки сообщений"""
+        return {
+            'validators': self._message_validators.copy(),
+            'prompt_enrichers': self._prompt_enrichers.copy(),
+            'context_enrichers': self._context_enrichers.copy(),
+            'response_processors': self._response_processors.copy(),
+            'send_filters': self._send_filters.copy()
+        }
+    
     def get_router_manager(self) -> RouterManager:
         """Получает менеджер роутеров событий"""
         return self.router_manager
@@ -374,6 +541,8 @@ class BotBuilder:
                 handlers_module.admin_manager = self.admin_manager
                 handlers_module.analytics_manager = self.analytics_manager
                 handlers_module.conversation_manager = self.conversation_manager
+                handlers_module.start_handlers = self._start_handlers  # Передаем обработчики on_start
+                handlers_module.message_hooks = self.get_message_hooks()  # Передаем хуки для обработки сообщений
                 logger.info("✅ Глобальные переменные установлены в handlers")
             except Exception as e:
                 logger.warning(f"⚠️ Не удалось установить глобальные переменные в handlers: {e}")