Добавлен модуль готовых промптов: создано окно с категориями промптов, добавлен поиск, реализовано добавление промптов в основное поле ввода
This commit is contained in:
1 parent
29dce906c5
commit
4b16298d05
3 files changed
+223
No files matched your search
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user