|
|
from PIL import Image
|
|
|
from PyQt5.QtWidgets import QApplication
|
|
|
|
|
|
from app_settings import AppSettings
|
|
|
from backend.models.lcmdiffusion_setting import DiffusionTask
|
|
|
from frontend.gui.img2img_widget import Img2ImgWidget
|
|
|
from frontend.webui.image_variations_ui import generate_image_variations
|
|
|
|
|
|
|
|
|
class ImageVariationsWidget(Img2ImgWidget):
|
|
|
def __init__(self, config: AppSettings, parent):
|
|
|
super().__init__(config, parent)
|
|
|
|
|
|
self.prompt.hide()
|
|
|
self.neg_prompt_label.hide()
|
|
|
self.neg_prompt.setEnabled(False)
|
|
|
|
|
|
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.config.settings.lcm_diffusion_setting.negative_prompt = ""
|
|
|
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 = generate_image_variations(
|
|
|
self.config.settings.lcm_diffusion_setting.init_image,
|
|
|
self.config.settings.lcm_diffusion_setting.strength,
|
|
|
)
|
|
|
self.prepare_images(images)
|
|
|
self.after_generation()
|
|
|
|
