Update config/app_config.py

config/app_config.py

@@ -107,6 +107,60 @@ class ProcessingConfig:
     edge_threshold: float = float(os.getenv('EDGE_THRESHOLD', '0.5'))
     edge_detect_method: str = os.getenv('EDGE_DETECT_METHOD', 'canny')
     
+    # Tracking and temporal consistency attributes (ADDED)
+    min_iou_to_accept: float = float(os.getenv('MIN_IOU_TO_ACCEPT', '0.3'))
+    max_iou_distance: float = float(os.getenv('MAX_IOU_DISTANCE', '0.7'))
+    min_track_length: int = int(os.getenv('MIN_TRACK_LENGTH', '3'))
+    track_buffer_size: int = int(os.getenv('TRACK_BUFFER_SIZE', '30'))
+    temporal_smoothing: float = float(os.getenv('TEMPORAL_SMOOTHING', '0.7'))
+    motion_threshold: float = float(os.getenv('MOTION_THRESHOLD', '0.1'))
+    
+    # Segmentation refinement attributes (ADDED)
+    refine_iterations: int = int(os.getenv('REFINE_ITERATIONS', '1'))
+    refine_threshold: float = float(os.getenv('REFINE_THRESHOLD', '0.5'))
+    use_crf: bool = os.getenv('USE_CRF', 'false').lower() == 'true'
+    crf_iterations: int = int(os.getenv('CRF_ITERATIONS', '5'))
+    bilateral_sigma_color: float = float(os.getenv('BILATERAL_SIGMA_COLOR', '80.0'))
+    bilateral_sigma_space: float = float(os.getenv('BILATERAL_SIGMA_SPACE', '10.0'))
+    
+    # Feathering and blending attributes (ADDED)
+    feather_amount: int = int(os.getenv('FEATHER_AMOUNT', '5'))
+    blend_mode: str = os.getenv('BLEND_MODE', 'normal')
+    alpha_matting: bool = os.getenv('ALPHA_MATTING', 'false').lower() == 'true'
+    alpha_threshold: float = float(os.getenv('ALPHA_THRESHOLD', '0.5'))
+    composite_mode: str = os.getenv('COMPOSITE_MODE', 'over')
+    
+    # Color correction attributes (ADDED)
+    color_correct: bool = os.getenv('COLOR_CORRECT', 'false').lower() == 'true'
+    brightness_adjust: float = float(os.getenv('BRIGHTNESS_ADJUST', '1.0'))
+    contrast_adjust: float = float(os.getenv('CONTRAST_ADJUST', '1.0'))
+    saturation_adjust: float = float(os.getenv('SATURATION_ADJUST', '1.0'))
+    hue_shift: float = float(os.getenv('HUE_SHIFT', '0.0'))
+    
+    # Noise reduction attributes (ADDED)
+    denoise: bool = os.getenv('DENOISE', 'false').lower() == 'true'
+    denoise_strength: float = float(os.getenv('DENOISE_STRENGTH', '10.0'))
+    median_filter_size: int = int(os.getenv('MEDIAN_FILTER_SIZE', '0'))
+    gaussian_sigma: float = float(os.getenv('GAUSSIAN_SIGMA', '1.0'))
+    
+    # Performance optimization attributes (ADDED)
+    use_gpu: bool = os.getenv('USE_GPU', 'true').lower() == 'true'
+    gpu_device_id: int = int(os.getenv('GPU_DEVICE_ID', '0'))
+    use_fp16: bool = os.getenv('USE_FP16', 'false').lower() == 'true'
+    use_tensorrt: bool = os.getenv('USE_TENSORRT', 'false').lower() == 'true'
+    use_quantization: bool = os.getenv('USE_QUANTIZATION', 'false').lower() == 'true'
+    
+    # Video codec attributes (ADDED)
+    use_nvenc: bool = os.getenv('USE_NVENC', 'false').lower() == 'true'
+    prefer_mp4: bool = os.getenv('PREFER_MP4', 'true').lower() == 'true'
+    video_codec: str = os.getenv('VIDEO_CODEC', 'mp4v')
+    audio_copy: bool = os.getenv('AUDIO_COPY', 'true').lower() == 'true'
+    ffmpeg_path: str = os.getenv('FFMPEG_PATH', 'ffmpeg')
+    
+    # Model size constraints (ADDED)
+    max_model_size: int = int(os.getenv('MAX_MODEL_SIZE', '0'))
+    max_model_size_bytes: int = int(os.getenv('MAX_MODEL_SIZE_BYTES', '0'))
+    
     # Output settings
     output_dir: str = os.getenv('OUTPUT_DIR', 'outputs')
     output_format: str = os.getenv('OUTPUT_FORMAT', 'mp4')
