| | 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()
|
| |
|
