Воркед
This commit is contained in:
kromskii2 committed
1 parent
ac32c55ab9
commit
ee94d8e529
1138 files changed
+55771
-77901
No files matched your search
+210
-168
@@ -1,195 +1,237 @@
|
||||
from PyQt6.QtWidgets import (QWidget, QVBoxLayout, QHBoxLayout,
|
||||
QLabel, QPushButton, QLineEdit, QTextEdit,
|
||||
QSpinBox, QDoubleSpinBox, QComboBox)
|
||||
from PyQt6.QtCore import Qt
|
||||
from PyQt6.QtWidgets import (QWidget, QVBoxLayout, QHBoxLayout,
|
||||
QLabel, QPushButton, QLineEdit, QSpinBox,
|
||||
QComboBox, QProgressBar)
|
||||
from PyQt6.QtCore import Qt, pyqtSignal
|
||||
from PyQt6.QtGui import QPixmap
|
||||
import requests
|
||||
from ..styles.style_constants import StyleConstants
|
||||
from ..styles.icons import IconManager, IconType
|
||||
from ..styles.page import BasePage
|
||||
from ..styles.theme import Colors, Spacing, Fonts
|
||||
from ..styles.theme_manager import ThemeManager
|
||||
|
||||
class HomePage(BasePage):
|
||||
def __init__(self, parent=None):
|
||||
def __init__(self, model_manager, sampler_manager, size_preset_manager, parent=None):
|
||||
super().__init__(parent)
|
||||
self.theme_manager = ThemeManager.get_instance()
|
||||
self.setup_content()
|
||||
self.model_manager = model_manager
|
||||
self.sampler_manager = sampler_manager
|
||||
self.size_preset_manager = size_preset_manager
|
||||
self.icon_manager = IconManager.get_instance()
|
||||
self.setup_ui()
|
||||
|
||||
def setup_content(self):
|
||||
"""Настройка содержимого страницы"""
|
||||
colors = self.theme_manager.current_theme.get_colors()
|
||||
spacing = self.theme_manager.current_theme.get_spacing()
|
||||
def setup_ui(self):
|
||||
# Верхняя панель с настройками
|
||||
settings_panel = QHBoxLayout()
|
||||
settings_panel.setContentsMargins(
|
||||
StyleConstants.SPACING['lg'],
|
||||
StyleConstants.SPACING['lg'],
|
||||
StyleConstants.SPACING['lg'],
|
||||
StyleConstants.SPACING['lg']
|
||||
)
|
||||
settings_panel.setSpacing(StyleConstants.SPACING['md'])
|
||||
|
||||
# Заголовок
|
||||
header = QLabel("Генерация изображений")
|
||||
header.setStyleSheet(f"""
|
||||
QLabel {{
|
||||
color: {colors['text']};
|
||||
font-size: 18px;
|
||||
font-weight: bold;
|
||||
}}
|
||||
""")
|
||||
self.layout.addWidget(header)
|
||||
# Выбор модели
|
||||
model_label = QLabel("Модель:")
|
||||
settings_panel.addWidget(model_label)
|
||||
|
||||
# Промпт
|
||||
prompt_layout = QVBoxLayout()
|
||||
self.model_combo = QComboBox()
|
||||
self.model_combo.currentTextChanged.connect(self.on_model_changed)
|
||||
settings_panel.addWidget(self.model_combo)
|
||||
|
||||
# Выбор сэмплера
|
||||
sampler_label = QLabel("Сэмплер:")
|
||||
settings_panel.addWidget(sampler_label)
|
||||
|
||||
self.sampler_combo = QComboBox()
|
||||
self.sampler_combo.currentTextChanged.connect(self.on_sampler_changed)
|
||||
settings_panel.addWidget(self.sampler_combo)
|
||||
|
||||
# Настройки размера
|
||||
size_label = QLabel("Размер:")
|
||||
settings_panel.addWidget(size_label)
|
||||
|
||||
self.width_spin = QSpinBox()
|
||||
self.width_spin.setRange(64, 2048)
|
||||
self.width_spin.setSingleStep(64)
|
||||
self.width_spin.setValue(512)
|
||||
settings_panel.addWidget(self.width_spin)
|
||||
|
||||
self.height_spin = QSpinBox()
|
||||
self.height_spin.setRange(64, 2048)
|
||||
self.height_spin.setSingleStep(64)
|
||||
self.height_spin.setValue(512)
|
||||
settings_panel.addWidget(self.height_spin)
|
||||
|
||||
settings_panel.addStretch()
|
||||
self.layout.addLayout(settings_panel)
|
||||
|
||||
# Панель промптов
|
||||
prompts_panel = QVBoxLayout()
|
||||
prompts_panel.setContentsMargins(
|
||||
StyleConstants.SPACING['lg'],
|
||||
0,
|
||||
StyleConstants.SPACING['lg'],
|
||||
StyleConstants.SPACING['lg']
|
||||
)
|
||||
prompts_panel.setSpacing(StyleConstants.SPACING['md'])
|
||||
|
||||
# Позитивный промпт
|
||||
prompt_label = QLabel("Промпт:")
|
||||
prompt_label.setStyleSheet(f"""
|
||||
QLabel {{
|
||||
color: {colors['text']};
|
||||
font-size: 14px;
|
||||
font-weight: 500;
|
||||
}}
|
||||
""")
|
||||
self.prompt_input = QTextEdit()
|
||||
self.prompt_input.setPlaceholderText("Опишите изображение, которое хотите сгенерировать...")
|
||||
self.prompt_input.setMinimumHeight(100)
|
||||
prompts_panel.addWidget(prompt_label)
|
||||
|
||||
prompt_layout.addWidget(prompt_label)
|
||||
prompt_layout.addWidget(self.prompt_input)
|
||||
self.layout.addLayout(prompt_layout)
|
||||
self.prompt_edit = QLineEdit()
|
||||
self.prompt_edit.setPlaceholderText("Опишите желаемое изображение...")
|
||||
prompts_panel.addWidget(self.prompt_edit)
|
||||
|
||||
# Негативный промпт
|
||||
negative_prompt_layout = QVBoxLayout()
|
||||
negative_prompt_label = QLabel("Негативный промпт:")
|
||||
negative_prompt_label.setStyleSheet(f"""
|
||||
QLabel {{
|
||||
color: {colors['text']};
|
||||
font-size: 14px;
|
||||
font-weight: 500;
|
||||
}}
|
||||
""")
|
||||
self.negative_prompt_input = QTextEdit()
|
||||
self.negative_prompt_input.setPlaceholderText("Опишите, что не должно появиться на изображении...")
|
||||
self.negative_prompt_input.setMinimumHeight(100)
|
||||
prompts_panel.addWidget(negative_prompt_label)
|
||||
|
||||
negative_prompt_layout.addWidget(negative_prompt_label)
|
||||
negative_prompt_layout.addWidget(self.negative_prompt_input)
|
||||
self.layout.addLayout(negative_prompt_layout)
|
||||
self.negative_prompt_edit = QLineEdit()
|
||||
self.negative_prompt_edit.setPlaceholderText("Опишите нежелательные элементы...")
|
||||
prompts_panel.addWidget(self.negative_prompt_edit)
|
||||
|
||||
# Параметры генерации
|
||||
params_layout = QHBoxLayout()
|
||||
self.layout.addLayout(prompts_panel)
|
||||
|
||||
# Размер изображения
|
||||
size_layout = QVBoxLayout()
|
||||
size_label = QLabel("Размер:")
|
||||
size_label.setStyleSheet(f"""
|
||||
QLabel {{
|
||||
color: {colors['text']};
|
||||
font-size: 14px;
|
||||
font-weight: 500;
|
||||
}}
|
||||
""")
|
||||
self.size_combo = QComboBox()
|
||||
self.size_combo.addItems(["512x512", "768x768", "1024x1024"])
|
||||
size_layout.addWidget(size_label)
|
||||
size_layout.addWidget(self.size_combo)
|
||||
params_layout.addLayout(size_layout)
|
||||
# Панель генерации
|
||||
generate_panel = QHBoxLayout()
|
||||
generate_panel.setContentsMargins(
|
||||
StyleConstants.SPACING['lg'],
|
||||
0,
|
||||
StyleConstants.SPACING['lg'],
|
||||
StyleConstants.SPACING['lg']
|
||||
)
|
||||
generate_panel.setSpacing(StyleConstants.SPACING['md'])
|
||||
|
||||
# Количество шагов
|
||||
steps_layout = QVBoxLayout()
|
||||
steps_label = QLabel("Шаги:")
|
||||
steps_label.setStyleSheet(f"""
|
||||
QLabel {{
|
||||
color: {colors['text']};
|
||||
font-size: 14px;
|
||||
font-weight: 500;
|
||||
}}
|
||||
""")
|
||||
self.steps_spin = QSpinBox()
|
||||
self.steps_spin.setRange(1, 150)
|
||||
self.steps_spin.setValue(20)
|
||||
steps_layout.addWidget(steps_label)
|
||||
steps_layout.addWidget(self.steps_spin)
|
||||
params_layout.addLayout(steps_layout)
|
||||
self.steps_spin.setPrefix("Шаги: ")
|
||||
generate_panel.addWidget(self.steps_spin)
|
||||
|
||||
# CFG Scale
|
||||
cfg_layout = QVBoxLayout()
|
||||
cfg_label = QLabel("CFG Scale:")
|
||||
cfg_label.setStyleSheet(f"""
|
||||
QLabel {{
|
||||
color: {colors['text']};
|
||||
font-size: 14px;
|
||||
font-weight: 500;
|
||||
}}
|
||||
""")
|
||||
self.cfg_spin = QDoubleSpinBox()
|
||||
self.cfg_spin.setRange(1.0, 30.0)
|
||||
self.cfg_spin.setValue(7.0)
|
||||
self.cfg_spin.setSingleStep(0.5)
|
||||
cfg_layout.addWidget(cfg_label)
|
||||
cfg_layout.addWidget(self.cfg_spin)
|
||||
params_layout.addLayout(cfg_layout)
|
||||
self.cfg_spin = QSpinBox()
|
||||
self.cfg_spin.setRange(1, 30)
|
||||
self.cfg_spin.setValue(7)
|
||||
self.cfg_spin.setPrefix("CFG: ")
|
||||
generate_panel.addWidget(self.cfg_spin)
|
||||
|
||||
# Сэмплер
|
||||
sampler_layout = QVBoxLayout()
|
||||
sampler_label = QLabel("Сэмплер:")
|
||||
sampler_label.setStyleSheet(f"""
|
||||
QLabel {{
|
||||
color: {colors['text']};
|
||||
font-size: 14px;
|
||||
font-weight: 500;
|
||||
}}
|
||||
""")
|
||||
self.sampler_combo = QComboBox()
|
||||
self.sampler_combo.addItems(["Euler a", "Euler", "DPM++ 2M Karras"])
|
||||
sampler_layout.addWidget(sampler_label)
|
||||
sampler_layout.addWidget(self.sampler_combo)
|
||||
params_layout.addLayout(sampler_layout)
|
||||
self.batch_size_spin = QSpinBox()
|
||||
self.batch_size_spin.setRange(1, 4)
|
||||
self.batch_size_spin.setValue(1)
|
||||
self.batch_size_spin.setPrefix("Количество: ")
|
||||
generate_panel.addWidget(self.batch_size_spin)
|
||||
|
||||
# Добавляем параметры на страницу
|
||||
self.layout.addLayout(params_layout)
|
||||
self.generate_btn = QPushButton("Сгенерировать")
|
||||
self.generate_btn.setIcon(self.icon_manager.get_icon(IconType.GENERATE))
|
||||
self.generate_btn.clicked.connect(self.generate_images)
|
||||
generate_panel.addWidget(self.generate_btn)
|
||||
|
||||
# Кнопки управления
|
||||
buttons_layout = QHBoxLayout()
|
||||
buttons_layout.setSpacing(spacing.MEDIUM)
|
||||
generate_panel.addStretch()
|
||||
self.layout.addLayout(generate_panel)
|
||||
|
||||
self.generate_button = QPushButton("Сгенерировать")
|
||||
self.generate_button.setStyleSheet(f"""
|
||||
QPushButton {{
|
||||
min-width: 150px;
|
||||
background-color: {colors['primary']};
|
||||
color: white;
|
||||
border: none;
|
||||
border-radius: {spacing.TINY}px;
|
||||
padding: {spacing.SMALL}px {spacing.MEDIUM}px;
|
||||
}}
|
||||
# Прогресс-бар
|
||||
progress_layout = QHBoxLayout()
|
||||
progress_layout.setContentsMargins(
|
||||
StyleConstants.SPACING['lg'],
|
||||
0,
|
||||
StyleConstants.SPACING['lg'],
|
||||
StyleConstants.SPACING['lg']
|
||||
)
|
||||
|
||||
self.progress_bar = QProgressBar()
|
||||
self.progress_bar.setTextVisible(True)
|
||||
self.progress_bar.setFormat("%p% - %v/%m шагов")
|
||||
progress_layout.addWidget(self.progress_bar)
|
||||
|
||||
self.layout.addLayout(progress_layout)
|
||||
|
||||
# Область для результатов
|
||||
result_layout = QVBoxLayout()
|
||||
result_layout.setContentsMargins(
|
||||
StyleConstants.SPACING['lg'],
|
||||
0,
|
||||
StyleConstants.SPACING['lg'],
|
||||
StyleConstants.SPACING['lg']
|
||||
)
|
||||
|
||||
self.result_label = QLabel()
|
||||
self.result_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
||||
result_layout.addWidget(self.result_label)
|
||||
|
||||
self.layout.addLayout(result_layout)
|
||||
self.layout.addStretch()
|
||||
|
||||
# Обновляем UI
|
||||
self.update_ui()
|
||||
|
||||
def update_ui(self):
|
||||
"""Обновление UI после загрузки данных"""
|
||||
# Обновляем список моделей
|
||||
self.model_combo.clear()
|
||||
for model in self.model_manager.get_models():
|
||||
self.model_combo.addItem(model.title, model.name)
|
||||
|
||||
# Устанавливаем текущую модель
|
||||
current_model = self.model_manager.get_current_model()
|
||||
if current_model:
|
||||
index = self.model_combo.findData(current_model.name)
|
||||
if index >= 0:
|
||||
self.model_combo.setCurrentIndex(index)
|
||||
|
||||
# Обновляем список сэмплеров
|
||||
self.sampler_combo.clear()
|
||||
for sampler in self.sampler_manager.get_samplers():
|
||||
self.sampler_combo.addItem(sampler.title, sampler.name)
|
||||
|
||||
# Устанавливаем текущий сэмплер
|
||||
current_sampler = self.sampler_manager.get_current_sampler()
|
||||
if current_sampler:
|
||||
index = self.sampler_combo.findData(current_sampler.name)
|
||||
if index >= 0:
|
||||
self.sampler_combo.setCurrentIndex(index)
|
||||
|
||||
def on_model_changed(self, model_title):
|
||||
"""Обработка изменения модели"""
|
||||
model_name = self.model_combo.currentData()
|
||||
if model_name:
|
||||
self.model_manager.switch_model(model_name)
|
||||
|
||||
def on_sampler_changed(self, sampler_title):
|
||||
"""Обработка изменения сэмплера"""
|
||||
sampler_name = self.sampler_combo.currentData()
|
||||
if sampler_name:
|
||||
self.sampler_manager.set_current_sampler(sampler_name)
|
||||
|
||||
def set_resolution(self, width: int, height: int):
|
||||
"""Установка разрешения"""
|
||||
self.width_spin.setValue(width)
|
||||
self.height_spin.setValue(height)
|
||||
|
||||
def generate_images(self):
|
||||
"""Генерация изображений"""
|
||||
try:
|
||||
# Получаем параметры
|
||||
params = {
|
||||
"prompt": self.prompt_edit.text(),
|
||||
"negative_prompt": self.negative_prompt_edit.text(),
|
||||
"steps": self.steps_spin.value(),
|
||||
"cfg_scale": self.cfg_spin.value(),
|
||||
"batch_size": self.batch_size_spin.value(),
|
||||
"width": self.width_spin.value(),
|
||||
"height": self.height_spin.value(),
|
||||
"sampler_name": self.sampler_combo.currentData(),
|
||||
}
|
||||
|
||||
QPushButton:hover {{
|
||||
background-color: {colors['hover']};
|
||||
}}
|
||||
# Отправляем запрос
|
||||
response = requests.post(
|
||||
f"{self.model_manager.api_url}/sdapi/v1/txt2img",
|
||||
json=params
|
||||
)
|
||||
|
||||
QPushButton:pressed {{
|
||||
background-color: {colors['active']};
|
||||
}}
|
||||
""")
|
||||
if response.status_code == 200:
|
||||
result = response.json()
|
||||
# TODO: Отобразить сгенерированные изображения
|
||||
print("Изображения успешно сгенерированы")
|
||||
else:
|
||||
print(f"Ошибка генерации: {response.status_code}")
|
||||
|
||||
self.clear_button = QPushButton("Очистить")
|
||||
self.clear_button.setStyleSheet(f"""
|
||||
QPushButton {{
|
||||
min-width: 150px;
|
||||
background-color: transparent;
|
||||
color: {colors['primary']};
|
||||
border: 1px solid {colors['primary']};
|
||||
border-radius: {spacing.TINY}px;
|
||||
padding: {spacing.SMALL}px {spacing.MEDIUM}px;
|
||||
}}
|
||||
|
||||
QPushButton:hover {{
|
||||
background-color: {colors['hover']};
|
||||
color: white;
|
||||
border: none;
|
||||
}}
|
||||
|
||||
QPushButton:pressed {{
|
||||
background-color: {colors['active']};
|
||||
color: white;
|
||||
border: none;
|
||||
}}
|
||||
""")
|
||||
|
||||
buttons_layout.addWidget(self.generate_button)
|
||||
buttons_layout.addWidget(self.clear_button)
|
||||
buttons_layout.addStretch()
|
||||
|
||||
self.layout.addLayout(buttons_layout)
|
||||
|
||||
# Растягиваем пространство
|
||||
self.layout.addStretch()
|
||||
except Exception as e:
|
||||
print(f"Ошибка при генерации: {str(e)}")
|
||||
Reference in new issue
Block a user