Добавлен модуль готовых промптов: создано окно с категориями промптов, добавлен поиск, реализовано добавление промптов в основное поле ввода

This commit is contained in:
kromskii2 committed 2025-03-14 19:34:01 +07:00
1 parent 29dce906c5
commit 4b16298d05
3 files changed
+223

No files matched your search

+23
View File
@@ -16,6 +16,7 @@ from io import BytesIO
from modules.console_window import ConsoleWindow
from modules.model_manager import ModelManager
from modules.sampler_manager import SamplerManager
from modules.prompt_window import PromptWindow
class GenerationThread(QThread):
progress_updated = pyqtSignal(float, str)
@@ -62,6 +63,7 @@ class MainWindow(QMainWindow):
super().__init__()
self.api_url = "http://localhost:7860" # URL по умолчанию
self.console_window = None
self.prompt_window = None
self.progress_timer = None
self.init_ui()
@@ -240,6 +242,11 @@ class MainWindow(QMainWindow):
self.console_button.clicked.connect(self.toggle_console)
settings_layout.addWidget(self.console_button)
# Добавляем кнопку промптов
self.prompt_button = QPushButton("Готовые промпты")
self.prompt_button.clicked.connect(self.toggle_prompt_window)
settings_layout.addWidget(self.prompt_button)
# Кнопка генерации
self.generate_button = QPushButton("Сгенерировать")
self.generate_button.clicked.connect(self.start_generation)
@@ -294,6 +301,22 @@ class MainWindow(QMainWindow):
self.console_window.log(message)
print(message)
def toggle_prompt_window(self):
"""Открыть/закрыть окно с промптами"""
if self.prompt_window is None:
self.prompt_window = PromptWindow(parent=self)
self.prompt_window.prompt_selected.connect(self.add_prompt)
self.prompt_window.show()
else:
self.prompt_window.close()
def add_prompt(self, prompt):
"""Добавить промпт в поле ввода"""
current_text = self.prompt_input.toPlainText()
if current_text:
current_text += ", "
self.prompt_input.setText(current_text + prompt)
def start_generation(self):
self.generate_button.setEnabled(False)
self.progress_bar.setValue(0)