Update core/app.py

core/app.py

@@ -5,42 +5,15 @@
 """
 from __future__ import annotations
 
-# ── Early env/threading hygiene (safe default to silence libgomp) ────────────
+# ── Critical environment defaults (set before any imports) ────────────────
 import os
-os.environ["OMP_NUM_THREADS"] = "1"  # Force valid value early
-
-# --- Stable runtime env defaults and optional secret-name mapping ---
-def _map_env(src: str, dst: str):
-    v = os.getenv(src)
-    if v and not os.getenv(dst):
-        os.environ[dst] = v
-
-# If HF blocks underscores in secret names, you can add simple names and map them:
-# Add secrets later like: OMPNUMTHREADS=1, TOKENIZERSPARALLELISM=false,
-# PYTORCHCUDAALLOCCONF=max_split_size_mb:128, LOGLEVEL=info, APPENV=production
-_map_env("OMPNUMTHREADS", "OMP_NUM_THREADS")
-_map_env("TOKENIZERSPARALLELISM", "TOKENIZERS_PARALLELISM")
-_map_env("PYTORCHCUDAALLOCCONF", "PYTORCH_CUDA_ALLOC_CONF")
-_map_env("LOGLEVEL", "LOG_LEVEL")
-_map_env("APPENV", "APP_ENV")
-
-# Provider secret mappings (underscore-free -> standard)
-_map_env("CLOUDINARYURL", "CLOUDINARY_URL")
-_map_env("HEYGENAPIKEY", "HEYGEN_API_KEY")
-_map_env("DATABASEURL", "DATABASE_URL")
-
-# Cache path mappings (underscore-free -> standard)
-_map_env("HFHOME", "HF_HOME")
-_map_env("TRANSFORMERSCACHE", "TRANSFORMERS_CACHE")
-_map_env("TORCHHOME", "TORCH_HOME")
-# Optional: GPU visibility alias
-_map_env("CUDAVISIBLEDEVICES", "CUDA_VISIBLE_DEVICES")
-
-# Critical defaults (safe even if already set elsewhere)
+
+# Set critical defaults directly - HF now supports underscores in env vars
+os.environ.setdefault("OMP_NUM_THREADS", "1")
 os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
+os.environ.setdefault("PYTORCH_CUDA_ALLOC_CONF", "max_split_size_mb:128")
 os.environ.setdefault("LOG_LEVEL", "info")
 os.environ.setdefault("APP_ENV", "production")
-os.environ.setdefault("PYTORCH_CUDA_ALLOC_CONF", "max_split_size_mb:128")
 
 # Ensure reasonable cache defaults if not set
 from pathlib import Path
@@ -74,10 +47,6 @@ def _ensure_cloudinary_url():
 import time
 from typing import Optional, Tuple, Dict, Any, Callable
 
-# Mitigate CUDA fragmentation (must be set before importing torch)
-if "PYTORCH_CUDA_ALLOC_CONF" not in os.environ:
-    os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True,max_split_size_mb:128"
-
 # ── Logging ──────────────────────────────────────────────────────────────────
 logging.basicConfig(
     level=logging.INFO,
@@ -462,7 +431,7 @@ def _process_single_stage(
             output_path=out_path,
             bg_config={
                 "background_choice": background_choice,
-                "custom_path": { "path": custom_background_path } if isinstance(custom_background_path, dict) else custom_background_path,
+                "custom_path": custom_background_path,
             },
             progress_callback=progress_callback,
         )