diffsynth/processors/sequencial_processor.py

from .base import VideoProcessor


class AutoVideoProcessor(VideoProcessor):
    def __init__(self):
        pass

    @staticmethod
    def from_model_manager(model_manager, processor_type, **kwargs):
        if processor_type == "FastBlend":
            from .FastBlend import FastBlendSmoother
            return FastBlendSmoother.from_model_manager(model_manager, **kwargs)
        elif processor_type == "Contrast":
            from .PILEditor import ContrastEditor
            return ContrastEditor.from_model_manager(model_manager, **kwargs)
        elif processor_type == "Sharpness":
            from .PILEditor import SharpnessEditor
            return SharpnessEditor.from_model_manager(model_manager, **kwargs)
        elif processor_type == "RIFE":
            from .RIFE import RIFESmoother
            return RIFESmoother.from_model_manager(model_manager, **kwargs)
        else:
            raise ValueError(f"invalid processor_type: {processor_type}")


class SequencialProcessor(VideoProcessor):
    def __init__(self, processors=[]):
        self.processors = processors

    @staticmethod
    def from_model_manager(model_manager, configs):
        processors = [
            AutoVideoProcessor.from_model_manager(model_manager, config["processor_type"], **config["config"])
            for config in configs
        ]
        return SequencialProcessor(processors)
    
    def __call__(self, rendered_frames, **kwargs):
        for processor in self.processors:
            rendered_frames = processor(rendered_frames, **kwargs)
        return rendered_frames