Request for Image + Mask Handling Support about cog-comfyui

[fofr]

Do you have a ComfyUI workflow for this?

[Six6pounder]

def handle_input_file(self, input_file: Path, isMaskFile: bool = False):
...
        if isMaskFile:

                shutil.copy(input_file, os.path.join(INPUT_DIR, f"mask{file_extension}"))
                
                
def predict(
        self,
        mask_file: Path = Input(
            description="Mask image, tar or zip file. Read guidance on workflows and input files here: https://github.com/fofr/cog-comfyui. Alternatively, you can replace inputs with URLs in your JSON workflow and the model will download them.",
            default=None,
        ),
        ....
       if mask_file:
            self.handle_input_file(mask_file, isMaskFile=True)