smart-bot-factory 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

smart_bot_factory/utils/prompt_loader.py

@@ -0,0 +1,427 @@
+# Обновленный prompt_loader.py с поддержкой финальных инструкций
+
+import logging
+import aiofiles
+from pathlib import Path
+from typing import List, Dict
+
+logger = logging.getLogger(__name__)
+
+class PromptLoader:
+    """Класс для загрузки промптов из локального каталога"""
+    
+    def __init__(self, prompts_dir: str):
+            self.prompts_dir = Path(prompts_dir)
+            self.welcome_file = self.prompts_dir / 'welcome_message.txt'
+            self.help_file = self.prompts_dir / 'help_message.txt'
+            self.final_instructions_file = self.prompts_dir / 'final_instructions.txt'
+            
+            # Автоматически находим все .txt файлы промптов (кроме специальных)
+            all_txt_files = list(self.prompts_dir.glob('*.txt'))
+            special_files = {'welcome_message.txt', 'help_message.txt', 'final_instructions.txt'}
+            self.prompt_files = [f.name for f in all_txt_files if f.name not in special_files]
+            
+            logger.info(f"Инициализирован загрузчик промптов: {self.prompts_dir}")
+            logger.info(f"Найдено файлов промптов: {len(self.prompt_files)}")
+            logger.info(f"Файлы промптов: {self.prompt_files}")
+
+    async def load_system_prompt(self) -> str:
+        """
+        Загружает и объединяет все файлы промптов в один системный промпт
+        
+        Returns:
+            Объединенный системный промпт с инструкциями по JSON
+        """
+        try:
+            prompt_parts = []
+            
+            for filename in self.prompt_files:
+                logger.debug(f"Загружаем промпт из {filename}")
+                content = await self._load_file(filename)
+                
+                if content:
+                    # Добавляем заголовок секции
+                    section_name = self._get_section_name(filename)
+                    prompt_parts.append(f"\n### {section_name} ###\n")
+                    prompt_parts.append(content.strip())
+                    prompt_parts.append("\n")
+                else:
+                    logger.warning(f"Файл {filename} пуст")
+            
+            if not prompt_parts:
+                error_msg = "Не удалось загрузить ни одного промпт файла"
+                logger.error(error_msg)
+                raise ValueError(error_msg)
+            
+            # Добавляем инструкции по JSON метаданным
+            json_instructions = self._get_json_instructions()
+            prompt_parts.append("\n")
+            prompt_parts.append(json_instructions)
+            
+            # Объединяем все части
+            full_prompt = "".join(prompt_parts).strip()
+            
+            logger.info(f"Системный промпт загружен успешно ({len(full_prompt)} символов)")
+            return full_prompt
+            
+        except Exception as e:
+            logger.error(f"Ошибка при загрузке системного промпта: {e}")
+            raise
+
+    async def load_final_instructions(self) -> str:
+        """
+        Загружает финальные инструкции из final_instructions.txt
+        
+        Returns:
+            Финальные инструкции или пустая строка если файла нет
+        """
+        try:
+            logger.debug(f"Загружаем финальные инструкции из {self.final_instructions_file.name}")
+            
+            if not self.final_instructions_file.exists():
+                logger.debug(f"Файл {self.final_instructions_file.name} не найден - пропускаем")
+                return ""
+            
+            async with aiofiles.open(self.final_instructions_file, 'r', encoding='utf-8') as f:
+                content = await f.read()
+            
+            if not content.strip():
+                logger.debug(f"Файл {self.final_instructions_file.name} пуст - пропускаем")
+                return ""
+            
+            logger.info(f"Финальные инструкции загружены ({len(content)} символов)")
+            return content.strip()
+                
+        except Exception as e:
+            logger.error(f"Ошибка при загрузке финальных инструкций: {e}")
+            # Не прерываем работу - финальные инструкции опциональны
+            return ""
+    
+    def _get_json_instructions(self) -> str:
+        """Возвращает инструкции по JSON метаданным для ИИ"""
+        return """
+### КРИТИЧЕСКИ ВАЖНЫЕ ИНСТРУКЦИИ ПО JSON МЕТАДАННЫМ ###
+
+В КОНЦЕ КАЖДОГО своего ответа ОБЯЗАТЕЛЬНО добавляй служебную информацию в JSON формате:
+
+{
+  "этап": "introduction|consult|offer|contacts",
+  "качество": 1-10,
+  "события": [
+    {
+      "тип": "телефон|консультация|покупка|отказ",
+      "инфо": "детали события"
+    }
+  ],
+  "файлы": ["file1.pdf", "file2.mp4"],
+  "каталоги": ["каталог1", "каталог2"]
+}
+
+ОПИСАНИЕ ЭТАПОВ:
+- introduction: знакомство, сбор базовой информации о клиенте
+- consult: консультирование, ответы на вопросы о конференции
+- offer: предложение тарифов, обработка возражений
+- contacts: получение контактов, завершение сделки
+
+СИСТЕМА ОЦЕНКИ КАЧЕСТВА (1-10):
+1-3: низкий интерес, много возражений, скептически настроен
+4-6: средний интерес, есть вопросы, обдумывает
+7-8: высокий интерес, готов к покупке, активно интересуется  
+9-10: горячий лид, предоставил контакты или готов к действию
+
+СОБЫТИЯ - добавляй ТОЛЬКО когда происходит что-то из этого:
+- "телефон": пользователь предоставил номер телефона
+- "консультация": пользователь просит живую консультацию по телефону
+- "покупка": пользователь готов купить/записаться на конференцию
+- "отказ": пользователь явно отказывается участвовать
+
+ВАЖНО:
+- Добавляй файлы и катологи только в том случае, если они относятся к этому этапу. (Прописаны в самом этапе) Если не относятся - отсылай пустой массив []
+
+ПРИМЕРЫ ПРАВИЛЬНОГО ИСПОЛЬЗОВАНИЯ:
+
+Пример 1 - обычный диалог:
+"Расскажу подробнее о конференции GrowthMED. Она пройдет 24-25 октября..."
+
+{
+  "этап": "consult",
+  "качество": 6,
+  "события": [],
+  "файлы": [],
+  "каталоги": []
+}
+
+Пример 2 - получен телефон:
+"Отлично! Записал ваш номер. Мы перезвоним в течение 10 минут!"
+
+{
+  "этап": "contacts", 
+  "качество": 9,
+  "события": [
+    {
+      "тип": "телефон",
+      "инфо": "Иван Петров +79219603144"
+    }
+  ],
+  "файлы": [],
+  "каталоги": []
+}
+
+Пример 3 - отправка презентации:
+"Отправляю вам презентацию о нашей компании и прайс-лист с актуальными ценами."
+
+{
+  "этап": "offer",
+  "качество": 7,
+  "события": [
+    {
+      "тип": "консультация", 
+      "инфо": "Запросил материалы"
+    }
+  ],
+  "файлы": ["презентация.pdf", "прайс.pdf"],
+  "каталоги": []
+}
+
+Пример 4 - отправка файлов из каталога:
+"В каталоге 'примеры_работ' вы можете посмотреть наши последние проекты."
+
+{
+  "этап": "presentation",
+  "качество": 8,
+  "события": [],
+  "файлы": [],
+  "каталоги": ["примеры_работ"]
+}
+
+Пример 5 - комбинированная отправка:
+"Отправляю вам коммерческое предложение и примеры похожих проектов из нашего портфолио."
+
+{
+  "этап": "offer",
+  "качество": 9,
+  "события": [
+    {
+      "тип": "предложение",
+      "инфо": "Отправлено КП"
+    }
+  ],
+  "файлы": ["коммерческое_предложение.pdf"],
+  "каталоги": ["портфолио_2023"]
+}
+
+ТРЕБОВАНИЯ К JSON:
+- JSON должен быть валидным и находиться в самом конце ответа
+- Всегда используй кавычки для строк
+- Массив "события" может быть пустым []
+- Если событий нет - не добавляй их в массив
+- Качество должно быть числом от 1 до 10
+
+ПОМНИ: Этот JSON критически важен для работы системы администрирования и аналитики!
+"""
+    
+    async def load_welcome_message(self) -> str:
+        """
+        Загружает приветственное сообщение из welcome_message.txt
+        
+        Returns:
+            Текст приветственного сообщения
+        """
+        try:
+            logger.debug(f"Загружаем приветственное сообщение из {self.welcome_file.name}")
+            
+            if not self.welcome_file.exists():
+                error_msg = f"Файл приветствия не найден: {self.welcome_file}"
+                logger.error(error_msg)
+                raise FileNotFoundError(error_msg)
+            
+            async with aiofiles.open(self.welcome_file, 'r', encoding='utf-8') as f:
+                content = await f.read()
+            
+            if not content.strip():
+                error_msg = f"Файл приветствия пуст: {self.welcome_file}"
+                logger.error(error_msg)
+                raise ValueError(error_msg)
+            
+            logger.info(f"Приветственное сообщение загружено ({len(content)} символов)")
+            return content.strip()
+                
+        except Exception as e:
+            logger.error(f"Ошибка при загрузке приветственного сообщения: {e}")
+            raise
+            
+    async def load_help_message(self) -> str:
+        """
+        Загружает справочное сообщение из help_message.txt
+        
+        Returns:
+            Текст справочного сообщения
+        """
+        try:
+            logger.debug(f"Загружаем справочное сообщение из {self.help_file.name}")
+            
+            if self.help_file.exists():
+                async with aiofiles.open(self.help_file, 'r', encoding='utf-8') as f:
+                    content = await f.read()
+                
+                if content.strip():
+                    logger.info(f"Справочное сообщение загружено ({len(content)} символов)")
+                    return content.strip()
+            
+            # Fallback если файл не найден или пуст
+            logger.warning("Файл help_message.txt не найден или пуст, используем дефолтную справку")
+            return "🤖 **Ваш помощник готов к работе!**\n\n**Команды:**\n/start - Начать диалог\n/help - Показать справку\n/status - Проверить статус"
+                
+        except Exception as e:
+            logger.error(f"Ошибка при загрузке справочного сообщения: {e}")
+            # Возвращаем простую справку в случае ошибки
+            return "🤖 Ваш помощник готов к работе! Напишите /start для начала диалога."
+
+    async def _load_file(self, filename: str) -> str:
+        """Загружает содержимое файла из каталога промптов"""
+        file_path = self.prompts_dir / filename
+        
+        try:
+            if not file_path.exists():
+                error_msg = f"Файл промпта не найден: {file_path}"
+                logger.error(error_msg)
+                raise FileNotFoundError(error_msg)
+            
+            async with aiofiles.open(file_path, 'r', encoding='utf-8') as f:
+                content = await f.read()
+            
+            if not content.strip():
+                logger.warning(f"Файл {filename} пуст")
+                return ""
+            
+            logger.debug(f"Загружен файл {filename} ({len(content)} символов)")
+            return content
+            
+        except Exception as e:
+            logger.error(f"Ошибка чтения файла {file_path}: {e}")
+            raise
+    
+    def _get_section_name(self, filename: str) -> str:
+        """Получает название секции по имени файла"""
+        name_mapping = {
+            'system_prompt.txt': 'СИСТЕМНЫЙ ПРОМПТ',
+            'sales_context.txt': 'КОНТЕКСТ ПРОДАЖ', 
+            'product_info.txt': 'ИНФОРМАЦИЯ О ПРОДУКТЕ',
+            'objection_handling.txt': 'ОБРАБОТКА ВОЗРАЖЕНИЙ',
+            '1sales_context.txt': 'КОНТЕКСТ ПРОДАЖ',
+            '2product_info.txt': 'ИНФОРМАЦИЯ О ПРОДУКТЕ', 
+            '3objection_handling.txt': 'ОБРАБОТКА ВОЗРАЖЕНИЙ',
+            'final_instructions.txt': 'ФИНАЛЬНЫЕ ИНСТРУКЦИИ'  # 🆕
+        }
+        
+        return name_mapping.get(filename, filename.replace('.txt', '').upper())
+    
+    async def reload_prompts(self) -> str:
+        """Перезагружает промпты (для обновления без перезапуска бота)"""
+        logger.info("Перезагрузка промптов...")
+        return await self.load_system_prompt()
+    
+    async def validate_prompts(self) -> Dict[str, bool]:
+        """Проверяет доступность всех файлов промптов и приветственного сообщения"""
+        results = {}
+        
+        # Проверяем файлы промптов
+        for filename in self.prompt_files:
+            file_path = self.prompts_dir / filename
+            try:
+                if file_path.exists():
+                    async with aiofiles.open(file_path, 'r', encoding='utf-8') as f:
+                        content = await f.read()
+                    results[filename] = bool(content.strip() and len(content.strip()) > 10)
+                else:
+                    results[filename] = False
+            except Exception:
+                results[filename] = False
+        
+        # Проверяем файл приветственного сообщения
+        try:
+            if self.welcome_file.exists():
+                async with aiofiles.open(self.welcome_file, 'r', encoding='utf-8') as f:
+                    content = await f.read()
+                results['welcome_message.txt'] = bool(content.strip() and len(content.strip()) > 5)
+            else:
+                results['welcome_message.txt'] = False
+        except Exception:
+            results['welcome_message.txt'] = False
+        
+        # Проверяем файл справки (опционально)
+        try:
+            if self.help_file.exists():
+                async with aiofiles.open(self.help_file, 'r', encoding='utf-8') as f:
+                    content = await f.read()
+                results['help_message.txt'] = bool(content.strip() and len(content.strip()) > 5)
+            else:
+                results['help_message.txt'] = False  # Не критично
+        except Exception:
+            results['help_message.txt'] = False
+        
+        # 🆕 Проверяем финальные инструкции (опционально)
+        try:
+            if self.final_instructions_file.exists():
+                async with aiofiles.open(self.final_instructions_file, 'r', encoding='utf-8') as f:
+                    content = await f.read()
+                results['final_instructions.txt'] = bool(content.strip() and len(content.strip()) > 5)
+            else:
+                results['final_instructions.txt'] = False  # Не критично - опциональный файл
+        except Exception:
+            results['final_instructions.txt'] = False
+        
+        return results
+    
+    def get_prompt_info(self) -> Dict[str, any]:
+        """Возвращает информацию о конфигурации промптов"""
+        return {
+            'prompts_dir': str(self.prompts_dir),
+            'prompt_files': self.prompt_files,
+            'welcome_file': 'welcome_message.txt',
+            'help_file': 'help_message.txt',
+            'final_instructions_file': 'final_instructions.txt',  # 🆕
+            'total_files': len(self.prompt_files) + 1,  # +1 для welcome message
+            'json_instructions_included': True
+        }
+    
+    async def test_json_parsing(self, test_response: str) -> Dict[str, any]:
+        """Тестирует парсинг JSON из ответа ИИ (для отладки)"""
+        import json
+        import re
+        
+        try:
+            # Используем тот же алгоритм что и в main.py
+            json_pattern = r'\{[^{}]*"этап"[^{}]*\}$'
+            match = re.search(json_pattern, test_response.strip())
+            
+            if match:
+                json_str = match.group(0)
+                response_text = test_response[:match.start()].strip()
+                
+                try:
+                    metadata = json.loads(json_str)
+                    return {
+                        'success': True,
+                        'response_text': response_text,
+                        'metadata': metadata,
+                        'json_str': json_str
+                    }
+                except json.JSONDecodeError as e:
+                    return {
+                        'success': False,
+                        'error': f"JSON decode error: {e}",
+                        'json_str': json_str
+                    }
+            else:
+                return {
+                    'success': False,
+                    'error': "JSON pattern not found",
+                    'response_text': test_response
+                }
+                
+        except Exception as e:
+            return {
+                'success': False,
+                'error': f"Parse error: {e}"
+            }

