ai-builder-photo

AI Builder Photo

AI Builder Photo - модуль для генерации скриншотов чата в стиле мессенджера. Идеально подходит для создания PNG изображений диалогов для презентаций и демонстраций.

Установка

pip install Pillow

Быстрый старт

from ai_builder_photo import create_chat_screenshot

# Простой диалог
dialog = """-Привет
-Привет! Как дела?
-Хорошо, а у тебя?
-Тоже отлично!"""

# Создание скриншота
file_path = create_chat_screenshot(dialog, "chat.png")
print(f"Скриншот создан: {file_path}")

Расширенное использование

from ai_builder_photo import ChatScreenshotGenerator

# Кастомный генератор
generator = ChatScreenshotGenerator(
    width=450,
    background_color="#1C1C1E"  # Темная тема
)

# Изменение цветов
generator.user_color = "#0A84FF"
generator.bot_color = "#2C2C2E"
generator.user_text_color = "#FFFFFF"
generator.bot_text_color = "#FFFFFF"

# Создание скриншота
file_path = generator.generate_from_dialog_string(dialog, "dark_chat.png")

Программное создание сообщений

messages = [
    ("Привет! Как дела?", True),  # True = пользователь
    ("Привет! Все отлично!", False),  # False = бот
    ("Отлично!", True),
]

file_path = generator.generate_chat_screenshot(messages, "chat.png")

Формат диалога

Диалог передается в виде строки, где каждая строка начинается с -:

-Первое сообщение (пользователь)
-Второе сообщение (бот)
-Третье сообщение (пользователь)
-Четвертое сообщение (бот)

Четные позиции (0, 2, 4...) = сообщения пользователя (справа, синие) Нечетные позиции (1, 3, 5...) = сообщения бота (слева, серые)

API

create_chat_screenshot(dialog_string, save_path, width=400)

Быстрая функция для создания скриншота.

Параметры:

• dialog_string - строка диалога
• save_path - путь для сохранения PNG файла
• width - ширина изображения

Возвращает: Путь к созданному файлу

ChatScreenshotGenerator

Основной класс для генерации скриншотов.

Методы:

• generate_from_dialog_string(dialog_string, save_path) - из строки диалога
• generate_chat_screenshot(messages, save_path) - из списка сообщений

Примеры

Запустите python example.py для демонстрации всех возможностей модуля.

Особенности

• ✅ Автоматический перенос длинных сообщений
• ✅ Поддержка темной и светлой темы
• ✅ Простое сохранение в PNG файлы
• ✅ Кастомизация цветов и размеров
• ✅ Поддержка русского языка
• ✅ Стиль современных мессенджеров

Лицензия

MIT License