Update ui_components.py

ui_components.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 """
-FIXED UI Components for Video Background Replacement
-Updated with enhanced debug logging for two-stage checkbox issue
+Final UI Components for Video Background Replacement
+Removed non-working timer code, kept progress display functionality
 """
 
 import gradio as gr
@@ -13,7 +13,7 @@
 import traceback
 from typing import Optional
 
-# Import FIXED core processing functions
+# Import core processing functions
 from app import load_models_with_validation, process_video_fixed, get_model_status, get_cache_status
 from utilities import PROFESSIONAL_BACKGROUNDS, create_procedural_background
 
@@ -106,22 +106,10 @@ def on_two_stage_change(value):
 def get_progress_info():
     """Read progress info from temporary file"""
     try:
-        with open("/tmp/video_progress.txt", 'r') as f:
+        with open("/tmp/processing_info.txt", 'r') as f:
             return f.read()
     except:
-        return ""
-
-# ============================================================================ #
-# PROGRESS WRAPPER
-# ============================================================================ #
-def gradio_progress_wrapper(progress_obj, pct, desc):
-    """Wrapper to safely call Gradio progress"""
-    try:
-        if progress_obj is not None:
-            progress_obj(pct, desc=desc)
-            add_debug_log(f"Progress: {pct:.1%} - {desc}")
-    except Exception:
-        pass
+        return "Processing..."
 
 # ============================================================================ #
 # MAIN PROCESSING FUNCTION WITH ENHANCED ERROR HANDLING
@@ -137,7 +125,7 @@ def process_video_enhanced_fixed(
         add_debug_log("ERROR: No video file provided")
         return None, "No video file provided.", get_debug_log()
     
-    # CRITICAL DEBUG: Log exact checkbox value received
+    # Log checkbox value for debugging
     add_debug_log(f"CHECKBOX RAW VALUE: use_two_stage = {use_two_stage}")
     add_debug_log(f"CHECKBOX TYPE: {type(use_two_stage)}")
     add_debug_log(f"CHECKBOX BOOL CONVERSION: {bool(use_two_stage)}")
@@ -146,20 +134,15 @@ def process_video_enhanced_fixed(
     use_two_stage = bool(use_two_stage)
     add_debug_log(f"FINAL use_two_stage value: {use_two_stage}")
     
-    # Store progress info for display
-    progress_info = {"message": "Starting..."}
-    
     def progress_callback(pct, desc):
-        # Update progress info dictionary
-        progress_info["message"] = desc
-        # Try to update Gradio progress (percentage only)
+        # Update Gradio progress bar
         if progress:
             try:
                 progress(pct)
             except:
                 pass
-        # Update result_text with detailed progress
-        return desc
+        # Log progress
+        add_debug_log(f"Progress: {pct:.1%} - {desc}")
     
     try:
         # Clear old logs for new processing
@@ -173,6 +156,9 @@ def progress_callback(pct, desc):
         mode_msg = "TWO-STAGE Green Screen" if use_two_stage else "Single-Stage"
         add_debug_log(f"Processing mode: {mode_msg}")
         
+        # Initial status for Processing Info box
+        initial_status = "🎬 PROCESSING STARTED\n━━━━━━━━━━━━━━━━━━━\n⏳ Initializing...\nPlease wait..."
+        
         if bg_method == "upload":
             if custom_img and os.path.exists(custom_img):
                 add_debug_log(f"Using uploaded image: {custom_img}")
@@ -182,7 +168,9 @@ def progress_callback(pct, desc):
                     use_two_stage=use_two_stage,
                     chroma_preset=chroma_preset
                 )
-                return result[0], result[1], get_debug_log()
+                # Check for progress info file to display
+                final_info = get_progress_info()
+                return result[0], result[1] if result[1] else final_info, get_debug_log()
             return None, "No image uploaded. Please upload a background image.", get_debug_log()
             
         elif bg_method == "professional":
@@ -194,7 +182,8 @@ def progress_callback(pct, desc):
                     use_two_stage=use_two_stage,
                     chroma_preset=chroma_preset
                 )
-                return result[0], result[1], get_debug_log()
+                final_info = get_progress_info()
+                return result[0], result[1] if result[1] else final_info, get_debug_log()
             return None, f"Invalid professional background: {prof_choice}", get_debug_log()
             
         elif bg_method == "colors":
@@ -221,7 +210,8 @@ def progress_callback(pct, desc):
                     use_two_stage=use_two_stage,
                     chroma_preset=chroma_preset
                 )
-                return result[0], result[1], get_debug_log()
+                final_info = get_progress_info()
+                return result[0], result[1] if result[1] else final_info, get_debug_log()
             except Exception as e:
                 error_msg = f"Error creating gradient: {str(e)}"
                 add_debug_log(f"ERROR: {error_msg}")
@@ -236,7 +226,8 @@ def progress_callback(pct, desc):
                     use_two_stage=use_two_stage,
                     chroma_preset=chroma_preset
                 )
