Update ui_components.py

ui_components.py

@@ -38,7 +38,7 @@ def create_interface():
     def enhanced_process_video(
         video_path, bg_method, custom_img, prof_choice, 
         use_two_stage, chroma_preset, quality_preset,
-        is_processing_state, frame_progress_state, progress: gr.Progress = None
+        is_processing_state, frame_progress_state, progress=gr.Progress()
     ):
         """
         Handles video processing (start/stop) and live frame/fps reporting.
@@ -113,6 +113,17 @@ def progress_callback(pct, desc, current_frame=None, total_frames=None):
             is_processing_state = False
             return None, f"Error: {str(e)}", "Error during processing.", False, frame_progress_state
 
+    def handle_model_loading(progress=gr.Progress()):
+        """
+        Wrapper function for model loading with proper progress handling
+        """
+        try:
+            # Call the model loading function with progress callback
+            result = load_models_with_validation(progress)
+            return result
+        except Exception as e:
+            return f"Model loading failed: {str(e)}"
+
     # --- UI setup ---
     with gr.Blocks(
         title="BackgroundFX Pro",
@@ -207,8 +218,9 @@ def update_frame_fields(frame_progress):
             show_progress=True
         )
 
+        # Fixed model loading button - let Gradio handle progress automatically
         load_models_btn.click(
-            fn=lambda progress=gr.Progress(): load_models_with_validation(progress),
+            fn=handle_model_loading,
             outputs=[status_text],
             show_progress=True
         )
@@ -216,4 +228,4 @@ def update_frame_fields(frame_progress):
     return demo
 
 def create_ui():
-    return create_interface()
+    return create_interface()