smart_bot_factory/utm_link_generator.py

@@ -0,0 +1,106 @@
+#!/usr/bin/env python3
+"""
+Генератор UTM-ссылок для Telegram ботов
+Создает ссылки в формате: @https://t.me/bot?start=source-vk_campaign-summer2025
+"""
+
+def get_user_input():
+    """Получает данные от пользователя через консоль"""
+    print("🔗 Генератор UTM-ссылок для Telegram")
+    print("=" * 50)
+    
+    # Основные параметры
+    bot_username = input("Введите username бота (без @): ").strip()
+    if not bot_username:
+        print("❌ Username бота обязателен!")
+        return None
+    
+    print("\n📊 Введите UTM-метки (нажмите Enter для пропуска):")
+    
+    # UTM параметры (соответствуют полям в базе данных)
+    utm_source = input("utm_source (источник): ").strip()
+    utm_medium = input("utm_medium (канал): ").strip()
+    utm_campaign = input("utm_campaign (кампания): ").strip()
+    utm_content = input("utm_content (контент): ").strip()
+    utm_term = input("utm_term (ключевое слово): ").strip()
+    
+    return {
+        'bot_username': bot_username,
+        'utm_source': utm_source,
+        'utm_medium': utm_medium,
+        'utm_campaign': utm_campaign,
+        'utm_content': utm_content,
+        'utm_term': utm_term
+    }
+
+def create_utm_string(utm_data):
+    """Создает строку UTM параметров в формате source-vk_campaign-summer2025"""
+    utm_parts = []
+    
+    # Маппинг полей базы данных на новый формат (без utm, в нижнем регистре)
+    field_mapping = {
+        'utm_source': 'source',
+        'utm_medium': 'medium', 
+        'utm_campaign': 'campaign',
+        'utm_content': 'content',
+        'utm_term': 'term'
+    }
+    
+    for db_field, utm_field in field_mapping.items():
+        value = utm_data.get(db_field)
+        if value:
+            utm_parts.append(f"{utm_field}-{value}")
+    
+    return "_".join(utm_parts)
+
+def generate_telegram_link(bot_username, utm_string):
+    """Генерирует полную ссылку на Telegram бота"""
+    return f"https://t.me/{bot_username}?start={utm_string}"
+
+def check_size_and_validate(utm_string):
+    """Проверяет размер строки после start= и валидирует"""
+    MAX_SIZE = 64
+    
+    if len(utm_string) > MAX_SIZE:
+        return False, f"Строка слишком большая: {len(utm_string)} символов (максимум {MAX_SIZE})"
+    
+    return True, f"Размер OK: {len(utm_string)} символов"
+
+
+def main():
+    """Основная функция"""
+    try:
+        # Получаем данные от пользователя
+        data = get_user_input()
+        if not data:
+            return
+        
+        # Создаем UTM строку
+        utm_string = create_utm_string(data)
+        
+        if not utm_string:
+            print("❌ Не указано ни одной UTM-метки!")
+            return
+        
+        # Проверяем размер
+        is_valid, size_message = check_size_and_validate(utm_string)
+        
+        print(f"\n📏 {size_message}")
+        
+        if not is_valid:
+            print("❌ Ссылка превышает максимальный размер!")
+            print("💡 Сократите значения UTM-меток или уберите менее важные")
+            return
+        
+        # Генерируем и выводим ссылку
+        telegram_link = generate_telegram_link(data['bot_username'], utm_string)
+        
+        print(f"\n✅ Сгенерированная ссылка:")
+        print(f"🔗 {telegram_link}")
+    except KeyboardInterrupt:
+        print("\n\n👋 Отменено пользователем")
+    except Exception as e:
+        print(f"\n❌ Ошибка: {e}")
+
+if __name__ == "__main__":
+    main()

