Update config/app_config.py
Browse files- 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:
|