Author SHA1 Message Date
kromskii2 36db738601 Update README.md 2025-05-06 17:50:39 +07:00
kromskii2 18dfe41e39 Create .gitignore 2025-05-06 17:46:56 +07:00
kromskii2 df447a1ecf Добавлено меню настроек с возможностью изменения API, интерфейса, генерации и горячих клавиш. Обновлен основной интерфейс с добавлением меню. Созданы диалоговые окна для настройки параметров. 2025-03-15 08:38:56 +07:00
kromskii2 01395a5aa4 Merge pull request #3 from kromskii2/feature/prompt-translations
Feature/prompt translations
2025-03-14 20:23:17 +07:00
kromskii2 74dd9824f3 Merge pull request #2 from kromskii2/interface
Обновлен интерфейс окон: изменены стили и размеры элементов, улучшена…
2025-03-14 20:22:49 +07:00
kromskii2 11d03556da Merge pull request #1 from kromskii2/feature/negative-prompts
Добавлено окно негативных промптов с категориями и автоматическим пер…
2025-03-14 20:22:25 +07:00
kromskii2 02c7b9483b Улучшен интерфейс менеджера пресетов размеров: добавлен фрейм для кнопок, изменены стили кнопок и заголовка, улучшено разделение пресетов на две группы. Изменены названия пресетов для упрощения. Добавлен метод создания кнопок с улучшенным дизайном. 2025-03-14 20:20:55 +07:00
kromskii2 828f3c6c12 Обновлен интерфейс окон: изменены стили и размеры элементов, улучшена читаемость текста, добавлены иконки для кнопок, оптимизированы отступы и размеры компонентов. 2025-03-14 20:11:38 +07:00
kromskii2 8e6e54c802 Добавлены новые бинарные файлы для кэширования модулей: negative_prompt_categories, negative_prompt_window, prompt_categories, prompt_translations, prompt_window, size_preset_manager и size_presets. 2025-03-14 20:05:25 +07:00
kromskii2 603f6827b6 Добавлены пресеты размеров изображения с красивыми кнопками 2025-03-14 20:03:37 +07:00
kromskii2 41999b31c1 Добавлена кнопка копирования изображения в буфер обмена 2025-03-14 19:59:11 +07:00
kromskii2 6e2defe3b6 Добавлены переводы промптов и изменен размер окна негативных промптов 2025-03-14 19:56:00 +07:00
23 changed files with 1128 additions and 70 deletions

No files matched your search

