|
|
from PyQt5.QtWidgets import (
|
|
|
QHBoxLayout,
|
|
|
QLabel,
|
|
|
QLineEdit,
|
|
|
QSlider,
|
|
|
QToolButton,
|
|
|
QFileDialog,
|
|
|
QApplication,
|
|
|
)
|
|
|
|
|
|
|
|
|
from PyQt5.QtCore import Qt, QEvent
|
|
|
|
|
|
from PIL import Image
|
|
|
from constants import DEVICE
|
|
|
from app_settings import AppSettings
|
|
|
from urllib.parse import urlparse, unquote
|
|
|
from frontend.gui.base_widget import BaseWidget
|
|
|
from backend.models.lcmdiffusion_setting import DiffusionTask
|
|
|
|
|
|
|
|
|
class Img2ImgWidget(BaseWidget):
|
|
|
def __init__(self, config: AppSettings, parent):
|
|
|
super().__init__(config, parent)
|
|
|
|
|
|
|
|
|
self.img_label = QLabel("Init image:")
|
|
|
self.img_path = QLineEdit()
|
|
|
self.img_path.setReadOnly(True)
|
|
|
self.img_path.setAcceptDrops(True)
|
|
|
self.img_path.installEventFilter(self)
|
|
|
self.img_browse = QToolButton()
|
|
|
self.img_browse.setText("...")
|
|
|
self.img_browse.setToolTip("Browse for an init image")
|
|
|
self.img_browse.clicked.connect(self.browse_click)
|
|
|
|
|
|
hlayout = QHBoxLayout()
|
|
|
hlayout.addWidget(self.img_label)
|
|
|
hlayout.addWidget(self.img_path)
|
|
|
hlayout.addWidget(self.img_browse)
|
|
|
|
|
|
self.strength_label = QLabel("Denoising strength: 0.3")
|
|
|
self.strength = QSlider(orientation=Qt.Orientation.Horizontal)
|
|
|
self.strength.setMaximum(10)
|
|
|
self.strength.setMinimum(1)
|
|
|
self.strength.setValue(3)
|
|
|
self.strength.valueChanged.connect(self.update_strength_label)
|
|
|
|
|
|
|
|
|
self.layout().addLayout(hlayout)
|
|
|
self.layout().addWidget(self.strength_label)
|
|
|
self.layout().addWidget(self.strength)
|
|
|
|
|
|
def browse_click(self):
|
|
|
filename = self.show_file_selection_dialog()
|
|
|
if filename[0] != "":
|
|
|
self.img_path.setText(filename[0])
|
|
|
|
|
|
def show_file_selection_dialog(self) -> str:
|
|
|
filename = QFileDialog.getOpenFileName(
|
|
|
self, "Open Image", "results", "Image Files (*.png *.jpg *.bmp)"
|
|
|
)
|
|
|
return filename
|
|
|
|
|
|
def eventFilter(self, source, event: QEvent):
|
|
|
"""This is the Drag and Drop event filter for the init image QLineEdit"""
|
|
|
if event.type() == QEvent.DragEnter:
|
|
|
if event.mimeData().hasFormat("text/plain"):
|
|
|
event.acceptProposedAction()
|
|
|
return True
|
|
|
elif event.type() == QEvent.Drop:
|
|
|
event.acceptProposedAction()
|
|
|
path = unquote(urlparse(event.mimeData().text()).path)
|
|
|
self.img_path.setText(path)
|
|
|
return True
|
|
|
|
|
|
return False
|
|
|
|
|
|
def before_generation(self):
|
|
|
super().before_generation()
|
|
|
self.img_browse.setEnabled(False)
|
|
|
self.img_path.setEnabled(False)
|
|
|
|
|
|
def after_generation(self):
|
|
|
super().after_generation()
|
|
|
self.img_browse.setEnabled(True)
|
|
|
self.img_path.setEnabled(True)
|
|
|
|
|
|
def generate_image(self):
|
|
|
self.parent.prepare_generation_settings(self.config)
|
|
|
self.config.settings.lcm_diffusion_setting.diffusion_task = (
|
|
|
DiffusionTask.image_to_image.value
|
|
|
)
|
|
|
self.config.settings.lcm_diffusion_setting.prompt = self.prompt.toPlainText()
|
|
|
self.config.settings.lcm_diffusion_setting.negative_prompt = (
|
|
|
self.neg_prompt.toPlainText()
|
|
|
)
|
|
|
self.config.settings.lcm_diffusion_setting.init_image = Image.open(
|
|
|
self.img_path.text()
|
|
|
)
|
|
|
self.config.settings.lcm_diffusion_setting.strength = self.strength.value() / 10
|
|
|
|
|
|
images = self.parent.context.generate_text_to_image(
|
|
|
self.config.settings,
|
|
|
self.config.reshape_required,
|
|
|
DEVICE,
|
|
|
)
|
|
|
self.parent.context.save_images(images, self.config.settings)
|
|
|
self.prepare_images(images)
|
|
|
self.after_generation()
|
|
|
|
|
|
def update_strength_label(self, value):
|
|
|
val = round(int(value) / 10, 1)
|
|
|
self.strength_label.setText(f"Denoising strength: {val}")
|
|
|
self.config.settings.lcm_diffusion_setting.strength = val
|
|
|
|
