Update app.py

app.py

@@ -5,40 +5,13 @@
 No inline JavaScript, CSP-compliant Gradio, safe environment vars, fallback AI models
 """
 
+import early_env  # <<< must be FIRST to sanitize threading/env before anything else
+
 import os
 import logging
 from pathlib import Path
 from typing import Optional, Tuple, Dict, Any, Callable
 
-# 0️⃣ Early threading/OMP sanitization (prevents libgomp + interop warnings)
-def _clean_int_env(name: str, default: str | None = None):
-    val = os.environ.get(name)
-    if val is None:
-        if default is not None:
-            os.environ[name] = str(default)
-        return
-    try:
-        int(str(val).strip())
-    except ValueError:
-        if default is None:
-            os.environ.pop(name, None)
-        else:
-            os.environ[name] = str(default)
-
-_clean_int_env("OMP_NUM_THREADS", "2")
-_clean_int_env("MKL_NUM_THREADS", "2")
-_clean_int_env("OPENBLAS_NUM_THREADS", "2")
-_clean_int_env("NUMEXPR_NUM_THREADS", "2")
-
-try:
-    import torch  # call thread setters BEFORE heavy parallel work starts
-    if hasattr(torch, "set_num_interop_threads"):
-        torch.set_num_interop_threads(2)
-    if hasattr(torch, "set_num_threads"):
-        torch.set_num_threads(2)
-except Exception:
-    pass
-
 # 1️⃣ Set CSP-safe environment variables BEFORE any imports (Gradio will see these)
 os.environ['GRADIO_ALLOW_FLAGGING'] = 'never'
 os.environ['GRADIO_ANALYTICS_ENABLED'] = 'False'
@@ -119,10 +92,8 @@ def load_models(self, progress_callback: Optional[Callable]=None) -> str:
         except Exception as e:
             logger.warning(f"Model loading failed ({e}) - Using CSP-safe fallbacks")
             sam2, matanyone = None, None
-        # Unwrap LoadedModel if needed
         sam2_model = getattr(sam2, "model", sam2) if sam2 else CSPSafeSAM2()
         matanyone_model = getattr(matanyone, "model", matanyone) if matanyone else CSPSafeMatAnyone()
-        # Set up Core Processor (replace models if None)
         self.core_processor = CoreVideoProcessor(config=ProcessorConfig(), models=None)
         self.core_processor.models = type('FakeModelManager', (), {
             'get_sam2': lambda self: sam2_model,
@@ -134,14 +105,11 @@ def load_models(self, progress_callback: Optional[Callable]=None) -> str:
     def process_video(self, video, bg_style, custom_bg_file):
         if not self.models_loaded:
             return None, "Models not loaded yet"
-        # Prepare output path
         import time
         output_path = f"/tmp/output_{int(time.time())}.mp4"
-        # Build background config
         cfg = PROFESSIONAL_BACKGROUNDS.get(bg_style, PROFESSIONAL_BACKGROUNDS["minimalist"])
         if custom_bg_file:
             cfg = {"custom_path": custom_bg_file.name}
-        # Validate and process
         ok, msg = validate_video_file(video)
         if not ok:
             return None, f"Invalid video: {msg}"
@@ -151,7 +119,6 @@ def process_video(self, video, bg_style, custom_bg_file):
                 output_path=output_path,
                 bg_config=cfg
             )
-            # Add audio back (optional, comment out if not needed)
             output_with_audio = self.audio_proc.add_audio_to_video(video, output_path)
             return output_with_audio, f"Processing complete ({result.get('frames', 'n/a')} frames, {bg_style})"
         except Exception as e:
@@ -187,7 +154,6 @@ def create_csp_safe_gradio():
                 status = gr.Textbox(label="Status", lines=4)
                 out_video = gr.Video(label="Processed Video")
 
-        # CSP-safe event bindings (no JS eval)
         def safe_load():
             return app.load_models()
         def safe_process(vid, style, custom_bg_file):
@@ -206,5 +172,5 @@ def safe_process(vid, style, custom_bg_file):
         server_port=7860,
         show_error=True,
         debug=False,
-        inbrowser=False  # Never open browser (CSP fails)
+        inbrowser=False
     )