from .utils import runwareUtils as rwUtils

class ipAdapter:
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "Reference Image": ("IMAGE",),
                "Model": ([
                    "runware:55@1 (IP Adapter SDXL)",
                    "runware:55@2 (IP Adapter SDXL Plus)",
                    "runware:55@3 (IP Adapter SDXL Plus Face)",
                    "runware:55@4 (IP Adapter SDXL Vit-H)",
                    "runware:55@5 (IP Adapter SD 1.5)",
                    "runware:55@6 (IP Adapter SD 1.5 Plus)",
                    "runware:55@7 (IP Adapter SD 1.5 Light)",
                    "runware:55@8 (IP Adapter SD 1.5 Plus Face)",
                    "runware:55@9 (IP Adapter SD 1.5 Full Face)",
                    "runware:55@10 (IP Adapter SD 1.5 Vit-G)"
                ], {
                    "tooltip": "Choose IP Adapter model to use for reference-based image generation.",
                    "default": "runware:55@1 (IP Adapter SDXL)",
                }),
                "weight": ("FLOAT", {
                    "default": 1.0,
                    "min": 0.0,
                    "max": 1.0,
                    "step": 0.1,
                    "tooltip": "Represents the strength or influence of this IP-Adapter in the generation process.\nA value of 0 means no influence, while 1 means maximum influence.",
                }),
            },
        }

    RETURN_TYPES = ("RUNWAREIPAdapter",)
    RETURN_NAMES = ("IPAdapter",)
    FUNCTION = "ipAdapter"
    CATEGORY = "Runware"
    DESCRIPTION = "IP-Adapters enable image-prompted generation, allowing you to use reference images to guide the style and content of your generations. Multiple IP Adapters can be used simultaneously."

    def ipAdapter(self, **kwargs):
        refImage = kwargs.get("Reference Image")
        Model = kwargs.get("Model")
        weight = kwargs.get("weight")
        modelAirCode = Model.split(" ")[0]
        guideImage = rwUtils.convertTensor2IMG(refImage)
        return ({
            "model": modelAirCode,
            "guideImage": guideImage,
            "weight": round(weight, 2),
        },)