@@ -243,6 +297,52 @@ def _validate_config(self):
         if self.edge_detect_method not in valid_edge_methods:
             self.edge_detect_method = 'canny'
         
+        # Validate tracking and temporal consistency (ADDED)
+        self.min_iou_to_accept = max(0.0, min(1.0, self.min_iou_to_accept))
+        self.max_iou_distance = max(0.0, min(1.0, self.max_iou_distance))
+        self.min_track_length = max(1, min(100, self.min_track_length))
+        self.track_buffer_size = max(1, min(300, self.track_buffer_size))
+        self.temporal_smoothing = max(0.0, min(1.0, self.temporal_smoothing))
+        self.motion_threshold = max(0.0, min(1.0, self.motion_threshold))
+        
+        # Validate segmentation refinement (ADDED)
+        self.refine_iterations = max(0, min(10, self.refine_iterations))
+        self.refine_threshold = max(0.0, min(1.0, self.refine_threshold))
+        self.crf_iterations = max(1, min(20, self.crf_iterations))
+        self.bilateral_sigma_color = max(1.0, min(200.0, self.bilateral_sigma_color))
+        self.bilateral_sigma_space = max(1.0, min(50.0, self.bilateral_sigma_space))
+        
+        # Validate feathering and blending (ADDED)
+        self.feather_amount = max(0, min(50, self.feather_amount))
+        valid_blend_modes = ['normal', 'multiply', 'screen', 'overlay', 'soft_light', 'hard_light']
+        if self.blend_mode not in valid_blend_modes:
+            self.blend_mode = 'normal'
+        self.alpha_threshold = max(0.0, min(1.0, self.alpha_threshold))
+        valid_composite_modes = ['over', 'under', 'atop', 'xor', 'plus']
+        if self.composite_mode not in valid_composite_modes:
+            self.composite_mode = 'over'
+        
+        # Validate color correction (ADDED)
+        self.brightness_adjust = max(0.0, min(2.0, self.brightness_adjust))
+        self.contrast_adjust = max(0.0, min(2.0, self.contrast_adjust))
+        self.saturation_adjust = max(0.0, min(2.0, self.saturation_adjust))
+        self.hue_shift = max(-180.0, min(180.0, self.hue_shift))
+        
+        # Validate noise reduction (ADDED)
+        self.denoise_strength = max(0.0, min(100.0, self.denoise_strength))
+        self.median_filter_size = max(0, min(15, self.median_filter_size))
+        if self.median_filter_size % 2 == 0 and self.median_filter_size > 0:
+            self.median_filter_size += 1  # Must be odd
+        self.gaussian_sigma = max(0.0, min(10.0, self.gaussian_sigma))
+        
+        # Validate performance optimization (ADDED)
+        self.gpu_device_id = max(0, self.gpu_device_id)
+        
+        # Validate video codec (ADDED)
+        valid_codecs = ['mp4v', 'h264', 'h265', 'xvid', 'mjpeg']
+        if self.video_codec not in valid_codecs:
+            self.video_codec = 'mp4v'
+        
         # Validate output settings
         valid_formats = ['mp4', 'avi', 'mov', 'webm', 'mkv']
         if self.output_format not in valid_formats: