import torch

NUM_FRAMES = None
FETA_WEIGHT = None
ENABLE_FETA= False

@torch.compiler.disable()
def set_num_frames(num_frames: int):
    global NUM_FRAMES
    NUM_FRAMES = num_frames

@torch.compiler.disable()
def get_num_frames() -> int:
    return NUM_FRAMES


def enable_enhance():
    global ENABLE_FETA
    ENABLE_FETA = True

def disable_enhance():
    global ENABLE_FETA
    ENABLE_FETA = False

@torch.compiler.disable()
def is_enhance_enabled() -> bool:
    return ENABLE_FETA

@torch.compiler.disable()
def set_enhance_weight(feta_weight: float):
    global FETA_WEIGHT
    FETA_WEIGHT = feta_weight

@torch.compiler.disable()
def get_enhance_weight() -> float:
    return FETA_WEIGHT