MogensR commited on
Commit
dcd56fe
·
1 Parent(s): 873f85d

Update config/app_config.py

Browse files
Files changed (1) hide show
  1. config/app_config.py +12 -0
config/app_config.py CHANGED
@@ -88,6 +88,12 @@ class ProcessingConfig:
88
  mask_edge_softness: int = int(os.getenv('MASK_EDGE_SOFTNESS', '2'))
89
  mask_spill_suppression: float = float(os.getenv('MASK_SPILL_SUPPRESSION', '0.35'))
90
 
 
 
 
 
 
 
91
  # Output settings
92
  output_dir: str = os.getenv('OUTPUT_DIR', 'outputs')
93
  output_format: str = os.getenv('OUTPUT_FORMAT', 'mp4')
@@ -203,6 +209,12 @@ def _validate_config(self):
203
  self.mask_edge_softness = max(0, min(10, self.mask_edge_softness))
204
  self.mask_spill_suppression = max(0.0, min(1.0, self.mask_spill_suppression))
205
 
 
 
 
 
 
 
206
  # Validate output settings
207
  valid_formats = ['mp4', 'avi', 'mov', 'webm', 'mkv']
208
  if self.output_format not in valid_formats:
 
88
  mask_edge_softness: int = int(os.getenv('MASK_EDGE_SOFTNESS', '2'))
89
  mask_spill_suppression: float = float(os.getenv('MASK_SPILL_SUPPRESSION', '0.35'))
90
 
91
+ # Hard cutoff thresholds for mask processing (ADDED)
92
+ hard_low: float = float(os.getenv('HARD_LOW', '0.2'))
93
+ hard_medium: float = float(os.getenv('HARD_MEDIUM', '0.5'))
94
+ hard_high: float = float(os.getenv('HARD_HIGH', '0.8'))
95
+ hard_threshold: float = float(os.getenv('HARD_THRESHOLD', '0.5'))
96
+
97
  # Output settings
98
  output_dir: str = os.getenv('OUTPUT_DIR', 'outputs')
99
  output_format: str = os.getenv('OUTPUT_FORMAT', 'mp4')
 
209
  self.mask_edge_softness = max(0, min(10, self.mask_edge_softness))
210
  self.mask_spill_suppression = max(0.0, min(1.0, self.mask_spill_suppression))
211
 
212
+ # Validate hard threshold settings (ADDED)
213
+ self.hard_low = max(0.0, min(1.0, self.hard_low))
214
+ self.hard_medium = max(0.0, min(1.0, self.hard_medium))
215
+ self.hard_high = max(0.0, min(1.0, self.hard_high))
216
+ self.hard_threshold = max(0.0, min(1.0, self.hard_threshold))
217
+
218
  # Validate output settings
219
  valid_formats = ['mp4', 'avi', 'mov', 'webm', 'mkv']
220
  if self.output_format not in valid_formats: