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)
        # Hide prompt and negative prompt widgets
        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()