+174
View File
@@ -0,0 +1,174 @@
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class
# C extensions
*.so
# Distribution / packaging
.Python
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
share/python-wheels/
*.egg-info/
.installed.cfg
*.egg
MANIFEST
# PyInstaller
# Usually these files are written by a python script from a template
# before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec
# Installer logs
pip-log.txt
pip-delete-this-directory.txt
# Unit test / coverage reports
htmlcov/
.tox/
.nox/
.coverage
.coverage.*
.cache
nosetests.xml
coverage.xml
*.cover
*.py,cover
.hypothesis/
.pytest_cache/
cover/
# Translations
*.mo
*.pot
# Django stuff:
*.log
local_settings.py
db.sqlite3
db.sqlite3-journal
# Flask stuff:
instance/
.webassets-cache
# Scrapy stuff:
.scrapy
# Sphinx documentation
docs/_build/
# PyBuilder
.pybuilder/
target/
# Jupyter Notebook
.ipynb_checkpoints
# IPython
profile_default/
ipython_config.py
# pyenv
# For a library or package, you might want to ignore these files since the code is
# intended to run in multiple environments; otherwise, check them in:
# .python-version
# pipenv
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
# However, in case of collaboration, if having platform-specific dependencies or dependencies
# having no cross-platform support, pipenv may install dependencies that don't work, or not
# install all needed dependencies.
#Pipfile.lock
# UV
# Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
# This is especially recommended for binary packages to ensure reproducibility, and is more
# commonly ignored for libraries.
#uv.lock
# poetry
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
# This is especially recommended for binary packages to ensure reproducibility, and is more
# commonly ignored for libraries.
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
#poetry.lock
# pdm
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
#pdm.lock
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
# in version control.
# https://pdm.fming.dev/latest/usage/project/#working-with-version-control
.pdm.toml
.pdm-python
.pdm-build/
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
__pypackages__/
# Celery stuff
celerybeat-schedule
celerybeat.pid
# SageMath parsed files
*.sage.py
# Environments
.env
.venv
env/
venv/
ENV/
env.bak/
venv.bak/
# Spyder project settings
.spyderproject
.spyproject
# Rope project settings
.ropeproject
# mkdocs documentation
/site
# mypy
.mypy_cache/
.dmypy.json
dmypy.json
# Pyre type checker
.pyre/
# pytype static type analyzer
.pytype/
# Cython debug symbols
cython_debug/
# PyCharm
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
# and can be added to the global gitignore or merged into this file. For a more nuclear
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
#.idea/
# Ruff stuff:
.ruff_cache/
# PyPI configuration file
.pypirc
BIN
View File
Binary file not shown.
+76 -4
View File
@@ -6,9 +6,9 @@ from time import sleep
from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
QHBoxLayout, QLabel, QLineEdit, QPushButton, QHBoxLayout, QLabel, QLineEdit, QPushButton,
QComboBox, QSpinBox, QDoubleSpinBox, QProgressBar, QComboBox, QSpinBox, QDoubleSpinBox, QProgressBar,
QScrollArea, QTextEdit, QFileDialog) QScrollArea, QTextEdit, QFileDialog, QMenuBar, QMenu)
from PyQt6.QtCore import Qt, QThread, pyqtSignal, QTimer from PyQt6.QtCore import Qt, QThread, pyqtSignal, QTimer
from PyQt6.QtGui import QPixmap, QImage from PyQt6.QtGui import QPixmap, QImage, QAction
from PIL import Image from PIL import Image
from io import BytesIO from io import BytesIO
@@ -18,6 +18,8 @@ from modules.model_manager import ModelManager
from modules.sampler_manager import SamplerManager from modules.sampler_manager import SamplerManager
from modules.prompt_window import PromptWindow from modules.prompt_window import PromptWindow
from modules.negative_prompt_window import NegativePromptWindow from modules.negative_prompt_window import NegativePromptWindow
from modules.size_preset_manager import SizePresetManager
from modules.settings_menu import SettingsMenu
class GenerationThread(QThread): class GenerationThread(QThread):
progress_updated = pyqtSignal(float, str) progress_updated = pyqtSignal(float, str)
@@ -67,8 +69,42 @@ class MainWindow(QMainWindow):
self.prompt_window = None self.prompt_window = None
self.negative_prompt_window = None self.negative_prompt_window = None
self.progress_timer = None self.progress_timer = None
# Создаем меню бар
self.create_menu_bar()
self.init_ui() self.init_ui()
def create_menu_bar(self):
"""Создание меню бара с настройками"""
menubar = QMenuBar(self)
menubar.setStyleSheet("""
QMenuBar {
background-color: #2b2b2b;
color: #ffffff;
border-bottom: 1px solid #3b3b3b;
padding: 2px;
}
QMenuBar::item {
background-color: transparent;
padding: 8px 12px;
margin: 0;
border-radius: 4px;
}
QMenuBar::item:selected {
background-color: #3b3b3b;
}
QMenuBar::item:pressed {
background-color: #404040;
}
""")
# Добавляем меню настроек
settings_menu = SettingsMenu(self)
menubar.addMenu(settings_menu)
self.setMenuBar(menubar)
def init_ui(self): def init_ui(self):
self.setWindowTitle("Stable Diffusion UI") self.setWindowTitle("Stable Diffusion UI")
self.setStyleSheet(""" self.setStyleSheet("""
@@ -202,6 +238,11 @@ class MainWindow(QMainWindow):
self.sampler_manager = SamplerManager(self.api_url) self.sampler_manager = SamplerManager(self.api_url)
settings_layout.addWidget(self.sampler_manager) settings_layout.addWidget(self.sampler_manager)
# Добавляем менеджер пресетов размеров
self.size_preset_manager = SizePresetManager()
self.size_preset_manager.size_changed.connect(self.on_size_preset_changed)
settings_layout.addWidget(self.size_preset_manager)
# Параметры генерации # Параметры генерации
params_layout = QVBoxLayout() params_layout = QVBoxLayout()
@@ -447,11 +488,20 @@ class MainWindow(QMainWindow):
save_button = QPushButton("Сохранить") save_button = QPushButton("Сохранить")
save_button.clicked.connect(lambda checked, img=img: self.save_image(img)) save_button.clicked.connect(lambda checked, img=img: self.save_image(img))
# Создаем контейнер для изображения и кнопки # Создаем кнопку копирования в буфер обмена
copy_button = QPushButton("Копировать")
copy_button.clicked.connect(lambda checked, img=img: self.copy_to_clipboard(img))
# Создаем контейнер для изображения и кнопок
container = QWidget() container = QWidget()
container_layout = QVBoxLayout(container) container_layout = QVBoxLayout(container)
container_layout.addWidget(img_label) container_layout.addWidget(img_label)
container_layout.addWidget(save_button)
# Создаем горизонтальный layout для кнопок
buttons_layout = QHBoxLayout()
buttons_layout.addWidget(save_button)
buttons_layout.addWidget(copy_button)
container_layout.addLayout(buttons_layout)
self.results_content_layout.addWidget(container) self.results_content_layout.addWidget(container)
self.log_to_console("Изображение успешно добавлено в интерфейс") self.log_to_console("Изображение успешно добавлено в интерфейс")
@@ -471,6 +521,22 @@ class MainWindow(QMainWindow):
if file_path: if file_path:
img.save(file_path) img.save(file_path)
def copy_to_clipboard(self, img):
"""Копирование изображения в буфер обмена"""
# Конвертируем изображение в QPixmap
buffer = BytesIO()
img.save(buffer, format='PNG')
buffer.seek(0)
pixmap = QPixmap()
pixmap.loadFromData(buffer.getvalue())
# Копируем в буфер обмена
clipboard = QApplication.clipboard()
clipboard.setPixmap(pixmap)
# Показываем уведомление
self.log_to_console("Изображение скопировано в буфер обмена")
def update_api_url(self): def update_api_url(self):
"""Обновление URL API во всех компонентах""" """Обновление URL API во всех компонентах"""
new_url = self.api_url_input.text() new_url = self.api_url_input.text()
@@ -478,6 +544,12 @@ class MainWindow(QMainWindow):
self.sampler_manager.update_api_url(new_url) self.sampler_manager.update_api_url(new_url)
self.log_to_console(f"API URL обновлен: {new_url}") self.log_to_console(f"API URL обновлен: {new_url}")
def on_size_preset_changed(self, width, height):
"""Обработчик изменения размера через пресет"""
self.width_input.setValue(width)
self.height_input.setValue(height)
self.log_to_console(f"Размер изменен на: {width}x{height}")
if __name__ == '__main__': if __name__ == '__main__':
app = QApplication(sys.argv) app = QApplication(sys.argv)
window = MainWindow() window = MainWindow()
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
+26 -19
View File
@@ -17,36 +17,42 @@ class ConsoleWindow(QDialog):
def setup_ui(self): def setup_ui(self):
self.setStyleSheet(""" self.setStyleSheet("""
QDialog { QDialog {
background-color: #2b2b2b; background-color: #1e1e1e;
min-width: 600px; min-width: 700px;
min-height: 400px; min-height: 500px;
} }
QTextEdit { QTextEdit {
background-color: #1e1e1e; background-color: #252526;
color: #ffffff; color: #d4d4d4;
border: 1px solid #3b3b3b; border: none;
border-radius: 4px; border-radius: 8px;
font-family: 'Consolas', 'Courier New', monospace; font-family: 'JetBrains Mono', 'Consolas', 'Courier New', monospace;
font-size: 12px; font-size: 13px;
padding: 8px; padding: 12px;
selection-background-color: #264f78;
} }
QPushButton { QPushButton {
background-color: #0d6efd; background-color: #2d2d2d;
color: white; color: #d4d4d4;
border: none; border: 1px solid #3e3e3e;
border-radius: 4px; border-radius: 6px;
padding: 8px 16px; padding: 8px 16px;
font-weight: bold; font-weight: 500;
min-width: 100px;
margin: 4px;
} }
QPushButton:hover { QPushButton:hover {
background-color: #0b5ed7; background-color: #3e3e3e;
border-color: #565656;
} }
QPushButton:pressed { QPushButton:pressed {
background-color: #0a58ca; background-color: #565656;
} }
""") """)
layout = QVBoxLayout(self) layout = QVBoxLayout(self)
layout.setContentsMargins(16, 16, 16, 16)
layout.setSpacing(12)
# Область для логов # Область для логов
self.console_output = QTextEdit() self.console_output = QTextEdit()
@@ -55,11 +61,12 @@ class ConsoleWindow(QDialog):
# Кнопки управления # Кнопки управления
buttons_layout = QHBoxLayout() buttons_layout = QHBoxLayout()
buttons_layout.setSpacing(8)
copy_button = QPushButton("Копировать всё") copy_button = QPushButton("📋 Копировать")
copy_button.clicked.connect(self.copy_all) copy_button.clicked.connect(self.copy_all)
clear_button = QPushButton("Очистить") clear_button = QPushButton("🗑️ Очистить")
clear_button.clicked.connect(self.clear_console) clear_button.clicked.connect(self.clear_console)
buttons_layout.addWidget(copy_button) buttons_layout.addWidget(copy_button)
+59
View File
@@ -13,8 +13,65 @@ class ModelManager(QWidget):
self.load_models() self.load_models()
def setup_ui(self): def setup_ui(self):
self.setStyleSheet("""
QWidget {
background-color: transparent;
}
QLabel {
color: #d4d4d4;
font-size: 13px;
font-weight: 500;
margin-right: 8px;
}
QComboBox {
background-color: #252526;
color: #d4d4d4;
border: 1px solid #3e3e3e;
border-radius: 6px;
padding: 6px 12px;
min-width: 200px;
font-size: 13px;
}
QComboBox:hover {
border-color: #565656;
}
QComboBox:focus {
border-color: #007acc;
}
QComboBox::drop-down {
border: none;
width: 20px;
}
QComboBox::down-arrow {
image: url(data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMTYiIGhlaWdodD0iMTYiIHZpZXdCb3g9IjAgMCAxNiAxNiIgZmlsbD0ibm9uZSIgeG1sbnM9Imh0dHA6Ly93d3cudzMub3JnLzIwMDAvc3ZnIj4KPHBhdGggZD0iTTggMTFMNCBTTDEyIDUiIHN0cm9rZT0iI2Q0ZDRkNCIgc3Ryb2tlLXdpZHRoPSIyIiBzdHJva2UtbGluZWNhcD0icm91bmQiIHN0cm9rZS1saW5lam9pbj0icm91bmQiLz4KPC9zdmc+);
}
QComboBox QAbstractItemView {
background-color: #252526;
color: #d4d4d4;
border: 1px solid #3e3e3e;
selection-background-color: #094771;
outline: none;
}
QPushButton {
background-color: #2d2d2d;
color: #d4d4d4;
border: 1px solid #3e3e3e;
border-radius: 6px;
padding: 6px 12px;
font-size: 13px;
}
QPushButton:hover {
background-color: #3e3e3e;
border-color: #565656;
}
QPushButton:pressed {
background-color: #565656;
}
""")
layout = QHBoxLayout(self) layout = QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0) layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(8)
# Метка # Метка
label = QLabel("Модель:") label = QLabel("Модель:")
@@ -22,11 +79,13 @@ class ModelManager(QWidget):
# Выпадающий список моделей # Выпадающий список моделей
self.model_combo = QComboBox() self.model_combo = QComboBox()
self.model_combo.setFixedHeight(32)
self.model_combo.currentTextChanged.connect(self.on_model_changed) self.model_combo.currentTextChanged.connect(self.on_model_changed)
layout.addWidget(self.model_combo) layout.addWidget(self.model_combo)
# Кнопка обновления # Кнопка обновления
refresh_button = QPushButton("🔄") refresh_button = QPushButton("🔄")
refresh_button.setFixedSize(32, 32)
refresh_button.setToolTip("Обновить список моделей") refresh_button.setToolTip("Обновить список моделей")
refresh_button.clicked.connect(self.load_models) refresh_button.clicked.connect(self.load_models)
layout.addWidget(refresh_button) layout.addWidget(refresh_button)
+2 -2
View File
@@ -18,8 +18,8 @@ class NegativePromptWindow(QDialog):
self.setStyleSheet(""" self.setStyleSheet("""
QDialog { QDialog {
background-color: #2b2b2b; background-color: #2b2b2b;
min-width: 1500px; min-width: 1700px;
min-height: 700px; min-height: 800px;
} }
QScrollArea { QScrollArea {
background-color: #1e1e1e; background-color: #1e1e1e;
+316
View File
@@ -0,0 +1,316 @@
PROMPT_TRANSLATIONS = {
# Персонажи
"Девушка": "girl",
"Парень": "boy",
"Мужчина": "man",
"Женщина": "woman",
"Ребенок": "child",
"Малыш": "baby",
"Пожилой человек": "elderly person",
"Старик": "old man",
"Старушка": "old woman",
# Профессии
"Художник": "artist",
"Музыкант": "musician",
"Писатель": "writer",
"Учитель": "teacher",
"Врач": "doctor",
"Инженер": "engineer",
"Программист": "programmer",
"Повар": "chef",
"Полицейский": "police officer",
"Пожарный": "firefighter",
# Животные
"Кошка": "cat",
"Собака": "dog",
"Птица": "bird",
"Рыба": "fish",
"Лошадь": "horse",
"Корова": "cow",
"Овца": "sheep",
"Коза": "goat",
"Кролик": "rabbit",
"Хомяк": "hamster",
# Природа
"Гора": "mountain",
"Море": "sea",
"Океан": "ocean",
"Река": "river",
"Лес": "forest",
"Поле": "field",
"Цветок": "flower",
"Дерево": "tree",
"Трава": "grass",
"Песок": "sand",
# Погода
"Солнце": "sun",
"Дождь": "rain",
"Снег": "snow",
"Ветер": "wind",
"Туман": "fog",
"Облако": "cloud",
"Гроза": "thunderstorm",
"Радуга": "rainbow",
"Мороз": "frost",
"Иней": "hoarfrost",
# Время суток
"Утро": "morning",
"День": "day",
"Вечер": "evening",
"Ночь": "night",
"Рассвет": "dawn",
"Закат": "sunset",
"Полдень": "noon",
"Полночь": "midnight",
# Сезоны
"Весна": "spring",
"Лето": "summer",
"Осень": "autumn",
"Зима": "winter",
# Эмоции
"Счастье": "happiness",
"Грусть": "sadness",
"Гнев": "anger",
"Страх": "fear",
"Удивление": "surprise",
"Отвращение": "disgust",
"Любовь": "love",
"Ненависть": "hate",
"Спокойствие": "calmness",
"Волнение": "excitement",
# Действия
"Бег": "running",
"Ходьба": "walking",
"Прыжок": "jumping",
"Плавание": "swimming",
"Танцы": "dancing",
"Пение": "singing",
"Игра": "playing",
"Работа": "working",
"Отдых": "resting",
"Сон": "sleeping",
# Одежда
"Платье": "dress",
"Рубашка": "shirt",
"Брюки": "pants",
"Юбка": "skirt",
"Куртка": "jacket",
"Пальто": "coat",
"Шапка": "hat",
"Обувь": "shoes",
"Перчатки": "gloves",
"Шарф": "scarf",
# Цвета
"Красный": "red",
"Синий": "blue",
"Зеленый": "green",
"Желтый": "yellow",
"Белый": "white",
"Черный": "black",
"Серый": "gray",
"Коричневый": "brown",
"Оранжевый": "orange",
"Фиолетовый": "purple",
# Материалы
"Дерево": "wood",
"Металл": "metal",
"Стекло": "glass",
"Пластик": "plastic",
"Ткань": "fabric",
"Бумага": "paper",
"Камень": "stone",
"Керамика": "ceramics",
"Кожа": "leather",
"Резина": "rubber",
# Технологии
"Компьютер": "computer",
"Телефон": "phone",
"Планшет": "tablet",
"Робот": "robot",
"Дрон": "drone",
"Смартфон": "smartphone",
"Ноутбук": "laptop",
"Принтер": "printer",
"Камера": "camera",
"Наушники": "headphones",
# Транспорт
"Машина": "car",
"Велосипед": "bicycle",
"Мотоцикл": "motorcycle",
"Поезд": "train",
"Самолет": "airplane",
"Корабль": "ship",
"Автобус": "bus",
"Трамвай": "tram",
"Метро": "subway",
"Вертолет": "helicopter",
# Еда
"Фрукты": "fruits",
"Овощи": "vegetables",
"Мясо": "meat",
"Рыба": "fish",
"Хлеб": "bread",
"Молоко": "milk",
"Сок": "juice",
"Кофе": "coffee",
"Чай": "tea",
"Сладости": "sweets",
# Дом
"Дом": "house",
"Квартира": "apartment",
"Комната": "room",
"Кухня": "kitchen",
"Спальня": "bedroom",
"Ванная": "bathroom",
"Гостиная": "living room",
"Балкон": "balcony",
"Подвал": "basement",
"Чердак": "attic",
# Музыка
"Гитара": "guitar",
"Пианино": "piano",
"Скрипка": "violin",
"Барабаны": "drums",
"Флейта": "flute",
"Саксофон": "saxophone",
"Труба": "trumpet",
"Бас": "bass",
"Синтезатор": "synthesizer",
"Микрофон": "microphone",
# Спорт
"Футбол": "football",
"Баскетбол": "basketball",
"Волейбол": "volleyball",
"Теннис": "tennis",
"Хоккей": "hockey",
"Бег": "running",
"Плавание": "swimming",
"Бокс": "boxing",
"Йога": "yoga",
"Гимнастика": "gymnastics",
# Искусство
"Картина": "painting",
"Рисунок": "drawing",
"Скульптура": "sculpture",
"Фотография": "photography",
"Кино": "movie",
"Театр": "theater",
"Музыка": "music",
"Танец": "dance",
"Поэзия": "poetry",
"Литература": "literature",
# Наука
"Физика": "physics",
"Химия": "chemistry",
"Биология": "biology",
"Математика": "mathematics",
"Астрономия": "astronomy",
"Геология": "geology",
"Экология": "ecology",
"Психология": "psychology",
"История": "history",
"География": "geography",
# Время
"Секунда": "second",
"Минута": "minute",
"Час": "hour",
"День": "day",
"Неделя": "week",
"Месяц": "month",
"Год": "year",
"Век": "century",
"Эпоха": "era",
"Период": "period",
# География
"Страна": "country",
"Город": "city",
"Деревня": "village",
"Река": "river",
"Озеро": "lake",
"Море": "sea",
"Океан": "ocean",
"Гора": "mountain",
"Пустыня": "desert",
"Джунгли": "jungle",
# Космос
"Солнце": "sun",
"Луна": "moon",
"Звезда": "star",
"Планета": "planet",
"Галактика": "galaxy",
"Вселенная": "universe",
"Астероид": "asteroid",
"Комета": "comet",
"Спутник": "satellite",
"Космический корабль": "spaceship",
# Фэнтези
"Дракон": "dragon",
"Единорог": "unicorn",
"Фея": "fairy",
"Волшебник": "wizard",
"Ведьма": "witch",
"Гоблин": "goblin",
"Эльф": "elf",
"Гном": "dwarf",
"Орк": "orc",
"Тролль": "troll",
# Мифология
"Бог": "god",
"Богиня": "goddess",
"Герой": "hero",
"Миф": "myth",
"Легенда": "legend",
"Демон": "demon",
"Ангел": "angel",
"Призрак": "ghost",
"Вампир": "vampire",
"Оборотень": "werewolf",
# Стили искусства
"Реализм": "realism",
"Импрессионизм": "impressionism",
"Экспрессионизм": "expressionism",
"Сюрреализм": "surrealism",
"Абстракционизм": "abstract art",
"Поп-арт": "pop art",
"Модернизм": "modernism",
"Постмодернизм": "postmodernism",
"Кубизм": "cubism",
"Футуризм": "futurism",
# Эпохи
"Древний мир": "ancient world",
"Средневековье": "middle ages",
"Ренессанс": "renaissance",
"Барокко": "baroque",
"Классицизм": "classicism",
"Романтизм": "romanticism",
"Викторианская эпоха": "victorian era",
"Модерн": "art nouveau",
"Ар-деко": "art deco",
"Современность": "modern era"
}
+70 -45
View File
@@ -3,6 +3,7 @@ from PyQt6.QtWidgets import (QDialog, QVBoxLayout, QHBoxLayout, QTreeWidget,
QScrollArea, QWidget, QCheckBox, QFrame) QScrollArea, QWidget, QCheckBox, QFrame)
from PyQt6.QtCore import Qt, pyqtSignal from PyQt6.QtCore import Qt, pyqtSignal
from .prompt_categories import PROMPT_CATEGORIES from .prompt_categories import PROMPT_CATEGORIES
from .prompt_translations import PROMPT_TRANSLATIONS
class PromptWindow(QDialog): class PromptWindow(QDialog):
prompt_selected = pyqtSignal(str) prompt_selected = pyqtSignal(str)
@@ -18,87 +19,105 @@ class PromptWindow(QDialog):
def setup_ui(self): def setup_ui(self):
self.setStyleSheet(""" self.setStyleSheet("""
QDialog { QDialog {
background-color: #2b2b2b; background-color: #1e1e1e;
min-width: 1500px; min-width: 1200px;
min-height: 700px; min-height: 800px;
} }
QScrollArea { QScrollArea {
background-color: #1e1e1e; background-color: #252526;
border: 1px solid #3b3b3b; border: none;
border-radius: 4px; border-radius: 8px;
padding: 8px;
} }
QScrollArea > QWidget > QWidget { QScrollArea > QWidget > QWidget {
background-color: #1e1e1e; background-color: #252526;
} }
QLabel { QLabel {
color: #ffffff; color: #d4d4d4;
font-size: 14px; font-size: 14px;
font-weight: bold; font-weight: 600;
padding: 8px; padding: 12px;
background-color: #3b3b3b; background-color: #2d2d2d;
border-radius: 4px; border-radius: 6px;
margin-bottom: 8px;
} }
QCheckBox { QCheckBox {
color: #ffffff; color: #d4d4d4;
font-size: 14px; font-size: 13px;
padding: 4px; padding: 6px;
spacing: 8px; spacing: 8px;
} }
QCheckBox:hover {
background-color: #2d2d2d;
border-radius: 4px;
}
QCheckBox::indicator { QCheckBox::indicator {
width: 18px; width: 18px;
height: 18px; height: 18px;
border-radius: 4px;
} }
QCheckBox::indicator:unchecked { QCheckBox::indicator:unchecked {
border: 2px solid #555555; border: 2px solid #565656;
background-color: #3b3b3b; background-color: #252526;
border-radius: 3px;
} }
QCheckBox::indicator:checked { QCheckBox::indicator:checked {
border: 2px solid #0d6efd; border: 2px solid #007acc;
background-color: #0d6efd; background-color: #007acc;
border-radius: 3px;
} }
QCheckBox::indicator:hover { QCheckBox::indicator:hover {
border-color: #0b5ed7; border-color: #007acc;
} }
QPushButton { QPushButton {
background-color: #0d6efd; background-color: #2d2d2d;
color: white; color: #d4d4d4;
border: none; border: 1px solid #3e3e3e;
border-radius: 4px; border-radius: 6px;
padding: 8px 16px; padding: 8px 16px;
font-weight: bold; font-weight: 500;
min-width: 120px; min-width: 120px;
font-size: 13px;
} }
QPushButton:hover { QPushButton:hover {
background-color: #0b5ed7; background-color: #3e3e3e;
border-color: #565656;
} }
QPushButton:pressed { QPushButton:pressed {
background-color: #0a58ca; background-color: #565656;
} }
QLineEdit { QLineEdit {
background-color: #3b3b3b; background-color: #252526;
color: white; color: #d4d4d4;
border: 1px solid #555555; border: 1px solid #3e3e3e;
border-radius: 4px; border-radius: 6px;
padding: 8px; padding: 8px 12px;
font-size: 14px; font-size: 13px;
min-height: 24px; min-height: 32px;
}
QLineEdit:focus {
border-color: #007acc;
} }
QFrame { QFrame {
background-color: #1e1e1e; background-color: #252526;
border: 1px solid #3b3b3b; border: 1px solid #3e3e3e;
border-radius: 4px; border-radius: 8px;
padding: 12px;
margin: 4px;
}
QFrame:hover {
border-color: #565656;
} }
""") """)
layout = QVBoxLayout(self) layout = QVBoxLayout(self)
layout.setContentsMargins(16, 16, 16, 16)
layout.setSpacing(12)
# Поиск # Поиск
search_layout = QHBoxLayout() search_layout = QHBoxLayout()
search_label = QLabel("Поиск:") search_layout.setSpacing(8)
search_label = QLabel("🔍 Поиск:")
search_label.setStyleSheet("background: transparent; padding: 0;")
self.search_input = QLineEdit() self.search_input = QLineEdit()
self.search_input.setPlaceholderText("Введите текст для поиска...")
self.search_input.textChanged.connect(self.filter_prompts) self.search_input.textChanged.connect(self.filter_prompts)
search_layout.addWidget(search_label) search_layout.addWidget(search_label)
search_layout.addWidget(self.search_input) search_layout.addWidget(self.search_input)
@@ -109,22 +128,26 @@ class PromptWindow(QDialog):
categories_scroll.setWidgetResizable(True) categories_scroll.setWidgetResizable(True)
categories_scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOn) categories_scroll.setHorizontalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOn)
categories_scroll.setVerticalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOff) categories_scroll.setVerticalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOff)
categories_scroll.setMinimumHeight(600)
categories_widget = QWidget() categories_widget = QWidget()
self.categories_layout = QHBoxLayout(categories_widget) self.categories_layout = QHBoxLayout(categories_widget)
self.categories_layout.setSpacing(10) self.categories_layout.setSpacing(12)
self.categories_layout.setContentsMargins(12, 12, 12, 12)
categories_scroll.setWidget(categories_widget) categories_scroll.setWidget(categories_widget)
layout.addWidget(categories_scroll) layout.addWidget(categories_scroll)
# Кнопки # Кнопки
buttons_layout = QHBoxLayout() buttons_layout = QHBoxLayout()
add_button = QPushButton("Добавить выбранные") buttons_layout.setSpacing(8)
add_button = QPushButton("✨ Добавить выбранные")
add_button.clicked.connect(self.add_selected) add_button.clicked.connect(self.add_selected)
close_button = QPushButton("Закрыть") close_button = QPushButton("❌ Закрыть")
close_button.clicked.connect(self.close) close_button.clicked.connect(self.close)
buttons_layout.addWidget(add_button) buttons_layout.addWidget(add_button)
buttons_layout.addWidget(close_button) buttons_layout.addWidget(close_button)
buttons_layout.addStretch()
layout.addLayout(buttons_layout) layout.addLayout(buttons_layout)
def load_categories(self): def load_categories(self):
@@ -195,5 +218,7 @@ class PromptWindow(QDialog):
def add_selected(self): def add_selected(self):
"""Добавление выбранных промптов""" """Добавление выбранных промптов"""
if self.selected_prompts: if self.selected_prompts:
prompts_text = ", ".join(sorted(self.selected_prompts)) # Преобразуем русские названия в английские
translated_prompts = [PROMPT_TRANSLATIONS.get(prompt, prompt) for prompt in self.selected_prompts]
prompts_text = ", ".join(sorted(translated_prompts))
self.prompt_selected.emit(prompts_text) self.prompt_selected.emit(prompts_text)
+43
View File
@@ -13,8 +13,50 @@ class SamplerManager(QWidget):
self.load_samplers() self.load_samplers()
def setup_ui(self): def setup_ui(self):
self.setStyleSheet("""
QWidget {
background-color: transparent;
}
QLabel {
color: #d4d4d4;
font-size: 13px;
font-weight: 500;
margin-right: 8px;
}
QComboBox {
background-color: #252526;
color: #d4d4d4;
border: 1px solid #3e3e3e;
border-radius: 6px;
padding: 6px 12px;
min-width: 180px;
font-size: 13px;
}
QComboBox:hover {
border-color: #565656;
}
QComboBox:focus {
border-color: #007acc;
}
QComboBox::drop-down {
border: none;
width: 20px;
}
QComboBox::down-arrow {
image: url(data:image/svg+xml;base64,PHN2ZyB3aWR0aD0iMTYiIGhlaWdodD0iMTYiIHZpZXdCb3g9IjAgMCAxNiAxNiIgZmlsbD0ibm9uZSIgeG1sbnM9Imh0dHA6Ly93d3cudzMub3JnLzIwMDAvc3ZnIj4KPHBhdGggZD0iTTggMTFMNCBTTDEyIDUiIHN0cm9rZT0iI2Q0ZDRkNCIgc3Ryb2tlLXdpZHRoPSIyIiBzdHJva2UtbGluZWNhcD0icm91bmQiIHN0cm9rZS1saW5lam9pbj0icm91bmQiLz4KPC9zdmc+);
}
QComboBox QAbstractItemView {
background-color: #252526;
color: #d4d4d4;
border: 1px solid #3e3e3e;
selection-background-color: #094771;
outline: none;
}
""")
layout = QHBoxLayout(self) layout = QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0) layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(8)
# Метка # Метка
label = QLabel("Сэмплер:") label = QLabel("Сэмплер:")
@@ -22,6 +64,7 @@ class SamplerManager(QWidget):
# Выпадающий список сэмплеров # Выпадающий список сэмплеров
self.sampler_combo = QComboBox() self.sampler_combo = QComboBox()
self.sampler_combo.setFixedHeight(32)
self.sampler_combo.currentTextChanged.connect(self.on_sampler_changed) self.sampler_combo.currentTextChanged.connect(self.on_sampler_changed)
layout.addWidget(self.sampler_combo) layout.addWidget(self.sampler_combo)
+261
View File
@@ -0,0 +1,261 @@
from PyQt6.QtWidgets import (QMenu,
QDialog, QVBoxLayout, QHBoxLayout,
QLabel, QLineEdit, QCheckBox,
QPushButton, QComboBox, QSpinBox)
from PyQt6.QtGui import QIcon, QAction, QActionGroup
from PyQt6.QtCore import Qt
class SettingsMenu(QMenu):
def __init__(self, parent=None):
super().__init__("⚙️ Настройки", parent)
self.setStyleSheet("""
QMenu {
background-color: #2b2b2b;
border: 1px solid #3b3b3b;
border-radius: 8px;
padding: 5px;
}
QMenu::item {
background-color: transparent;
padding: 8px 25px;
margin: 2px;
border-radius: 4px;
color: #ffffff;
font-size: 13px;
}
QMenu::item:selected {
background-color: #3b3b3b;
}
QMenu::separator {
height: 1px;
background-color: #3b3b3b;
margin: 5px 15px;
}
""")
self.setup_menu()
def setup_menu(self):
# API настройки
api_action = QAction("🌐 API настройки", self)
api_action.triggered.connect(self.show_api_settings)
# Настройки интерфейса
ui_action = QAction("🎨 Настройки интерфейса", self)
ui_action.triggered.connect(self.show_ui_settings)
# Настройки генерации
generation_action = QAction("⚡ Настройки генерации", self)
generation_action.triggered.connect(self.show_generation_settings)
# Горячие клавиши
hotkeys_action = QAction("⌨️ Горячие клавиши", self)
hotkeys_action.triggered.connect(self.show_hotkeys_settings)
# Добавляем действия в меню
self.addAction(api_action)
self.addAction(ui_action)
self.addAction(generation_action)
self.addSeparator()
self.addAction(hotkeys_action)
def show_api_settings(self):
dialog = APISettingsDialog(self.parent())
dialog.exec()
def show_ui_settings(self):
dialog = UISettingsDialog(self.parent())
dialog.exec()
def show_generation_settings(self):
dialog = GenerationSettingsDialog(self.parent())
dialog.exec()
def show_hotkeys_settings(self):
dialog = HotkeysSettingsDialog(self.parent())
dialog.exec()
class BaseSettingsDialog(QDialog):
def __init__(self, title, parent=None):
super().__init__(parent)
self.setWindowTitle(title)
self.setup_base_ui()
def setup_base_ui(self):
self.setStyleSheet("""
QDialog {
background-color: #2b2b2b;
min-width: 400px;
}
QLabel {
color: #ffffff;
font-size: 13px;
}
QLineEdit, QComboBox, QSpinBox {
background-color: #3b3b3b;
border: 1px solid #555555;
border-radius: 4px;
padding: 5px;
color: #ffffff;
min-width: 200px;
}
QPushButton {
background-color: #0d6efd;
color: white;
border: none;
border-radius: 4px;
padding: 8px 16px;
font-weight: bold;
min-width: 100px;
}
QPushButton:hover {
background-color: #0b5ed7;
}
QCheckBox {
color: #ffffff;
}
""")
self.layout = QVBoxLayout(self)
self.layout.setSpacing(15)
self.layout.setContentsMargins(20, 20, 20, 20)
class APISettingsDialog(BaseSettingsDialog):
def __init__(self, parent=None):
super().__init__("🌐 API настройки", parent)
self.setup_ui()
def setup_ui(self):
# API URL
url_layout = QHBoxLayout()
url_label = QLabel("API URL:")
self.url_input = QLineEdit(self.parent().api_url_input.text())
url_layout.addWidget(url_label)
url_layout.addWidget(self.url_input)
self.layout.addLayout(url_layout)
# Кнопки
buttons_layout = QHBoxLayout()
save_button = QPushButton("Сохранить")
save_button.clicked.connect(self.save_settings)
cancel_button = QPushButton("Отмена")
cancel_button.clicked.connect(self.reject)
buttons_layout.addWidget(save_button)
buttons_layout.addWidget(cancel_button)
self.layout.addLayout(buttons_layout)
def save_settings(self):
self.parent().api_url_input.setText(self.url_input.text())
self.accept()
class UISettingsDialog(BaseSettingsDialog):
def __init__(self, parent=None):
super().__init__("🎨 Настройки интерфейса", parent)
self.setup_ui()
def setup_ui(self):
# Тема
theme_layout = QHBoxLayout()
theme_label = QLabel("Тема:")
self.theme_combo = QComboBox()
self.theme_combo.addItems(["Тёмная", "Светлая"])
theme_layout.addWidget(theme_label)
theme_layout.addWidget(self.theme_combo)
self.layout.addLayout(theme_layout)
# Размер шрифта
font_layout = QHBoxLayout()
font_label = QLabel("Размер шрифта:")
self.font_spin = QSpinBox()
self.font_spin.setRange(8, 24)
self.font_spin.setValue(14)
font_layout.addWidget(font_label)
font_layout.addWidget(self.font_spin)
self.layout.addLayout(font_layout)
# Кнопки
buttons_layout = QHBoxLayout()
save_button = QPushButton("Сохранить")
save_button.clicked.connect(self.save_settings)
cancel_button = QPushButton("Отмена")
cancel_button.clicked.connect(self.reject)
buttons_layout.addWidget(save_button)
buttons_layout.addWidget(cancel_button)
self.layout.addLayout(buttons_layout)
def save_settings(self):
# TODO: Применить настройки интерфейса
self.accept()
class GenerationSettingsDialog(BaseSettingsDialog):
def __init__(self, parent=None):
super().__init__("⚡ Настройки генерации", parent)
self.setup_ui()
def setup_ui(self):
# Качество по умолчанию
quality_layout = QHBoxLayout()
quality_label = QLabel("Качество по умолчанию:")
self.quality_combo = QComboBox()
self.quality_combo.addItems(["Низкое", "Среднее", "Высокое"])
quality_layout.addWidget(quality_label)
quality_layout.addWidget(self.quality_combo)
self.layout.addLayout(quality_layout)
# Размер по умолчанию
size_layout = QHBoxLayout()
size_label = QLabel("Размер по умолчанию:")
self.size_combo = QComboBox()
self.size_combo.addItems(["512x512", "768x512", "512x768"])
size_layout.addWidget(size_label)
size_layout.addWidget(self.size_combo)
self.layout.addLayout(size_layout)
# Кнопки
buttons_layout = QHBoxLayout()
save_button = QPushButton("Сохранить")
save_button.clicked.connect(self.save_settings)
cancel_button = QPushButton("Отмена")
cancel_button.clicked.connect(self.reject)
buttons_layout.addWidget(save_button)
buttons_layout.addWidget(cancel_button)
self.layout.addLayout(buttons_layout)
def save_settings(self):
# TODO: Применить настройки генерации
self.accept()
class HotkeysSettingsDialog(BaseSettingsDialog):
def __init__(self, parent=None):
super().__init__("⌨️ Горячие клавиши", parent)
self.setup_ui()
def setup_ui(self):
# Генерация
generate_layout = QHBoxLayout()
generate_label = QLabel("Генерация:")
self.generate_input = QLineEdit("Ctrl+G")
generate_layout.addWidget(generate_label)
generate_layout.addWidget(self.generate_input)
self.layout.addLayout(generate_layout)
# Открыть консоль
console_layout = QHBoxLayout()
console_label = QLabel("Открыть консоль:")
self.console_input = QLineEdit("Ctrl+`")
console_layout.addWidget(console_label)
console_layout.addWidget(self.console_input)
self.layout.addLayout(console_layout)
# Кнопки
buttons_layout = QHBoxLayout()
save_button = QPushButton("Сохранить")
save_button.clicked.connect(self.save_settings)
cancel_button = QPushButton("Отмена")
cancel_button.clicked.connect(self.reject)
buttons_layout.addWidget(save_button)
buttons_layout.addWidget(cancel_button)
self.layout.addLayout(buttons_layout)
def save_settings(self):
# TODO: Применить настройки горячих клавиш
self.accept()
+93
View File
@@ -0,0 +1,93 @@
from PyQt6.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel, QFrame
from PyQt6.QtCore import Qt, pyqtSignal
from .size_presets import SIZE_PRESETS
class SizePresetManager(QWidget):
size_changed = pyqtSignal(int, int) # Сигнал для изменения размера
def __init__(self, parent=None):
super().__init__(parent)
self.setup_ui()
def setup_ui(self):
layout = QVBoxLayout(self)
layout.setSpacing(2)
# Заголовок
title = QLabel("Размеры:")
title.setStyleSheet("""
QLabel {
color: #ffffff;
font-size: 12px;
font-weight: bold;
padding: 4px;
background-color: #3b3b3b;
border-radius: 4px;
}
""")
layout.addWidget(title)
# Создаем фрейм для кнопок
buttons_frame = QFrame()
buttons_frame.setStyleSheet("""
QFrame {
background-color: #2b2b2b;
border: 1px solid #3b3b3b;
border-radius: 4px;
padding: 2px;
}
""")
buttons_layout = QVBoxLayout(buttons_frame)
buttons_layout.setSpacing(2)
# Создаем две строки кнопок
row1_layout = QHBoxLayout()
row2_layout = QHBoxLayout()
# Разделяем пресеты на две группы
presets = list(SIZE_PRESETS.items())
half = len(presets) // 2
# Добавляем кнопки в первую строку
for name, (width, height) in presets[:half]:
button = self.create_preset_button(name, width, height)
row1_layout.addWidget(button)
# Добавляем кнопки во вторую строку
for name, (width, height) in presets[half:]:
button = self.create_preset_button(name, width, height)
row2_layout.addWidget(button)
buttons_layout.addLayout(row1_layout)
buttons_layout.addLayout(row2_layout)
layout.addWidget(buttons_frame)
def create_preset_button(self, name, width, height):
"""Создание кнопки пресета с улучшенным дизайном"""
button = QPushButton(name)
button.setStyleSheet(f"""
QPushButton {{
background-color: #3b3b3b;
color: white;
border: 1px solid #555555;
border-radius: 4px;
padding: 4px;
font-size: 11px;
min-width: 80px;
text-align: center;
}}
QPushButton:hover {{
background-color: #4a4a4a;
border-color: #666666;
}}
QPushButton:pressed {{
background-color: #0d6efd;
border-color: #0b5ed7;
}}
""")
button.clicked.connect(lambda checked, w=width, h=height: self.on_preset_clicked(w, h))
return button
def on_preset_clicked(self, width, height):
"""Обработчик нажатия на кнопку пресета"""
self.size_changed.emit(width, height)
+8
View File
@@ -0,0 +1,8 @@
SIZE_PRESETS = {
"512x768": (512, 768),
"768x512": (768, 512),
"512x512": (512, 512),
"768x432": (768, 432),
"432x768": (432, 768),
"600x800": (600, 800)
}