from typing import List from PIL import Image import numpy as np class DesignModel: """Interface for the design model""" def __init__(self): self.seed = None self.neg_prompt = None self.additional_quality_suffix = None def generate_design(self, image: Image.Image, num_variations: int = 1, **kwargs) -> List[np.ndarray]: """Generate design variations from the input image""" raise NotImplementedError("This method should be implemented by concrete model classes")