84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
from .utils import runwareUtils as rwUtils
|
|
|
|
|
|
class vectorize:
|
|
"""Vectorize node for converting images to vector format"""
|
|
|
|
RUNWARE_VECTORIZE_MODELS = {
|
|
"recraft:1@1": "recraft:1@1",
|
|
"picsart:1@1": "picsart:1@1",
|
|
}
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"Image": ("IMAGE", {
|
|
"tooltip": "Input image to be vectorized."
|
|
}),
|
|
"Model": (list(cls.RUNWARE_VECTORIZE_MODELS.keys()), {
|
|
"tooltip": "Select the vectorization model to use.",
|
|
"default": "recraft:1@1",
|
|
}),
|
|
},
|
|
}
|
|
|
|
DESCRIPTION = "Convert images to vector format using Runware's vectorization service."
|
|
FUNCTION = "vectorizeImage"
|
|
RETURN_TYPES = ("STRING",)
|
|
RETURN_NAMES = ("IMAGE",)
|
|
CATEGORY = "Runware"
|
|
|
|
def vectorizeImage(self, **kwargs):
|
|
"""Vectorize image and return SVG URL"""
|
|
image = kwargs.get("Image")
|
|
modelName = kwargs.get("Model", "recraft:1@1")
|
|
|
|
model = self.RUNWARE_VECTORIZE_MODELS.get(modelName, "recraft:1@1")
|
|
imageUuid = rwUtils.convertTensor2IMG(image)
|
|
|
|
genConfig = self._buildGenConfig(model, imageUuid)
|
|
|
|
print(f"[DEBUG] Sending Vectorize Request:")
|
|
print(f"[DEBUG] Request Payload: {rwUtils.safe_json_dumps(genConfig, indent=2)}")
|
|
|
|
genResult = rwUtils.inferenecRequest(genConfig)
|
|
|
|
print(f"[DEBUG] Received Vectorize Response:")
|
|
print(f"[DEBUG] Response: {rwUtils.safe_json_dumps(genResult, indent=2)}")
|
|
|
|
self._validateResponse(genResult)
|
|
|
|
svg_url = rwUtils.extractImageURLs(genResult)
|
|
|
|
return (svg_url,)
|
|
|
|
def _buildGenConfig(self, model, imageUuid):
|
|
"""Build generation configuration for API request"""
|
|
return [{
|
|
"taskType": "vectorize",
|
|
"taskUUID": rwUtils.genRandUUID(),
|
|
"model": model,
|
|
"inputs": {
|
|
"image": imageUuid
|
|
},
|
|
"outputType": "URL",
|
|
"outputFormat": "svg",
|
|
}]
|
|
|
|
def _validateResponse(self, genResult):
|
|
"""Validate API response"""
|
|
if "errors" in genResult:
|
|
errorMessage = genResult["errors"][0]["message"]
|
|
raise Exception(f"Vectorization failed: {errorMessage}")
|
|
|
|
|
|
# Node class mappings
|
|
NODE_CLASS_MAPPINGS = {
|
|
"RunwareVectorize": vectorize,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"RunwareVectorize": "Runware Vectorize",
|
|
}
|