from backend.annotators.control_interface import ControlInterface
from controlnet_aux import OpenposeDetector
from PIL import Image


class PoseControl(ControlInterface):
    def get_control_image(self, image: Image) -> Image:
        openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
        image = openpose(image)
        return image