model.py

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