editings/editor.py

from .interfacegan import InterFaceGAN
from .ganspace import GanSpace
from .styleclip import StyleClip
from options import Settings

"""
Entry class for all the edits. 
"""
class Editor():
    def __init__(self) -> None:
        self.interfacegan_editor = InterFaceGAN()
        self.ganspace_editor = GanSpace()
        self.styleclip_editor = StyleClip()

    def edit(self, latent, cfg):
        # Finds the corresponding function using method name
        if cfg.method == 'inversion':
            return latent

        editor = getattr(self, f'{cfg.method}_editor')
        return editor.edit(latent, cfg)