Update app.py

app.py

@@ -5,7 +5,6 @@
 NEW: Added automatic device detection for Hugging Face Spaces compatibility,
      improved error handling, and better resource management
 """
-import sys
 import cv2
 import numpy as np
 from pathlib import Path
@@ -18,6 +17,7 @@
 from typing import Optional, Tuple, Dict, Any
 import logging
 from huggingface_hub import hf_hub_download
+import os
 
 # Import utilities - CRITICAL: Use these functions, don't duplicate!
 from utilities import (
@@ -61,7 +61,7 @@
 models_loaded = False
 loading_lock = threading.Lock()
 two_stage_processor = None
-PROCESS_CANCELLED = False
+PROCESS_CANCELLED = threading.Event()
 
 # ============================================================================ #
 # DEVICE DETECTION FOR HUGGING FACE SPACES
@@ -115,7 +115,7 @@ def _prog(pct: float, desc: str):
 
         # Download checkpoint with caching
         checkpoint_path = hf_hub_download(
-            repo_id="facebook/sam2-hiera-large",
+            repo_id="facebookresearch/sam2",
             filename="sam2_hiera_large.pt",
             cache_dir=str(CACHE_DIR / "sam2_checkpoint"),
             force_download=False
@@ -223,11 +223,11 @@ def load_models_with_validation(progress_callback: Optional[callable] = None) ->
     global sam2_predictor, matanyone_model, models_loaded, two_stage_processor, PROCESS_CANCELLED
 
     with loading_lock:
-        if models_loaded and not PROCESS_CANCELLED:
+        if models_loaded and not PROCESS_CANCELLED.is_set():
             return "Models already loaded and validated"
 
         try:
-            PROCESS_CANCELLED = False
+            PROCESS_CANCELLED.clear()
             start_time = time.time()
             logger.info(f"Starting model loading on {DEVICE}")
 
@@ -237,13 +237,13 @@ def load_models_with_validation(progress_callback: Optional[callable] = None) ->
             # Load SAM2 with validation
             sam2_predictor = load_sam2_predictor_fixed(device=DEVICE, progress_callback=progress_callback)
 
-            if PROCESS_CANCELLED:
+            if PROCESS_CANCELLED.is_set():
                 return "Model loading cancelled by user"
 
             # Load MatAnyone with validation
             matanyone_model = load_matanyone_fixed(progress_callback=progress_callback)
 
-            if PROCESS_CANCELLED:
+            if PROCESS_CANCELLED.is_set():
                 return "Model loading cancelled by user"
 
             models_loaded = True
@@ -282,7 +282,7 @@ def process_video_fixed(
     """Optimized video processing using proper functions from utilities"""
     global PROCESS_CANCELLED
 
-    if PROCESS_CANCELLED:
+    if PROCESS_CANCELLED.is_set():
         return None, "Processing cancelled by user"
 
     if not models_loaded:
@@ -297,7 +297,7 @@ def process_video_fixed(
         return None, f"Invalid video: {validation_msg}"
 
     def _prog(pct: float, desc: str):
-        if PROCESS_CANCELLED:
+        if PROCESS_CANCELLED.is_set():
             raise Exception("Processing cancelled by user")
 
         if progress_callback:
@@ -376,7 +376,7 @@ def _prog(pct: float, desc: str):
                 progress_callback=_prog
             )
 
-            if PROCESS_CANCELLED:
+            if PROCESS_CANCELLED.is_set():
                 return None, "Processing cancelled by user"
 
             if result is None:
@@ -391,7 +391,7 @@ def _prog(pct: float, desc: str):
                     f'ffmpeg -y -i "{final_output}" -i "{video_path}" '
                     f'-c:v libx264 -crf 18 -preset medium '
                     f'-c:a aac -b:a 192k -ac 2 -ar 48000 '
-                    f'-map 0:v:0 -map 1:a:0? -shortest "{final_with_audio}"'
+                    f'-map 0:v:0 -map 1:a:0? -shortest "{final_with_audio}" -loglevel error'
                 )
                 os.system(audio_cmd)
                 if os.path.exists(final_with_audio):
@@ -479,7 +479,7 @@ def _prog(pct: float, desc: str):
         start_time = time.time()
 
         while True:
-            if PROCESS_CANCELLED:
+            if PROCESS_CANCELLED.is_set():
                 cap.release()
                 final_writer.release()
                 if os.path.exists(final_path):
@@ -568,7 +568,7 @@ def _prog(pct: float, desc: str):
         cap.release()
         final_writer.release()
 
-        if PROCESS_CANCELLED:
+        if PROCESS_CANCELLED.is_set():
             if os.path.exists(final_path):
                 os.remove(final_path)
             return None, "Processing cancelled by user"
@@ -594,7 +594,7 @@ def _prog(pct: float, desc: str):
                     f'ffmpeg -y -i "{final_path}" -i "{video_path}" '
                     f'-c:v libx264 -crf 18 -preset medium '
                     f'-c:a aac -b:a 192k -ac 2 -ar 48000 '
-                    f'-map 0:v:0 -map 1:a:0? -shortest "{final_output}"'
+                    f'-map 0:v:0 -map 1:a:0? -shortest "{final_output}" -loglevel error'
                 )
                 result = os.system(audio_cmd)
                 if result != 0 or not os.path.exists(final_output):
@@ -667,4 +667,4 @@ def main():
         print(f"Startup failed: {e}")
 
 if __name__ == "__main__":
-    main()
+    main()