ComfyUI Generator with LAN Ollama

PyQt5 приложение для генерации изображений в ComfyUI с использованием локальных и облачных LLM для улучшения промптов.

Описание

Графическое приложение для работы с ComfyUI, которое использует языковые модели (Ollama, LM Studio, облачные API) для автоматического улучшения и перевода промптов перед генерацией изображений.

Возможности

• 🎨 Интеграция с ComfyUI через WebSocket API
• 🤖 Поддержка множества LLM провайдеров:
  • Ollama (локально)
  • LM Studio (локальная сеть)
  • Облачные API (Gemini, DeepSeek, Grok)
• 🌐 Автоматический перевод промптов
• ✨ Улучшение промптов через AI
• 📊 Детальное логирование процесса
• 💾 Сохранение настроек

Установка

pip install PyQt5 websocket-client requests Pillow

Конфигурация

Настройки хранятся в settings.json:

{
  "comfyui": {
    "server_address": "192.168.1.118:8188"
  },
  "ollama": {
    "base_url": "http://127.0.0.1:11434/v1"
  },
  "lmstudio": {
    "base_url": "http://192.168.1.118:1234/v1",
    "api_key": "lm-studio",
    "models": ["luna-ai-llama2"]
  },
  "cloud_ai": {
    "base_url": "https://neuroapi.host/v1",
    "api_key": "YOUR_KEY_HERE",
    "models": ["gemini-2.5-flash", "deepseek-v3.2"]
  }
}

Использование

python app.py

1. Введите промпт на любом языке
2. Выберите модель для улучшения промпта
3. Настройте параметры генерации
4. Нажмите "Генерировать"

Логи

Детальные логи сохраняются в app_detailed_log.txt.