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