smart_bot_factory-0.1.4.dist-info/METADATA

@@ -0,0 +1,126 @@
+Metadata-Version: 2.4
+Name: smart-bot-factory
+Version: 0.1.4
+Summary: Библиотека для создания умных чат-ботов
+Author-email: Kopatych <kopatych@example.com>
+License: MIT
+License-File: LICENSE
+Keywords: chatbot,cli,openai,supabase,telegram
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Framework :: AsyncIO
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Communications :: Chat
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.9
+Requires-Dist: aiofiles>=23.0.0
+Requires-Dist: aiogram>=3.4.1
+Requires-Dist: click>=8.0.0
+Requires-Dist: openai>=1.12.0
+Requires-Dist: project-root-finder>=1.9
+Requires-Dist: python-dotenv>=1.0.1
+Requires-Dist: pytz>=2023.3
+Requires-Dist: pyyaml>=6.0.2
+Requires-Dist: supabase>=2.3.4
+Description-Content-Type: text/markdown
+
+# Smart Bot Factory
+
+Библиотека для создания умных чат-ботов с использованием OpenAI, Telegram и Supabase.
+
+## Установка
+
+```bash
+pip install smart-bot-factory
+```
+
+## Быстрый старт
+
+1. Создайте нового бота:
+```bash
+sbf create my-bot
+```
+
+2. Настройте конфигурацию в `bots/my-bot/.env`
+
+3. Запустите бота:
+```bash
+sbf run my-bot
+```
+
+## Возможности
+
+- 🤖 Интеграция с OpenAI GPT для умных ответов
+- 📱 Поддержка Telegram Bot API через aiogram
+- 💾 Хранение данных в Supabase
+- 🔄 Система событий и обработчиков
+- ⏰ Планировщик задач
+- 🧪 Встроенная система тестирования
+- 📝 Управление промптами
+- 🛠️ Удобный CLI интерфейс
+
+## CLI команды
+
+```bash
+# Создать нового бота
+sbf create my-bot
+
+# Запустить бота
+sbf run my-bot
+
+# Показать список ботов
+sbf list
+
+# Управление промптами
+sbf prompts my-bot --list
+sbf prompts my-bot --edit welcome_message
+sbf prompts my-bot --add new_prompt
+
+# Запустить тесты
+sbf test my-bot
+```
+
+## Пример использования
+
+```python
+from smart_bot_factory import BotBuilder, event_handler, schedule_task
+
+# Обработчик события
+@event_handler("book_appointment", "Запись на прием")
+async def handle_booking(user_id: int, event_data: dict):
+    # Логика обработки записи на прием
+    return {"status": "success"}
+
+# Запланированная задача
+@schedule_task("send_reminder", "Отправка напоминания")
+async def send_reminder(user_id: int, message: str):
+    # Логика отправки напоминания
+    return {"status": "sent"}
+
+# Запуск бота
+async def main():
+    bot = BotBuilder("my-bot")
+    await bot.build()
+    await bot.start()
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+## Требования
+
+- Python 3.9+
+- OpenAI API ключ
+- Telegram Bot Token
+- Supabase проект
+
+## Лицензия
+
+MIT