from typing import Dict, List, Any
from diffusers import ControlNetModel, StableDiffusionControlNetPipeline, StableDiffusionPipeline, DiffusionPipeline


class EndpointHandler():
    def __init__(self, path=""):
        self.pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
        self.pipeline.load_lora_weights(".", weight_name="/repository/safetensors.safetensors")
        print('hiiiii')

    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
        print('data:', data)
        input = data['inputs']
        print('input:', input)
        result = self.pipeline(input)
        print('result:', result)
        return result