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