-                return result[0], result[1], get_debug_log()
+                final_info = get_progress_info()
+                return result[0], result[1] if result[1] else final_info, get_debug_log()
             return None, "No AI background generated. Click 'Generate Background' first.", get_debug_log()
             
         else:
@@ -250,13 +241,25 @@ def progress_callback(pct, desc):
 def load_models_with_progress_fixed(progress: Optional[gr.Progress] = None):
     """Model loading with debug logging"""
     def progress_callback(pct, desc):
-        gradio_progress_wrapper(progress, pct, desc)
+        if progress:
+            try:
+                progress(pct)
+            except:
+                pass
+        add_debug_log(f"Model loading: {pct:.1%} - {desc}")
     
     add_debug_log("Starting model loading...")
     result = load_models_with_validation(progress_callback)
     add_debug_log(f"Model loading result: {result}")
     return result, get_debug_log()
 
+# ============================================================================ #
+# REFRESH PROCESSING INFO
+# ============================================================================ #
+def refresh_processing_info():
+    """Refresh the processing info display"""
+    return get_progress_info()
+
 # ============================================================================ #
 # MAIN INTERFACE CREATION WITH DEBUG PANEL
 # ============================================================================ #
@@ -287,6 +290,13 @@ def create_interface():
             border-radius: 8px;
             margin: 10px 0;
         }
+        .processing-info-box {
+            font-family: monospace;
+            background: #f0f0f0;
+            border: 2px solid #3498db;
+            border-radius: 8px;
+            padding: 12px;
+        }
         """
     ) as demo:
         gr.Markdown("# 🎬 Video Background Replacement")
@@ -407,15 +417,21 @@ def create_interface():
                 gr.Markdown("### 🎥 Result")
                 video_output = gr.Video(label="Processed Video", height=400)
                 
+                # Processing Info box with better styling
                 result_text = gr.Textbox(
-                    label="Processing Info",
+                    label="📊 Processing Info",
                     interactive=False,
-                    lines=6,
-                    placeholder="Results will appear here..."
+                    lines=10,
+                    placeholder="Processing statistics will appear here...\n\nWhen processing starts, you'll see:\n• Frame count\n• Processing speed (fps)\n• Time elapsed\n• ETA\n\nFinal results will show total time and performance stats.",
+                    elem_classes=["processing-info-box"]
                 )
                 
-                # Debug panel - OPEN by default for testing
-                with gr.Accordion("🔧 Debug Log", open=True):
+                # Manual refresh button for processing info
+                with gr.Row():
+                    refresh_info_btn = gr.Button("🔄 Refresh Progress", size="sm", visible=False)
+                
+                # Debug panel - can be closed
+                with gr.Accordion("🔧 Debug Log", open=False):
                     debug_output = gr.Textbox(
                         label="Debug Output",
                         value=get_debug_log(),
@@ -434,29 +450,7 @@ def create_interface():
             outputs=[status_text, debug_output]
         )
         
-        # Add timer to update progress display during processing
-        def update_progress_display():
-            """Update result_text with current progress"""
-            progress_info = get_progress_info()
-            if progress_info:
-                return progress_info
-            return gr.update()
-        
-        # Timer that updates every second during processing
-        timer = gr.Timer(1.0, active=False)
-        timer.tick(
-            fn=update_progress_display,
-            outputs=[result_text]
-        )
-        
-        # Main processing - ensure all inputs are connected
-        def process_with_timer(*args):
-            """Wrapper to activate timer during processing"""
-            timer.active = True
-            result = process_video_enhanced_fixed(*args)
-            timer.active = False
-            return result
-        
+        # Main processing
         process_btn.click(
             fn=process_video_enhanced_fixed,
             inputs=[
@@ -472,13 +466,19 @@ def process_with_timer(*args):
                 ai_prompt,            # ai_prompt
                 ai_style,             # ai_style
                 ai_generated_image,   # ai_img
-                use_two_stage,        # use_two_stage - THIS IS THE KEY INPUT
+                use_two_stage,        # use_two_stage
                 chroma_preset         # chroma_preset
             ],
             outputs=[video_output, result_text, debug_output],
             show_progress=True
         )
         
+        # Refresh processing info manually
+        refresh_info_btn.click(
+            fn=refresh_processing_info,
+            outputs=[result_text]
+        )
+        
         # Debug controls
         refresh_debug_btn.click(
             fn=get_debug_log,
@@ -497,10 +497,15 @@ def process_with_timer(*args):
             - **Single-Stage**: Fast, good quality for most videos
             - **Two-Stage**: Slower but cinema-quality edges using green screen
             
+            ### Performance:
+            - **720p videos**: ~2-3 fps processing speed
+            - **1080p videos**: ~1-2 fps processing speed
+            - **1-minute video**: Takes approximately 5-10 minutes
+            
             ### Tips:
-            - Start with 720p videos for testing
+            - Start with shorter clips (10-30 seconds) for testing
             - Two-stage mode works best with clear subjects
-            - Processing takes 2-4 minutes for 1-minute videos
+            - Check debug log if issues occur
             """)
 
     return demo