Платформа для генерации изображений с автоматической постобработкой
API-платформа для генерации изображений с поддержкой нескольких AI-моделей, автоматической постобработкой и системой уведомлений.
О проекте
API-платформа для генерации изображений с поддержкой нескольких AI-моделей, автоматической постобработкой и системой уведомлений.
Содержание
Разработана серверная платформа для автоматизации процесса создания изображений с использованием нейросетей. Система объединяет несколько AI-моделей генерации в единый интерфейс и автоматизирует последующую обработку результатов.
Платформа решает задачу централизованного управления генерацией изображений через API, обеспечивая работу с различными моделями искусственного интеллекта, включая Midjourney, Stable Diffusion XL, FLUX и Luma Photon-1. Система автоматически обрабатывает результаты генерации: удаляет фон, анализирует цветовую палитру, обрезает изображения и подготавливает их для дальнейшего использования.
Решение
Архитектура построена на FastAPI и включает модульную систему обработки запросов. Основные компоненты:
API-сервер на FastAPI предоставляет REST-интерфейс для создания генераций, управления очередями и получения результатов. Реализована система приоритетов для обработки задач и поддержка асинхронных операций.
Интеграция с AI-моделями осуществляется через отдельные сервисы для каждой модели. Для Midjourney используется внешний API с webhook-уведомлениями, для локальных моделей (SDXL, FLUX) — выделенные микросервисы на базе библиотеки Diffusers.
Автоматическая постобработка включает удаление фона через внешний сервис, автоматическую обрезку прозрачных областей, подгонку изображений под квадратный формат и анализ цветовой палитры. Система определяет доминирующие цвета, средние значения светлых и тёмных тонов, что полезно для дальнейшей работы с изображениями.
Система уведомлений через webhook позволяет отслеживать статус генераций в реальном времени. Реализована поддержка множественных подписок и автоматические повторные попытки при сбоях.
База данных на SQLite (Peewee ORM) хранит информацию о генерациях, файлах и их метаданных. Поддерживается создание вариаций изображений с привязкой к оригинальным генерациям.
Контейнеризация через Docker обеспечивает простой деплой и изоляцию окружения. Настроена поддержка hot-reload для разработки.
Используемые технологии: Python, FastAPI, Peewee ORM, SQLite, Diffusers, Pillow, Docker, APScheduler, aiohttp.
Достигнутые результаты
Создана рабочая платформа, которая объединяет несколько AI-моделей в единый API. Система автоматизирует процесс от создания запроса до получения готовых обработанных изображений.
Реализована поддержка четырёх различных моделей генерации с единым интерфейсом. Автоматическая постобработка сокращает ручную работу и обеспечивает консистентный формат результатов. Система очередей с приоритетами позволяет эффективно управлять нагрузкой.
Webhook-уведомления обеспечивают интеграцию с внешними системами и отслеживание прогресса генераций. Модульная архитектура упрощает добавление новых моделей и расширение функциональности.
Хотите такой же проект?
Оставьте заявку и получите бесплатную консультацию по вашему проекту