another fix

ui.py

@@ -5,13 +5,10 @@
 - Compatible with pipeline.process()
 - Aligned with torch==2.3.1+cu121, MatAnyone v1.0.0, SAM2 commit 3c76f73c1a7e7b4a2e8a0a9a3e5b92f7e6e3f2f5
 
-Changes (2025-09-16):
-- Aligned with updated pipeline.py and models/
-- Updated progress callback to pass percentages to pipeline.process
-- Added input path validation logging
-- Simplified SAM2 arguments to use pipeline defaults
-- Added MatAnyone version logging in GPU diagnostics
-- Enhanced temporary file cleanup
+Changes (2025-09-17):
+- Fixed progress callback with comprehensive exception handling
+- Added debug logging around pipeline call to identify hanging issues
+- Enhanced error recovery to prevent silent failures
 """
 
 import io
@@ -176,6 +173,7 @@ def process_video_with_background_stoppable(
 ):
     import importlib
     try:
+        logger.info("=== UI CALLBACK STARTED ===")
         STOP.stop = False
         yield gr.update(visible=False), gr.update(visible=True), None, "Starting…"
 
@@ -318,56 +316,84 @@ def process_video_with_background_stoppable(
         }
 
         def format_status():
-            return (gpu_status + f"\n\nPROCESSING: {stage_status['current_stage']}\n\n" +
-                   f"PIPELINE STAGES:\n" +
-                   f"SAM2 Segmentation: {stage_status['sam2_status']}\n" +
-                   f"MatAnyone Matting: {stage_status['matany_status']}\n" +
-                   f"Video Compositing: {stage_status['composite_status']}\n" +
-                   f"Audio Muxing: {stage_status['audio_status']}\n" +
-                   (f"\n{stage_status['frame_progress']}" if stage_status['frame_progress'] else ""))
+            try:
+                return (gpu_status + f"\n\nPROCESSING: {stage_status['current_stage']}\n\n" +
+                       f"PIPELINE STAGES:\n" +
+                       f"SAM2 Segmentation: {stage_status['sam2_status']}\n" +
+                       f"MatAnyone Matting: {stage_status['matany_status']}\n" +
+                       f"Video Compositing: {stage_status['composite_status']}\n" +
+                       f"Audio Muxing: {stage_status['audio_status']}\n" +
+                       (f"\n{stage_status['frame_progress']}" if stage_status['frame_progress'] else ""))
+            except Exception as e:
+                logger.error(f"Format status error: {e}")
+                return f"Processing: {stage_status.get('current_stage', 'Unknown')}"
 
         processing_status = format_status()
         yield gr.update(visible=False), gr.update(visible=True), None, processing_status
 
-        # Create progress callback to update UI status with detailed tracking
+        # Create progress callback with comprehensive exception handling
         def progress_callback(pct: float, msg: str):
-            nonlocal stage_status
-
-            # Update current stage and frame progress
-            stage_status['current_stage'] = msg
+            try:
+                nonlocal stage_status
+                logger.debug(f"Progress callback called: {pct:.1%} - {msg}")
 
-            # Track specific stages
-            if "SAM2" in msg or "segmentation" in msg.lower():
-                if "complete" in msg.lower() or "Complete" in msg:
-                    stage_status['sam2_status'] = "Complete"
-                else:
-                    stage_status['sam2_status'] = "Running..."
-            elif "MatAnyone" in msg or "matting" in msg.lower():
-                if "complete" in msg.lower() or "Complete" in msg:
-                    stage_status['matany_status'] = "Complete"
-                elif "failed" in msg.lower() or "fallback" in msg.lower():
-                    stage_status['matany_status'] = "Failed -> Fallback"
-                else:
-                    stage_status['matany_status'] = "Running..."
-            elif "composit" in msg.lower():
-                if "complete" in msg.lower() or "Complete" in msg:
-                    stage_status['composite_status'] = "Complete"
-                else:
-                    stage_status['composite_status'] = "Running..."
-            elif "audio" in msg.lower() or "mux" in msg.lower():
-                if "complete" in msg.lower() or "Complete" in msg:
-                    stage_status['audio_status'] = "Complete"
-                else:
-                    stage_status['audio_status'] = "Running..."
+                # Update current stage and frame progress
+                stage_status['current_stage'] = msg
 
-            # Extract frame progress
-            if "/" in msg and any(word in msg.lower() for word in ["frame", "matting", "chunking"]):
-                stage_status['frame_progress'] = msg
+                # Track specific stages
+                if "SAM2" in msg or "segmentation" in msg.lower():
+                    if "complete" in msg.lower() or "Complete" in msg:
+                        stage_status['sam2_status'] = "Complete"
+                    else:
+                        stage_status['sam2_status'] = "Running..."
+                elif "MatAnyone" in msg or "matting" in msg.lower():
+                    if "complete" in msg.lower() or "Complete" in msg:
+                        stage_status['matany_status'] = "Complete"
+                    elif "failed" in msg.lower() or "fallback" in msg.lower():
+                        stage_status['matany_status'] = "Failed -> Fallback"
+                    else:
+                        stage_status['matany_status'] = "Running..."
+                elif "composit" in msg.lower():
+                    if "complete" in msg.lower() or "Complete" in msg:
+                        stage_status['composite_status'] = "Complete"
+                    else:
+                        stage_status['composite_status'] = "Running..."
+                elif "audio" in msg.lower() or "mux" in msg.lower():
+                    if "complete" in msg.lower() or "Complete" in msg:
+                        stage_status['audio_status'] = "Complete"
+                    else:
+                        stage_status['audio_status'] = "Running..."
 
-            updated_status = format_status()
-            return gr.update(visible=False), gr.update(visible=True), None, updated_status
+                # Extract frame progress
+                if "/" in msg and any(word in msg.lower() for word in ["frame", "matting", "chunking"]):
+                    stage_status['frame_progress'] = msg
 
+                updated_status = format_status()
+                logger.debug(f"Progress callback returning status update")
+                return gr.update(visible=False), gr.update(visible=True), None, updated_status
+            
+            except Exception as e:
+                logger.error(f"Progress callback error: {e}")
+                import traceback
+                logger.error(f"Progress callback traceback: {traceback.format_exc()}")
+                
+                # Don't let progress callback errors crash the pipeline
+                # Return a simple status update instead
+                try:
+                    simple_status = gpu_status + f"\n\nProcessing: {msg}"
+                    return gr.update(visible=False), gr.update(visible=True), None, simple_status
+                except Exception as e2:
+                    logger.error(f"Even simple progress update failed: {e2}")
+                    # If even that fails, return nothing to avoid crashing
+                    return None, None, None, None
+
+        # Add extensive debugging around the pipeline call
+        logger.info(f"=== ABOUT TO CALL PIPELINE.PROCESS ===")
+        logger.info(f"Arguments: video_path={video_path}, bg_path={bg_path}")
+        logger.info(f"Pipeline process function exists: {hasattr(pipe, 'process')}")
+        
         try:
+            logger.info(f"=== CALLING pipe.process() NOW ===")
             out_path, diag = pipe.process(
                 video_path=video_path,
                 bg_image_path=bg_path,
@@ -456,8 +482,10 @@ def progress_callback(pct: float, msg: str):
             return
 
         if out_path:
+            logger.info(f"=== UI CALLBACK RETURNING SUCCESS: {out_path} ===")
             yield gr.update(visible=True), gr.update(visible=False), out_path, status_msg
         else:
+            logger.info(f"=== UI CALLBACK RETURNING FAILURE ===")
             yield gr.update(visible=True), gr.update(visible=False), None, status_msg
 
     except Exception as e: