Update ui.py

ui.py

@@ -2,9 +2,7 @@
 """
 Modern UI for Video Background Replacer (PRO)
 - Clean, responsive Streamlit layout
-- File uploaders, previews, and processing controls
-- Supports: Color, Image upload, Stock image, AI Prompt background
-- Log viewer, session state, robust error handling
+- Shows live stage progress and all errors during processing
 """
 import streamlit as st
 import os
@@ -164,11 +162,17 @@ def render_ui(process_video_func):
         with col2:
             background, bg_type = _render_background_settings()
             st.header("3. Process Video")
+            status_box = st.empty()  # <--- Status/progress area
             can_process = (
                 uploaded_video is not None
                 and not st.session_state.get('processing', False)
                 and (background is not None)
             )
+
+            def progress_callback(msg):
+                status_box.info(msg)
+                logger.info(f"[STAGE] {msg}")
+
             if st.button("Process Video", disabled=not can_process, use_container_width=True):
                 try:
                     logger.info("Process Video button clicked")
@@ -177,13 +181,16 @@ def render_ui(process_video_func):
                         return
                     with st.spinner("Processing video..."):
                         # bg_type always lowercase for pipeline
-                        success = process_video_func(uploaded_video, background, bg_type.lower())
+                        success = process_video_func(uploaded_video, background, bg_type.lower(), progress_callback)
                     if success:
+                        status_box.success("✅ Video processing complete!")
                         st.success("Video processing complete!")
                     else:
+                        status_box.error("❌ Video processing failed. Check logs for details.")
                         st.error("Video processing failed. Check logs for details.")
                 except Exception as e:
                     logger.error(f"[UI] Process video error: {e}", exc_info=True)
+                    status_box.error(f"Processing error: {str(e)}. Check logs for details.")
                     st.error(f"Processing error: {str(e)}. Check logs for details.")
 
         # --- Results Display ---